B Mt\6@sddlmZddlmZddlZddlZddlZddlZddlm Z e Z yddl Z ddl m ZWnek r|dZdZ YnXGdddeZddejdfd d Zd d Zd dZddZdddZddZdddZdddZdS)) generators) bisect_rightN) ModuleType) PyCF_ONLY_ASTc@seZdZdZdZddZddZddZd d Zd d Z d dZ d%ddZ d&ddZ d'ddZ d(ddZd)ddZd*ddZd d!Zdd"ejddfd#d$ZdS)+SourcezY a immutable object holding a source code fragment, possibly deindenting it. rcOsg|_}|dd}|dd}x|D]}|s4g}t|trF|j}njt|ttfrddd|D}nLt|tjjr| d}|rx.|r|d rP| qWnt ||dj}|rt |}||q(WdS) NdeindentTrstripcSsg|]}|dqS) )r).0xr .lib/python3.7/site-packages/py/_code/source.py !sz#Source.__init__..r )r)linesget isinstancertuplelistpybuiltinZ _basestringsplitstrippop getsourcerextend)selfpartskwargsrZderpartZ partlinesr r r __init__s*       zSource.__init__cCs<y |j|jkStk r6t|tr2t||kSdSXdS)NF)rAttributeErrorrstr)rotherr r r __eq__/s    z Source.__eq__cCs:t|tr|j|S|jdkr&td||j|jSdS)N)Nz!cannot slice a Source with a step)rintrstep IndexError __getslice__startstop)rkeyr r r __getitem__7s    zSource.__getitem__cCs t|jS)N)lenr)rr r r __len__?szSource.__len__cCst}|j|||_|S)N)rr)rr*end newsourcer r r r)BszSource.__getslice__cCs|dt|}}x"||kr0|j|s0|d7}qWx&||krX|j|dsX|d8}q4Wt}|j|||jdd<|S)z] return new source object with trailing and leading blank lines removed. rr%N)r.rrr)rr*r0sourcer r r rGs  z Source.strip cs@t|}t|}t}fdd|jD}|j||j|_|S)ze return a copy of the source object with 'before' and 'after' wrapped around it. csg|] }|qSr r )r line)indentr r r[sz$Source.putaround..)rr)rZbeforeZafterr6r1rr )r6r putaroundTs zSource.putaroundcs t}fdd|jD|_|S)zl return a copy of the source object with all lines indented by the given indent-string. csg|] }|qSr r )r r5)r6r r rdsz!Source.indent..)rr)rr6r1r )r6r r6_sz Source.indentFcCs|||\}}|||S)zc return Source statement which contains the given linenumber (counted from 0). )getstatementrange)rlineno assertionr*r0r r r getstatementgszSource.getstatementcCs:d|krt|ks"ntdt||\}}}||fS)z{ return (start, end) tuple which spans the minimal statement region which containing the given lineno. rzlineno out of range)r.r(getstatementrange_ast)rr9r:astr*r0r r r r8nszSource.getstatementrangeNcCs t}t|j||jdd<|S)a: return a new source object deindented by offset. If offset is None then guess an indentation offset from the first non-blank line. Subsequent lines which have a lower indentation offset will be copied verbatim as they are assumed to be part of multilines. N)rrr)roffsetr1r r r rws zSource.deindentTcCsy ddl}Wntk r(dd}YnX|j}|rBt|}nt|}y||dWn(tk rpYntk rdSXdSdS)zb return True if source is parseable, heuristically deindenting it by default. rNcSs t|ddS)NZasdexec)compile)r r r r sz$Source.isparseable..r FT)parser ImportErrorZsuiter"rKeyboardInterrupt Exception)rrrBZsyntax_checkerr2r r r isparseables zSource.isparseablecCs d|jS)Nr )joinr)rr r r __str__szSource.__str__r?cCsd|rtj|jddr||dkr*td}|jj|j}}d|j }|j j d7_ |sj|d||f}n|d|||f}d |j d} yt | |||} Wntk r"td} |j d| j} | jr| d | jd | d |td | } | j| _| j| _| j| _| Yn>X|t@r2| Sd d |j D}ddl}dd||f|j|<| SdS)z return compiled code object. if filename is None invent an artificial filename which displays the source/line position of the caller frame. r)fileNr%z <%d-codegen z%s:%d>z %r %s:%d>r  ^z*(code was compiled probably from here: %s)cSsg|] }|dqS)r r )r r r r r rsz"Source.compile..)rpathlocalZchecksys _getframef_code co_filenamef_lineno_compilecounter __class__rGr cpy_compile SyntaxErrorexc_infor9r>appendtext _AST_FLAG linecachecache)rfilenamemodeflag dont_inherit _genframefnr9baser2coexZmsglinesZnewexrr[r r r r@s:     zSource.compile)r3r3r4)r4)F)F)N)T)__name__ __module__ __qualname____doc__rSr r$r-r/r)rr7r6r;r8rrFrHr compiler_flagr@r r r r rs$   rr?cCsLtdk r$t|tjr$t|||||Std}t|}|j||||d}|S)z compile the given source to a raw code object, and maintain an internal cache which allows later retrieval of the source code for the code object and any recursively created code objects. Nr%)ra)_astrZASTrUrNrOrr@)r2r]r^flagsr`rasrdr r r compile_s  rnc Csytj|}Wntk ryt|p4t|}Wntk rLdSX|r^tj|p`d}d}|ryt |\}}Wnt k rYnXYnX|j}|j }t |t st||fS)zx Return source location (path, lineno) for the given object. If the source cannot be determined return ("", -1) )r3rNr)rcodeZCode TypeErrorinspectZ getsourcefileZgetfilerLrM findsourceIOErrorZ firstlinenorr&AssertionError)objrorbfspathr9_r r r getfslinenos&  rxcCsTyt|\}}Wn"tjjk r,Yn dSt}dd|D|_||fS)N)NrcSsg|] }|qSr )r)r r5r r r rszfindsource..)rqrrrrZ_sysexrr)ruZ sourcelinesr9r2r r r rrsrrcKsNtj|}yt|}Wntk r2d}YnXt|tsBtt |f|S)Nz<"Buggy python version consider upgrading, cannot get source") rroZ getrawcoderqrIndentationErrorrr"rtr)rurZstrsrcr r r rs  rc sH|dkr@x6|D]*}|}|}|rt|t|}PqWd}|dkrPt|Sg}dd}||yxtfddD]\}}\}}\}}}|t|krP|t|kr||d}|r|d|r||d}||x t||D]} ||| qWqzWWnt tj fk r,YnX| |t|d|S)Nrcss(x|D]}|dVqWx dVqWdS)Nr r3r )rr5r r r readline_generators z$deindent..readline_generatorcstS)N)nextr )itr r rA$szdeindent..r%) expandtabslstripr.rtokenizegenerate_tokensisspacerXrangery TokenErrorr) rr>r5rmnewlinesrzrwZslineZelineir )r|r rs8 *    rc Csddl}g}xp||D]b}t|tjs4t|tjr||jdx4dD],}t||d}|rJ||djddqJWqW| t ||}||d}|t |krd} n||} || fS)Nrr%)Z finalbodyZorelse) r=walkrrkZstmtZ ExceptHandlerrXr9getattrsortrr.) r9Znoder=lr namevalZ insert_indexr*r0r r r get_statement_startend28s       rFc s^|dkrNt|}yt|ddd}Wn*tk rLt|||\}}d||fSXt||\}}|dkrnt|j}||dkrt}|j|d |_ dd|j||Dy*x$t fdd D]}|j |qWWn:tjtfk r|j|}Yntk rYnXx<|rR|j|d} | d sB| sL|d8}nPqW|||fS) Nr2r?ir%rcss|]}|dVqdS)r Nr )r r r r r esz(getstatementrange_ast..cstS)N)r{r )r|r r rAgsz'getstatementrange_ast..#)r"r@ ValueErrorgetstatementrange_oldrr.rrqZ BlockFinderrZstartedrrZ tokeneaterZ EndOfBlockryZlastrEr~ startswith) r9r2r:ZastnodeZcontentr*r0Z block_findertokr5r )r|r r<Os6   r<c Csddlm}xt|ddD]}|r^|j|}d|krLd|krLd|krLtdd|kr^d |kr^q|j||d }|dd d |}y ||Wnttt fk rwYnXx:t|d t |d D] }|||}| r||fSqWqWtd |fdS)z return (start, end) tuple which spans the minimal statement region which containing the given lineno. raise an IndexError if no such statementrange can be found. r)compile_commandrsuperrr zlikely a subclassassertraiser%z def xxx():z z%no valid source range around line %d N) Zcodeoprrrr(insertrGrV OverflowErrorrr.rF) r9r2r:rr*r5ZtrylinesZ trysourcer0r r r rxs(      r)N)FN)F)Z __future__rZbisectrrNrqrrtypesrr@rUrkrrZrCobjectrrjrnrxrrrrrr<rr r r r s.    ;  * )