U nu f @sddlZddlZddlZddlmZmZmZddlmZmZddl m Z m Z m Z m Z mZmZmZmZddlmZedZedZedd d Ze ee ed d d Zeeje eefdddZe ee e eefdddZeedddZeedddZeeefedddZeeefe eeefdddZ eeefe eefddd Z!e eeegefeee efd!d"d#Z"Gd$d%d%e#Z$d&d'Z%d-d)d*Z&d+d,Z'dS).N) OrderedDictCounter defaultdict) FrameType TracebackType)IteratorListTupleIterableCallableUnionTypeVarMapping)ASTTextTR) max_lengthcCsNt||krJ|t|d}|t||}|d|||| d}|S)N)len)seqrZmiddlerightleftr4/tmp/pip-unpacked-wheel-8ocghofw/stack_data/utils.pytruncates  r)itreturncCstt|SN)listrfromkeys)rrrrunique_in_ordersr )atoknodercCsht|ttddr>t||j\}}t||jd\}}||fS|j|dd\\}}\}}||dfSdS)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. Z match_caserF)ZpaddedN) isinstancegetattrast line_rangepatternbodyZget_text_positions)r!r"startZ_end_startend_rrrr(s r()lstrc#st|tj|fdddD]\}|rtdgtfdd}tD]D|fdddk r|fdd|fd dqTn td tEdHqdS) Ncs |dkS)Nrx)countsrr/z"highlight_unique..keyFcs.z |}Wntk r YdSXd|<|S)NT) ValueError)fi) highlightedrrhighlight_index4s  z)highlight_unique..highlight_indexcs Srindexrgroupitemrrr4=r5csdSNr$r=r)firstr@rArrr4?r5csddddS)Nr#r=rr?rrr4@r5T)r itertoolsgroupbyrrsetrepeatzip)r/Z is_commonr<r)r3rCr@r;rArhighlight_unique,s   rI)r2rcCs|Srrr1rrridentityGsrJ)mapperr7c csztt||}tjt|t|dddD]J\}}t|\}}|rTt||EdHq*t|\} } |t|t| Vq*dS)NcSs |ddSrBr)trrrr4Or5z#collapse_repeated..r6)rmaprDrErHrI) r/Z collapserrKr7ZkeyedZis_highlightedr@Zoriginal_groupZhighlighted_groupZ keyed_groupr.rrrcollapse_repeatedKs   rN) frame_or_tbrcCs$tt|tjtjft|tjfSr)assert_r%typesrrrOrrris_frameYsrSccs*|}|r&|Vt|r|j}q|j}qdSr)rSf_backtb_next)rOcurrentrrr iter_stack^s rWcCs"t|r||jfS|j|jfSdSr)rSf_linenotb_frame tb_linenorRrrrframe_and_linenohs r[)iterablekey_funcrcCs(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\r]resultrArrrgroup_by_key_funcos r`c@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)selfrcrrr__init__szcached_property.__init__cCs(|dkr |S||}|j|jj<|Sr)rc__dict____name__)rdobj_clsvaluerrrcached_property_wrappersz'cached_property.cached_property_wrapperN)rg __module__ __qualname__rbrerk__get__rrrrrasracsnddl}ddlm}Gfdddt|d}|dd}z||||}Wntk rd|}YnX|S)Nr)get_lexer_by_namecseZdZfddZZS)z'_pygmented_with_ranges..MyLexerc3sPdt|D]:\}}tfddDr4|j}t|7||fVqdS)Nrc3s*|]"\}}|ko|knVqdSrr).0r+r-lengthrr szE_pygmented_with_ranges..MyLexer.get_tokens..)super get_tokensanyZ ExecutingNoder)rdtextttyperj) __class__rangesrqrrus  z2_pygmented_with_ranges..MyLexer.get_tokens)rgrlrmru __classcell__rrz)ryrMyLexersr}python3F)Zstripnl)pygmentsZpygments.lexersrotypeZ highlight Exception splitlines) formattercoderzrror}lexerr;rr|r_pygmented_with_rangess   rcCs|st|trt|}|dSr)r%strAssertionError) conditionerrorrrrrPs rPcCs*z t|WSdt|jYSXdS)Nz)rrrg)rjrrrsome_strs r)r)(r'rDrQ collectionsrrrrrtypingrrr r r r r rZ asttokensrrrintrr ASTr(boolrIrJrNrSrWr[r`objectrarrPrrrrrs,(  $ ,