o tdf  @sddlZddlZddlZddlmZmZmZddlmZmZddl m Z m Z m Z m Z mZmZmZmZddlmZedZedZdefd d Zd e ed e efd dZdedejd e eeffddZde ed e e eeffddZded efddZeedddZdeeefd efddZdeeefd e eeeffdd Z deeefd e eeffd!d"Z!d#e ed$eegefd eee effd%d&Z"Gd'd(d(e#Z$d)d*Z%d0d,d-Z&d.d/Z'dS)1N) OrderedDictCounter defaultdict) FrameType TracebackType)IteratorListTupleIterableCallableUnionTypeVarMapping)ASTTextTR max_lengthcCsNt||kr%|t|d}|t||}|d|||| d}|S)N)len)seqrmiddlerightleftrY/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/stack_data/utils.pytruncates ritreturncCstt|SN)listrfromkeys)rrrrunique_in_ordersr!atoknodecCsdt|ttddrt||j\}}t||jd\}}||fS|j|dd\\}}\}}||dfS)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. match_caserF)padded) isinstancegetattrast line_rangepatternbodyZget_text_positions)r"r#start_end_startend_rrrr+s  r+lstc#st|tj|fdddD]L\}|rOtdgtfdd}tD]"|fdddurD|fdd|fd dq+ntd tEdHqdS) Ncs |dkS)Nrx)countsrr/ z"highlight_unique..keyFcs,z|}Wn tyYdSwd|<|S)NT) ValueError)fi) highlightedrrhighlight_index4s  z)highlight_unique..highlight_indexcs Srindexrgroupitemrrr8=s csdSNr'rAr)firstrDrErrr8?scsddddS)Nr%rArrCrrr8@sT)r itertoolsgroupbyrrsetrepeatzip)r3Z is_commonr@r)r7rGrDr?rErhighlight_unique,s    rMr6cCs|Srrr5rrridentityGsrN)mapperr;c cs|tt||}tjt|t|dddD]%\}}t|\}}|r+t||EdHqt|\} } |t|t| VqdS)NcSs |ddSrFr)trrrr8Or9z#collapse_repeated..r:)rmaprHrIrLrM) r3Z collapserrOr;ZkeyedZis_highlightedrDZoriginal_groupZhighlighted_groupZ keyed_groupr2rrrcollapse_repeatedKs   rR frame_or_tbcCs$tt|tjtjft|tjfSr)assert_r(typesrrrSrrris_frameYsrWccs2|}|r|Vt|r|j}n|j}|sdSdSr)rWf_backtb_next)rScurrentrrr iter_stack^s r[cCst|r ||jfS|j|jfSr)rWf_linenotb_frame tb_linenorVrrrframe_and_linenohs  r_iterablekey_funccCs(tt}|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])] )rrappend)r`raresultrErrrgroup_by_key_funcos rdc@s$eZdZdZddZddZeZdS)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 cCs|j|_||_dSr)__doc__func)selfrgrrr__init__s zcached_property.__init__cCs(|dur|S||}|j|jj<|Sr)rg__dict____name__)rhobj_clsvaluerrrcached_property_wrappersz'cached_property.cached_property_wrapperN)rk __module__ __qualname__rfriro__get__rrrrres recspddl}ddlm}Gfdddt|d}|dd}z ||||}W|Sty7|}Y|Sw)Nr)get_lexer_by_namecseZdZfddZZS)z'_pygmented_with_ranges..MyLexerc3sRdt|D]\}}tfddDr|j}t|7||fVq dS)Nrc3s,|]\}}|ko|knVqdSrr).0r.r1lengthrr s*zE_pygmented_with_ranges..MyLexer.get_tokens..)super get_tokensanyZ ExecutingNoder)rhtextttypern) __class__rangesrurrys  z2_pygmented_with_ranges..MyLexer.get_tokens)rkrprqry __classcell__rr~)r}rMyLexersrpython3F)stripnl)pygmentsZpygments.lexersrstype highlight Exception splitlines) formattercoder~rrsrlexerr?rrr_pygmented_with_rangess   rcCs|s t|tr t|}|dSr)r(strAssertionError) conditionerrorrrrrTs  rTcCs$zt|WSdt|jYS)Nz)rrrk)rnrrrsome_strs r)r)(r*rHrU collectionsrrrrrtypingrrr r r r r rZ asttokensrrrintrr!ASTr+boolrMrNrRrWr[r_rdobjectrerrTrrrrrs.(  "&" 0