U
    ARd-                     @   s   d dl Z d dlmZmZmZmZmZ d dlmZ	 d dl
mZ d dlmZ d dlmZ erdd dlmZ eeeee f dd	d
ZG dd dZdS )    N)TYPE_CHECKINGAnyListUnioncast)Json)gather_metrics)SessionStateProxy)UserInfoProxy)DeltaGenerator)oreturnc                 C   s   t | trt| S t| S )zRA repr function for json.dumps default arg, which tries to serialize sets as lists)
isinstancesetlistrepr)r    r   ;/tmp/pip-unpacked-wheel-b9et7o5g/streamlit/elements/json.py_ensure_serialization   s    
r   c                   @   s>   e Zd ZedddeeddddZeddd	d
ZdS )	JsonMixinjsonT)expandedr   )bodyr   r   c             
   C   s   ddl }t|ttfr| }t|tsztj|td}W nD t	k
r~ } z&|
d| d tj|dtd}W 5 d}~X Y nX t }||_||_| jd|S )	a  Display object or string as a pretty-printed JSON string.

        Parameters
        ----------
        body : object or str
            The object to print as JSON. All referenced objects should be
            serializable to JSON as well. If object is a string, we assume it
            contains serialized JSON.

        expanded : bool
            An optional boolean that allows the user to set whether the initial
            state of this json element should be expanded. Defaults to True.
            This argument can only be supplied by keyword.

        Example
        -------
        >>> import streamlit as st
        >>>
        >>> st.json({
        ...     'foo': 'bar',
        ...     'baz': 'boz',
        ...     'stuff': [
        ...         'stuff 1',
        ...         'stuff 2',
        ...         'stuff 3',
        ...         'stuff 5',
        ...     ],
        ... })

        .. output::
           https://doc-json.streamlitapp.com/
           height: 385px

        r   N)defaultzqWarning: this data structure was not fully serializable as JSON due to one or more unexpected keys.  (Error was: )T)skipkeysr   r   )Z	streamlitr   r	   r
   Zto_dictstrr   dumpsr   	TypeErrorwarning	JsonProtor   r   dgZ_enqueue)selfr   r   sterrZ
json_protor   r   r   r   #   s    )

"zJsonMixin.json)r   c                 C   s
   t d| S )zGet our DeltaGenerator.r   )r   )r"   r   r   r   r!   a   s    zJsonMixin.dgN)	__name__
__module____qualname__r   objectboolr   propertyr!   r   r   r   r   r   "   s   =r   )r   typingr   r   r   r   r   Zstreamlit.proto.Json_pb2r   r    Zstreamlit.runtime.metrics_utilr   Zstreamlit.runtime.stater	   Zstreamlit.user_infor
   Zstreamlit.delta_generatorr   r(   r   r   r   r   r   r   r   <module>   s   