o MTh  @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)seqrmiddlerightleftrS/home/air/segue/gemini/backup/venv/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_rangepatternbodyget_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 Srindexrgroupitemrrr9=s csdSNr'rBr)firstrErFrrr9?scsddddS)Nr%rBrrDrrr9@sT)r itertoolsgroupbyrrsetrepeatzip)r4 is_commonrAr)r8rHrEr@rFrhighlight_unique,s    rOr7cCs|Srrr6rrridentityGsrP)mapperr<c cs|tt||}tjt|t|dddD]%\}}t|\}}|r+t||EdHqt|\} } |t|t| VqdS)NcSs |ddSrGr)trrrr9Or:z#collapse_repeated..r;)rmaprIrJrMrO) r4 collapserrQr<keyedis_highlightedrEoriginal_grouphighlighted_group keyed_groupr3rrrcollapse_repeatedKs   rZ frame_or_tbcCs$tt|tjtjft|tjfSr)assert_r(typesrrr[rrris_frameYsr_ccs2|}|r|Vt|r|j}n|j}|sdSdSr)r_f_backtb_next)r[currentrrr iter_stack^s rccCst|r ||jfS|j|jfSr)r_f_linenotb_frame tb_linenor^rrrframe_and_linenohs  rgiterablekey_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)rhriresultrFrrrgroup_by_key_funcos rlc@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)selfrorrr__init__s zcached_property.__init__cCs(|dur|S||}|j|jj<|Sr)ro__dict____name__)rpobj_clsvaluerrrcached_property_wrappersz'cached_property.cached_property_wrapperN)rs __module__ __qualname__rnrqrw__get__rrrrrms rmcspddl}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/r2lengthrr s*zE_pygmented_with_ranges..MyLexer.get_tokens..)super get_tokensany ExecutingNoder)rptextttyperv) __class__rangesr}rrs  z2_pygmented_with_ranges..MyLexer.get_tokens)rsrxryr __classcell__rr)rrMyLexersrpython3F)stripnl)pygmentspygments.lexersr{type highlight Exception splitlines) formattercoderrr{rlexerr@rrr_pygmented_with_rangess   rcCs|s t|tr t|}|dSr)r(strAssertionError) conditionerrorrrrr\s  r\cCs$zt|WSdt|jYS)Nz)rrrs)rvrrrsome_strs r)r)(r*rIr] collectionsrrrrrtypingrrr r r r r r asttokensrrrintrr!ASTr+boolrOrPrZr_rcrgrlobjectrmrr\rrrrrs.(  "&" 0