a
    -gh                     @   sl   d dl Z d dlmZ d dlT d dlT dddZG d	d
 d
eZG dd deZdddZ	dd Z
dd ZdS )    N)PrettyPrinter)*      P   c           
   	   C   s   t | trd\}}n$t | tr(d\}}ntdt|  t| |krJ| }n(t| d| dg t| | d  }|ddd |D  | }d	tj||d
|d  d
|d  d	 }	|	S )a  
    Return a string representation for of a list where list is elided if
    it has more than n elements

    Parameters
    ----------
    v : list
        Input list
    threshold :
        Maximum number of elements to display

    Returns
    -------
    str
    )[])()zInvalid value of type: %sNz...z, c                 S   s   g | ]}t |qS  )str).0er   r   \/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/plotly/utils.py
<listcomp>$       z%_list_repr_elided.<locals>.<listcomp>
    )widthinitial_indentsubsequent_indent)

isinstancelisttuple
ValueErrortypelenjointextwrapwrapstrip)
v	threshold	edgeitemsindentr   Z	open_charZ
close_charZdisp_vZv_strZ	v_wrappedr   r   r   _list_repr_elided   s$    



(

r&   c                   @   s,   e Zd ZdZdd Zedd Zdd ZdS )	ElidedWrapperz
    Helper class that wraps values of certain types and produces a custom
    __repr__() that may be elided and is suitable for use during pretty
    printing
    c                 C   s   || _ || _|| _d S N)r"   r%   r#   )selfr"   r#   r%   r   r   r   __init__8   s    zElidedWrapper.__init__c                 C   s^   t d}t| ttfr4t| dkr4t| d ts4dS |rHt| |jrHdS t| trVdS dS d S )Nnumpyr   TF)
get_moduler   r   r   r   dictndarrayr   )r"   r+   r   r   r   is_wrappable=   s    (
zElidedWrapper.is_wrappablec              	   C   s
  t d}t| jttfr0t| j| j| jd}|S |rt| j|jr|	 }|j
f i t|| jddd | j }|d}dd| j  |}|j
f i | |S t| jt rt| jdkrdt| jd d	  d
 t| jdd   d S | j S n
| j S d S )Nr+   )r#   r%   r   r   )r#   r$   Z	linewidthr   r   r	      z ... ir
   )r,   r   r"   r   r   r&   r#   r%   r.   Zget_printoptionsZset_printoptionsr-   __repr__splitr   r   r   repr)r)   r+   resZ	orig_optsZ	res_linesr   r   r   r1   I   s*    

0zElidedWrapper.__repr__N)__name__
__module____qualname____doc__r*   staticmethodr/   r1   r   r   r   r   r'   1   s
   
r'   c                   @   s    e Zd ZdZdd Zdd ZdS )ElidedPrettyPrinterzF
    PrettyPrinter subclass that elides long lists/arrays/strings
    c                 O   s*   | dd| _tj| g|R i | d S )Nr#   r   )popr#   r   r*   )r)   argskwargsr   r   r   r*   t   s    zElidedPrettyPrinter.__init__c              	   C   sF   t |r,t || j|}| ||||||S t| ||||||S d S r(   )r'   r/   r#   _formatr   )r)   valstreamr%   Z	allowancecontextlevelZ
elided_valr   r   r   r>   x   s    
zElidedPrettyPrinter._formatN)r5   r6   r7   r8   r*   r>   r   r   r   r   r:   o   s   r:   r   c                 c   sT   t | tsdS | |fV  |  D ].\}}t |tr t|||f D ]
}|V  qBq dS )a  
    General, node-yielding generator.

    Yields (node, path) tuples when it finds values that are dict
    instances.

    A path is a sequence of hashable values that can be used as either keys to
    a mapping (dict) or indices to a sequence (list). A path is always wrt to
    some object. Given an object, a path explains how to get from the top level
    of that object to a nested value in the object.

    :param (dict) node: Part of a dict to be traversed.
    :param (tuple[str]) path: Defines the path of the current node.
    :return: (Generator)

    Example:

        >>> for node, path in node_generator({'a': {'b': 5}}):
        ...     print(node, path)
        {'a': {'b': 5}} ()
        {'b': 5} ('a',)

    N)r   r-   itemsnode_generator)nodepathkeyr?   itemr   r   r   rD      s    


rD   c                 C   s   |D ]}| | } q| S )aR  
    Iteratively get on obj for each key in path.

    :param (list|dict) obj: The top-level object.
    :param (tuple[str]|tuple[int]) path: Keys to access parts of obj.

    :return: (*)

    Example:

        >>> figure = {'data': [{'x': [5]}]}
        >>> path = ('data', 0, 'x')
        >>> get_by_path(figure, path)
        [5]
    r   )objrF   rG   r   r   r   get_by_path   s    
rJ   c              	   C   s   t | trht| D ]R\}}t |ttfr6t|| |< qt |trzt|| |< W q tyb   Y q0 qnt | trt|  t|   }}t	||D ]d\}}t |ttfrt|| |< n.t |trzt|| |< W n ty   Y n0 | 
|| t|< q| S r(   )r   r   	enumerater-   decode_unicoder   UnicodeEncodeErrorkeysvalueszipr;   )ZcollnoentryrN   valsrG   r?   r   r   r   rL      s*    




rL   )r   r   r   r   )r   )r   pprintr   Z_plotly_utils.utilsZ_plotly_utils.data_utilsr&   objectr'   r:   rD   rJ   rL   r   r   r   r   <module>   s   
)>
!