B `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 W/home/kop/projects/devel/pgwui/test_venv/lib/python3.7/site-packages/py/_code/source.py !sz#Source.__init__..r )r)linesget isinstancertuplelistpybuiltinZ _basestringsplitstrippop getsourcerextend)selfpartskwargsrderpartZ 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+r1sourcer 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)rbeforeafterr7r2rr )r7r 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 r6)r7r r rdsz!Source.indent..)rr)rr7r2r )r7r r7_sz Source.indentFcCs|||\}}|||S)zc return Source statement which contains the given linenumber (counted from 0). )getstatementrange)rlineno assertionr+r1r 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)rr<r=astr+r1r r r r;nszSource.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)roffsetr2r 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)Nasdexec)compile)r r r r z$Source.isparseable..r FT)parser ImportErrorZsuiter#rKeyboardInterrupt Exception)rrrGZsyntax_checkerr3r r r isparseables zSource.isparseablecCs d|jS)Nr )joinr)rr r r __str__szSource.__str__rCcCsd|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..)rpathlocalchecksys _getframef_code co_filenamef_lineno_compilecounter __class__rLr cpy_compile SyntaxErrorexc_infor<rAappendtext _AST_FLAG linecachecache)rfilenamemodeflag dont_inherit _genframefnr<baser3coexZmsglinesZnewexrrar r r rDs:     zSource.compile)r4r4r5)r5)F)F)N)T)__name__ __module__ __qualname____doc__rYr!r%r.r0r*rr:r7r>r;rrKrMr compiler_flagrDr r r r rs$   rrCcCsLtdk 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&)rg)_astrASTr[rTrUrrD)r3rcrdflagsrfrgsrjr r r compile_s  ruc 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) )r4rNr)rcodeZCode TypeErrorinspect getsourcefilegetfilerQrR findsourceIOErrorZ firstlinenorr'AssertionError)objrvrhfspathr<_r r r getfslinenos&  rcCsTyt|\}}Wn"tjjk r,Yn dSt}dd|D|_||fS)N)NrcSsg|] }|qSr )r)r r6r r r rszfindsource..)rxr{rrZ_sysexrr)r~ sourcelinesr<r3r r r r{sr{cKsNtj|}yt|}Wntk r2d}YnXt|tsBtt |f|S)Nz<"Buggy python version consider upgrading, cannot get source") rrvZ getrawcoderxrIndentationErrorrr#r}r)r~rZstrsrcr 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 r4r )rr6r r r readline_generators z$deindent..readline_generatorcstS)N)nextr )itr r rE$rFzdeindent..r&) expandtabslstripr/rtokenizegenerate_tokensisspacer^ranger TokenErrorr) rrAr6rtnewlinesrrslineZelineir )rr 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@walkrrqZstmtZ ExceptHandlerr^r<getattrsortrr/) r<noder@lr namevalZ insert_indexr+r1r 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) Nr3rCir&rcss|]}|dVqdS)r Nr )r r r r r esz(getstatementrange_ast..cstS)N)rr )rr r rEgrFz'getstatementrange_ast..#)r#rD ValueErrorgetstatementrange_oldrr/rrx BlockFinderrstartedrr tokeneater EndOfBlockrlastrJr startswith) r<r3r=Zastnodecontentr+r1Z block_findertokr6r )rr 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)insertrLr\ OverflowErrorrr/rK) r<r3r=rr+r6ZtrylinesZ trysourcer1r r r rxs(      r)N)FN)F) __future__rbisectrrTrxrrtypesrrDr[rqrr`rHobjectrrprurr{rrrr?rr r r r s.    ;  * )