^Yc@s#ddlmZddlmZddlZddlZddlZddlZddlm Z e Z y ddl Z ddl m ZWnek rdZdZ nXdefdYZdd ejdd Zd Zd Zd ZddZdZeddZedZdS(i(t generators(t bisect_rightN(t ModuleType(t PyCF_ONLY_ASTitSourcecBseZdZdZdZdZdZdZdZdZ dddd Z dd Z e d Z e dZddZedZdZddejdddZRS(sY a immutable object holding a source code fragment, possibly deindenting it. ic Os<g|_}|jdt}|jdt}x|D]}|sMg}nt|trh|j}nt|ttfrg|D]}|jd^q}npt|tj j r|j d}|rx+|r|dj rPn|j qWqnt|d|j}|r't|}n|j|q8WdS(Ntdeindenttrstrips i(tlinestgettTruet isinstanceRttupletlistRtpytbuiltint _basestringtsplittstriptpopt getsourceRtextend( tselftpartstkwargsRtdeRtpartt partlinestx((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt__init__s*    % cCsLy|j|jkSWn1tk rGt|trCt||kStSXdS(N(RtAttributeErrorR tstrtFalse(Rtother((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt__eq__/s  cCsRt|tr|j|S|jdkr8tdn|j|j|jSdS(Nis!cannot slice a Source with a step(Ni( R tintRtsteptNonet IndexErrort __getslice__tstarttstop(Rtkey((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt __getitem__7s  cCs t|jS(N(tlenR(R((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt__len__?scCs t}|j||!|_|S(N(RR(RR'tendt newsource((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyR&Bs cCsdt|}}x.||krC|j|j rC|d7}qWx2||krx|j|dj rx|d8}qGWt}|j||!|j(|S(s] return new source object with trailing and leading blank lines removed. ii(R+RRR(RR'R-tsource((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRGs#' tt icCs\t|}t|}t}g|jD]}||^q+}|j||j|_|S(se return a copy of the source object with 'before' and 'after' wrapped around it. (RR(RtbeforetaftertindentR.tlineR((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt putaroundTs     cCs0t}g|jD]}||^q|_|S(sl return a copy of the source object with all lines indented by the given indent-string. (RR(RR4R.R5((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyR4_s #cCs#|j||\}}|||!S(sc return Source statement which contains the given linenumber (counted from 0). (tgetstatementrange(Rtlinenot assertionR'R-((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt getstatementgscCsSd|kot|kns1tdnt||\}}}||fS(s{ return (start, end) tuple which spans the minimal statement region which containing the given lineno. islineno out of range(R+R%tgetstatementrange_ast(RR8R9tastR'R-((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyR7ns"cCs#t}t|j||j(|S(s: 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. (RRR(RtoffsetR.((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRws cCsyddl}Wntk r,d}n X|j}|rQt|j}n t|}y||dWn%tk rntk rtSXtSdS(sb return True if source is parseable, heuristically deindenting it by default. iNcSst|ddS(Ntasdtexec(tcompile(R((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pytss ( tparsert ImportErrortsuiteRRtKeyboardInterruptt ExceptionRR (RRRBtsyntax_checkerR/((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt isparseables      cCsdj|jS(Ns (tjoinR(R((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt__str__sR?cCs| s%tjj|jddr|d krCtjd}n|jj|j }}d|j }|j j d7_ |s|d||f}q|d|||f}ndj |j d} yt| |||} Wntk rtjd} |j | j } | jr2| jd| jd n| jd |tdj | } | j| _| j| _| j| _| nRX|t@r| Sg|j D]}|d^q}dd ||ftjjj|<| Sd S( s return compiled code object. if filename is None invent an artificial filename which displays the source/line position of the caller frame. tfileiis <%d-codegen s%s:%d>s %r %s:%d>s R1t^s*(code was compiled probably from here: %s)N(R tpathtlocaltcheckR$tsyst _getframetf_codet co_filenametf_linenot_compilecountert __class__RIRt cpy_compilet SyntaxErrortexc_infoR8R=tappendttextt _AST_FLAGtstdt linecachetcache(Rtfilenametmodetflagt dont_inheritt _genframetfnR8tbaseR/tcotextmsglinestnewexRR((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyR@s8%          s s N(t__name__t __module__t__doc__RURR!R*R,R&RR6R4RR:R7R$RR RHRJRt compiler_flagR@(((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRs$          R?cCsntdk r4t|tjr4t|||||Stjd}t|}|j|||d|}|S(s 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. iRdN( t_astR$R tASTRWRPRQRR@(R/R`RatflagsRcRdtsRg((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pytcompile_s  cCsytjj|}Wntk ry.tjjj|pPtjjj|}Wntk rhdSX|rtjj |pd}d}|ryt |\}}Wqt k rqXqnX|j}|j }t|tst||fS(sx Return source location (path, lineno) for the given object. If the source cannot be determined return ("", -1) R0i(R0iN(R tcodetCodet TypeErrorR]tinspectt getsourcefiletgetfileRMRNR$t findsourcetIOErrort firstlinenoR R"tAssertionError(tobjRtRetfspathR8t_((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyt getfslinenos&      cCsxytjjj|\}}Wn!tjjk r;ndSXt}g|D]}|j^qS|_ ||fS(Ni(Ni( R R]RwRzRt_sysexR$RRR(R~t sourcelinesR8R/R5((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRzs "cKsatjj|}ytj|}Wntk r>d}nXt|tsTtt ||S(Ns<"Buggy python version consider upgrading, cannot get source"( R Rtt getrawcodeRwRtIndentationErrorR RR}R(R~Rtstrsrc((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRs  c s|dkr^xO|D]>}|j}|j}|rt|t|}PqqWd}n|dkrtt|Sg}d}||yxtjfdD]\}}\}}\}}}|t|krPn|t|krA||dj}|jr1|| jr1||}n|j|nx(t ||D]} |j|| qQWqWWnt tj fk rnX|j |t||S(Nicss0x|D]}|dVqWxtr+dVqWdS(Ns R0(R (RR5((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pytreadline_generators   cs tS(N(tnext((tit(s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRA#si( R$t expandtabstlstripR+R ttokenizetgenerate_tokenstisspaceRZtrangeRt TokenErrorR( RR=R5RrtnewlinesRRtslinetelineti((Rs+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyR s8         : c Csddl}g}x|j|D]}t|tjsLt|tjr"|j|jdxHdD]=}t||d}|rg|j|djddqgqgWq"q"W|j t ||}||d}|t |krd} n ||} || fS(Niit finalbodytorelsei(RR( R<twalkR Rotstmtt ExceptHandlerRZR8tgetattrR$tsortRR+( R8tnodeR<tlRtnametvalt insert_indexR'R-((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pytget_statement_startend27s  $ *   c s|dkrt|}tjd kr4|d7}nyt|ddd}Wqtk rt|||\}}d||fSXnt||\}}|dkrt|j }n||dkrzt j }|j |dj |_ d |j ||!Dy4x-tjfd D]}|j|q%WWqzt jtfk rf|j|}qztk rvqzXnxE|r|j |dj} | jd s| r|d8}q}Pq}W|||fS( Niis R/R?iiicss|]}|dVqdS(s N((t.0R((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pys fscs tS(N(R((R(s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRAhst#(ii(R$RRPt version_infoR@t ValueErrortgetstatementrange_oldRR+RRwt BlockFinderRtstartedRRt tokeneatert EndOfBlockRtlastRFRt startswith( R8R/R9tastnodetcontentR'R-t block_finderttokR5((Rs+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyR;Ns:         c CsWddlm}x-t|ddD]}|r|j|}d|krod|krod|krotdnd|krd|krq#qn|j||d !}|jd d d j|}y||Wnttt fk rq#nXxHt|d t |d D])}|||!}|j r||fSqWq#Wtd |fdS(s 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. i(tcompile_commandtsuperRRslikely a subclasstasserttraiseiis def xxx():s s%no valid source range around line %d N( tcodeopRRRR%tinsertRIRXt OverflowErrorRR+RH( R8R/R9RR'R5ttrylinest trysourceR-((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyRys( $ $  (t __future__RtbisectRRPRwRR ttypesRR@RWRoRR\RCR$tobjectRRnRsRRzRRRRR;R(((s+/tmp/pip-build-hU8Cw8/py/py/_code/source.pyts,       * +