o
    ˷e                  	   @   s  d dl Z d dlZd dlZd dlmZmZmZ d dlmZmZ d dl	m
Z
mZmZmZmZmZmZmZ edZedZdefdd	Zd
ee dee fddZde jdeeef fddZdee de
eeef  fddZdedefddZeedddZdeeef defddZdeeef de
eeef  fddZdeeef deeef fdd Zd!ee d"eegef deeee f fd#d$Z G d%d& d&e!Z"d'd( Z#d,d*d+Z$dS )-    N)OrderedDictCounterdefaultdict)	FrameTypeTracebackType)IteratorListTupleIterableCallableUnionTypeVarMappingTR
max_lengthc                 C   sN   t | |kr%|t | d }|t | | }| d | | | | d   } | S )N   )len)seqr   middlerightleft r   G/var/www/ideatree/venv/lib/python3.10/site-packages/stack_data/utils.pytruncate   s
   r   itreturnc                 C   s   t t| S N)listr   fromkeys)r   r   r   r   unique_in_order   s   r    nodec                 C   sJ   z| j jd | jjd d fW S  ty$   | jt| d| jd f Y S w )z
    Returns a pair of numbers representing a half open range
    (i.e. suitable as arguments to the `range()` builtin)
    of line numbers of the given AST nodes.
    r      
end_lineno)first_tokenstart
last_tokenendAttributeErrorlinenogetattr)r!   r   r   r   
line_range   s   
r+   lstc                 #   s    t |  tj|  fdddD ]L\}|rOtdgt fdd}tD ]"|fddd urD|fdd |fd	d q+ntd
tE d H  qd S )Nc                    s    |  dkS )N   r   x)countsr   r   <lambda>0       z"highlight_unique.<locals>.<lambda>keyFc                    s,   z|  }W n
 t y   Y d S w d |< |S )NT)
ValueError)fi)highlightedr   r   highlight_index5   s   
z)highlight_unique.<locals>.highlight_indexc                      s
     S r   indexr   groupitemr   r   r1   >   s   
 c                      s     d S Nr"   r:   r   )firstr=   r>   r   r   r1   @   s    c                      s   d d d d   S )Nr:   r   r<   r   r   r1   A   s    T)r   	itertoolsgroupbyr   r   setrepeatzip)r,   	is_commonr9   r   )r0   r@   r=   r8   r>   r   highlight_unique-   s    
rH   r/   c                 C   s   | S r   r   r.   r   r   r   identityH   s   rI   )mapperr4   c                c   s|    t t|| }tjt| t|dd dD ]%\}}t| \}}|r+t||E d H  qt| \}	}
|t |t |	V  qd S )Nc                 S   s   | d d S r?   r   )tr   r   r   r1   P   r2   z#collapse_repeated.<locals>.<lambda>r3   )r   maprB   rC   rF   rH   )r,   	collapserrJ   r4   keyedis_highlightedr=   original_grouphighlighted_groupkeyed_group_r   r   r   collapse_repeatedL   s   rT   frame_or_tbc                 C   s$   t t| tjtjf t| tjfS r   )assert_
isinstancetypesr   r   rU   r   r   r   is_frameZ   s   rZ   c                 c   s.    | r| V  t | r| j} n| j} | sd S d S r   )rZ   f_backtb_nextrY   r   r   r   
iter_stack_   s   r]   c                 C   s   t | r	| | jfS | j| jfS r   )rZ   f_linenotb_frame	tb_linenorY   r   r   r   frame_and_linenoh   s   
ra   iterablekey_funcc                 C   s(   t t}| D ]}||| | q|S )a  
    Create a dictionary from an iterable such that the keys are the result of evaluating a key function on elements
    of the iterable and the values are lists of elements all of which correspond to the key.

    >>> def si(d): return sorted(d.items())
    >>> si(group_by_key_func("a bb ccc d ee fff".split(), len))
    [(1, ['a', 'd']), (2, ['bb', 'ee']), (3, ['ccc', 'fff'])]
    >>> si(group_by_key_func([-1, 0, 1, 3, 6, 8, 9, 2], lambda x: x % 2))
    [(0, [0, 6, 8, 2]), (1, [-1, 1, 3, 9])]
    )r   r   append)rb   rc   resultr>   r   r   r   group_by_key_funco   s   rf   c                   @   s$   e Zd ZdZdd Zdd ZeZdS )cached_propertyz
    A property that is only computed once per instance and then replaces itself
    with an ordinary attribute. Deleting the attribute resets the property.

    Based on https://github.com/pydanny/cached-property/blob/master/cached_property.py
    c                 C   s   |j | _ || _d S r   )__doc__func)selfri   r   r   r   __init__   s   
zcached_property.__init__c                 C   s(   |d u r| S |  | }|j| j j< |S r   )ri   __dict____name__)rj   obj_clsvaluer   r   r   cached_property_wrapper   s   z'cached_property.cached_property_wrapperN)rm   
__module____qualname__rh   rk   rq   __get__r   r   r   r   rg      s
    rg   c                    sL   dd l }ddlm} G  fdddt|d}|dd}||||  S )Nr   )get_lexer_by_namec                       s   e Zd Z fddZ  ZS )z'_pygmented_with_ranges.<locals>.MyLexerc                 3   sR    d t  |D ]\}}t fddD r|j} t|7  ||fV  q	d S )Nr   c                 3   s,    | ]\}}|   ko|k n  V  qd S r   r   ).0r%   r'   lengthr   r   	<genexpr>   s   * zE_pygmented_with_ranges.<locals>.MyLexer.get_tokens.<locals>.<genexpr>)super
get_tokensanyExecutingNoder   )rj   textttyperp   )	__class__rangesrw   r   r{      s   z2_pygmented_with_ranges.<locals>.MyLexer.get_tokens)rm   rr   rs   r{   __classcell__r   r   )r   r   MyLexer   s    r   python3F)stripnl)pygmentspygments.lexersru   type	highlight
splitlines)	formattercoder   r   ru   r   lexerr   r   r   _pygmented_with_ranges   s
   
	r    c                 C   s   | st |trt|}|d S r   )rW   strAssertionError)	conditionerrorr   r   r   rV      s
   
rV   )r   )%astrB   rX   collectionsr   r   r   r   r   typingr   r   r	   r
   r   r   r   r   r   r   intr   r    ASTr+   boolrH   rI   rT   rZ   r]   ra   rf   objectrg   r   rV   r   r   r   r   <module>   s*    ("&"	0