a j@svdZddlZddlZddlZddlZddlZddlZddlmZe e Z ddZ d ddZ d d ZGd d d ZdS)zSimple traceback introspection. Used to add additional information to AssertionErrors in tests, so that failure messages may be more informative.N)StringIOc CsXtd||jr|j}q |j}t|\}}t||\}}ttd |}t |j |j }|rzt |jD] }||qnWqt jy} zXtd| |d|d8}ttd |}t |j |j }WYd} ~ q\WYd} ~ qd} ~ 00qq\g} |jrN|jd} d} | D]6} | |kr4| d| n| d | | d7} qd | S) zInspect a traceback and its frame, returning source for the expression where the exception was raised, with simple variable replacement performed and the line on which the exception was raised marked with '>>' zinspect traceback %szTokenizer error: %srN z>> z )logdebugtb_nexttb_frametbsourcefind_inspectable_linesrtextwrapdedentjoinExpanderf_locals f_globalstokenizegenerate_tokensreadline TokenErrorpopexpanded_sourcesplitappend)tbframelinesZexc_lineZ inspect_linesZ mark_linesrcexptokepaddedZ exp_linesepliner$4E:\soccer\ml_env\Lib\site-packages\nose/inspector.pyinspect_tracebacks<    "   r&c Cs|j}|j}|dkr|d|d}td||zt|\}}Wntybdgd}}Yq0|}t|d}tdt|t ||}||||}|d|}|dkr||d dr|d8}||||}qn dgd}}td||||fS)a9Get source from a traceback object. A tuple of two things is returned: a list of lines of context from the source code, and the index of the current line within that list. The optional second argument specifies the number of lines of context to return, which are centered around the current line.rrzlineno: %s start: %sr\z'tbsource lines '''%s''' around index %s) tb_linenor rrinspect findsourceIOErrormaxminlenstripendswith) rcontextlinenorstartrdummyindex all_linesr$r$r%r 3s*    r c Cs4td}td}td}g}||}||d}|t|dd|}|||dt|dt|} |D]*} || d|kr|| q|qq||||t|d} | |} | D]H} | s|| d|kr| | s|| | | } qqqt d|| || fS) aFind lines in home that are inspectable. Walk back from the err line up to 3 lines, but don't walk back over changes in indent level. Walk forward up to 3 lines, counting \ separated lines as 1. Don't walk over changes in indent level (unless part of an extended line)z \\[\s\n]*$z :[\s\n]*$z^(\s*)rrz#Inspecting lines '''%s''' around %s) recompilematchgroupsr.reverser/r0rsearchrr) rposZcntZdfindZ toinspecthomeZ home_indentbeforeafterr#Zhome_pos continuedr$r$r%r Rs:         r c@s eZdZdZddZddZdS)rzsSimple expression expander. Uses tokenize to find the names and expands any that can be looked up in the frame.cCs||_||_d|_d|_dS)Nr)localsglobalslposr)selfrGrHr$r$r%__init__zszExpander.__init__c CsF|jdurZ|d|jkr6|jd|d|j7_n$|d|jkrZ|jd|d7_|d|_|tjkrpn|tjkr z*|j|}t|tjj r|}nt |}WnTt yz*|j |}t|tjj r|}nt |}Wnt y|}Yn0Yn0|j|7_n|j|7_t d||ddrB|jd7_dS)Nr z\s+\\\nz \ )rIrrINDENTNAMErG isinstance collectionsabcCallablereprKeyErrorrHr;r=)rJttyperr5endr#valr$r$r%__call__s4          zExpander.__call__N)__name__ __module__ __qualname____doc__rKrXr$r$r$r%rwsr)r')r\r+loggingr;r rrPior getLoggerrYrr&r r rr$r$r$r%s  % %