^Yc@`s%ddlmZmZmZmZddlmZddlZddlZddl Z ddl Z e Z y ddl Z ddl mZWnek rdZdZ nXdefdYZddejddZd Zd Zd Zdd Zd ZeddZedZdS(i(tabsolute_importtdivisiont generatorstprint_function(t bisect_rightN(t PyCF_ONLY_ASTtSourcecB`seZdZdZdZdZdZdZdZ dZ dddd Z dd Z e d Ze d ZddZedZdZddejdddZRS(sY a immutable object holding a source code fragment, possibly deindenting it. ic O`s<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( tselftpartstkwargsR tdeRtpartt partlinestx((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt__init__s*    % cC`sLy|j|jkSWn1tk rGt|trCt||kStSXdS(N(R tAttributeErrorR tstrtFalse(Rtother((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt__eq__.s  cC`sbt|tr|j|S|jdkr8tdnt}|j|j|j!|_|SdS(Nis!cannot slice a Source with a step(Ni( R tintR tsteptNonet IndexErrorRtstarttstop(Rtkeyt newsource((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt __getitem__8s  cC`s t|jS(N(tlenR (R((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt__len__BscC`sdt|}}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-R RR(RR(tendtsource((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyREs#' tt icC`s\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 ((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt putaroundRs     cC`s0t}g|jD]}||^q|_|S(sl return a copy of the source object with all lines indented by the given indent-string. (RR (RR5R+R6((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyR5]s #cC`s#|j||\}}|||!S(sc return Source statement which contains the given linenumber (counted from 0). (tgetstatementrange(Rtlinenot assertionR(R/((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt getstatementescC`sSd|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(RR9R:tastR(R/((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyR8ls"cC`s#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+((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyRus cC`syddl}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. iNcS`st|ddS(Ntasdtexec(tcompile(R((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pytss ( tparsert ImportErrortsuiteR RtKeyboardInterruptt ExceptionR!R (RRRCtsyntax_checkerR0((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt isparseables      cC`sdj|jS(Ns (tjoinR (R((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt__str__sR@cC`s| 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 R2t^s*(code was compiled probably from here: %s)N(RtpathtlocaltcheckR&tsyst _getframetf_codet co_filenametf_linenot_compilecountert __class__RJR t cpy_compilet SyntaxErrortexc_infoR9R>tappendttextt _AST_FLAGtstdt linecachetcache(Rtfilenametmodetflagt dont_inheritt _genframetfnR9tbaseR0tcotextmsglinestnewexRR ((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyRAs8%          Ns s (t__name__t __module__t__doc__RVRR#R&t__hash__R,R.RR7R5R!R;R8RR RIRKRt compiler_flagRA(((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyRs$        R@cC`sntdk 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. iReN( t_astR&R tASTRXRQRRRRA(R0RaRbtflagsRdRetsRh((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pytcompile_s  cC`sddl}y|jj|}Wntk ry.tjjj|p\tjjj|}Wntk rtdSX|rtj j |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) iNR1i(R1i(t _pytest._codet_codetCodet TypeErrorRR^tinspectt getsourcefiletgetfileRNROR&t findsourcetIOErrort firstlinenoR R$tAssertionError(tobjt_pytesttcodeRftfspathR9t_((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyt getfslinenos(       cC`sxytjjj|\}}Wn!tjjk r;ndSXt}g|D]}|j^qS|_ ||fS(Ni(Ni( RR^RzR}Rt_sysexR&RRR (Rt sourcelinesR9R0R6((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyR}s "cK`smddl}|jj|}ytj|}Wntk rJd}nXt|ts`tt ||S(Nis<"Buggy python version consider upgrading, cannot get source"( RvRwt getrawcodeRzRtIndentationErrorR R RR(RRRtstrsrc((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_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(Nics`s0x|D]}|dVqWxtr+dVqWdS(Ns R1(R (R R6((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pytreadline_generators   c`s tS(N(tnext((tit(s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyRB&si( R&t expandtabstlstripR-Rttokenizetgenerate_tokenstisspaceR[trangeRt TokenErrorR( R R>R6RttnewlinesRRtslinetelineti((Rs4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyRs8         : c C`sddl}g}x|j|D]}t|tjsLt|tjr"|j|jdxHdD]=}t||d}|rg|j|djddqgqgWq"q"W|j t ||}||d}|t |krd} n ||} || fS(Niit finalbodytorelse(RR( R=twalkR Rqtstmtt ExceptHandlerR[R9tgetattrR&tsortRR-( R9tnodeR=tlRtnametvalt insert_indexR(R/((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pytget_statement_startend2:s  $ *   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 R0R@iiics`s|]}|dVqdS(s N((t.0R((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pys isc`s tS(N(R((R(s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyRBkst#(ii(R&R RQt version_infoRAt ValueErrortgetstatementrange_oldRR-R Rzt BlockFinderRtstartedRRt tokeneatert EndOfBlockRtlastRGRt startswith( R9R0R:tastnodetcontentR(R/t block_finderttokR6((Rs4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyR<Qs:         c C`sWddlm}x-t|ddD]}|r|j|}d|krod|krod|krotdnd|krd |krq#qn|j||d !}|jdd 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_commanditsuperRRslikely a subclasstasserttraiseis def xxx():s s%no valid source range around line %d N( tcodeopRRR R'tinsertRJRYt OverflowErrorRR-RI( R9R0R:RR(R6ttrylinest trysourceR/((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyR|s( $ $  (t __future__RRRRtbisectRRQRzRRRARXRqRR]RDR&tobjectRRpRuRR}RRRR!R<R(((s4/tmp/pip-build-hU8Cw8/pytest/_pytest/_code/source.pyts*"       + +