g8ddlmZddlmZddlZddlZddlZddlZddlm Z e Z ddl Z ddl m Zn#e$rdZdZ YnwxYwGddeZddejdfd Zd Zd Zd Zdd ZdZddZddZdS)) generators) bisect_rightN) ModuleType) PyCF_ONLY_ASTceZdZdZdZdZdZdZdZdZ dZ dd Z dd Z ddZ ddZddZddZdZddejddfdZdS)SourcezY a immutable object holding a source code fragment, possibly deindenting it. rcZgx|_}|dd}|dd}|D]}|sg}t|tr|j}nt|tt fr d|D}nt|t jjrK| d}|r3|r1|d rn| |1nt||j}|rt|}||dS)NdeindentTrstripc8g|]}|dS r .0xs J/granary/edev/quantum/lib/python3.11/site-packages/_pytest/_code/source.py z#Source.__init__..!s":::QXXd^^:::r)r )linesget isinstancertuplelistpybuiltin _basestringsplitstrippop getsourcer extend)selfpartskwargsrder part partliness r__init__zSource.__init__sL U ZZ D ) )Hd++ $ $D  $'' ? J D5$-00 ?::T::: D"*"899 ? JJt,, (#($R=..00"!! $( &dR888>  0$Y// LL # # # #% $ $rc |j|jkS#t$r.t|trt||kcYSYdSwxYw)NF)rAttributeErrorrstr)r$others r__eq__z Source.__eq__/sa :, ,   %%% *4yyE))))55 s2A  A Nct|tr |j|S|jdvrt dt }|j|j|j|_|S)N)Nz!cannot slice a Source with a step)rintrstep IndexErrorrstartstop)r$key newsources r __getitem__zSource.__getitem__9sb c3   :c? "xy(( !DEEEI"j38);.Zs999d6D=999rrr)r$beforeafterrIr8rs ` r putaroundzSource.putaroundSsZu HH 9999dj999 ,.%+= rcTt}fd|jD|_|S)zl return a copy of the source object with all lines indented by the given indent-string. cg|]}|zSrFrFrGs rrz!Source.indent..cs@@@TF4K@@@rrJ)r$rIr8s ` rrIz Source.indent^s1HH @@@@TZ@@@ rFcH|||\}}|||S)zc return Source statement which contains the given linenumber (counted from 0). )getstatementrange)r$lineno assertionr5r@s r getstatementzSource.getstatementfs+++FI>> sE#Ircd|cxkrt|ksntdt||\}}}||fS)z{ return (start, end) tuple which spans the minimal statement region which containing the given lineno. rzlineno out of range)r<r4getstatementrange_ast)r$rRrSastr5r@s rrQzSource.getstatementrangemsWV''''c$ii''''233 3/==UCczrc`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)rr r)r$offsetr8s rr zSource.deindentvs/HH %dj&99 rTc ddl}|j}n#t$rd}YnwxYw|r"t|}nt|} ||dzdS#t $rt $rYdSwxYw)zb return True if source is parseable, heuristically deindenting it by default. rNc$t|ddS)Nasdexec)compile)rs rz$Source.isparseable..swq%'@'@rrTF)parsersuite ImportErrorr-r KeyboardInterrupt Exception)r$r r`syntax_checkerrAs r isparseablezSource.isparseables * MMM$\NN A A A@@NNN A  ))FFYYF  N6$; ' ' ' 4 !       55 s A%%A=<A=c6d|jS)Nr)joinrr=s r__str__zSource.__str__syy$$$rr]cD|r3tj|dr`|t jd}|jj|j}}d|j z}|j xj dz c_ |s |d||fzz}n |d|||fzz}d |j dz} t| |||} |tzr| Sd |j D} tjdd krHt!d } dtjjj|<| tjjjd<d| _dd| |ftjjj|<| S#t0$rt jd} |j d| j}| jr |d | jzd z|d|zt1d |}| j|_| j|_| j|_|wxYw)z return compiled code object. if filename is None invent an artificial filename which displays the source/line position of the caller frame. r)fileNr1z <%d-codegen z%s:%d>z %r %s:%d>rcg|]}|dzSr rFrs rrz"Source.compile..s444Aa$h444r_pycodecompile_pseudo_module ^z*(code was compiled probably from here: %s))rpathlocalchecksys _getframef_code co_filenamef_lineno_compilecounter __class__rhr cpy_compile _AST_FLAG version_inforstdinspect modulesbyfilemodules __loader__ linecachecache SyntaxErrorexc_inforRrYappendtext)r$filenamemodeflag dont_inherit _genframefnrRbaserAcormexmsglinesnewexs rr^zSource.compiles. G27==2288a8@@ G M!,, !(4i6HvB!D$88D N * *a / * * G(b&\"99+2v0F"FF4:&&- VXtT::Bi  44444E"a''=>>9=,X6+, "4(  014/IBF  "8 ,I3   "Bz*29*-Hy 5BI  3444 OOH8S T T T ( 3 344E9EL9ELEJK s 4E B?H)rBrBrC)rCFr;)T)__name__ __module__ __qualname____doc__ryr*r/__hash__r9r>r rMrIrTrQr rfrir compiler_flagr^rFrrrrs$O$$$0H           2%%% $&-$------rrr]ct-t|tjrt|||||St jd}t |}|||||}|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. Nr1)r)_astrASTr{rtrurr^)rArrflagsrrsrs rcompile_rsi Jvtx8868T5,GGG a  IvA 8T5I > >B Ircddl} |j|}|j}|j}n#t $r t jj |p#t jj |}n#t $rYYdSwxYw|rt j |pd}d}|r$ t|\}}n#t$rYnwxYwYnwxYwt|tsJ||fS)zx Return source location (path, lineno) for the given object. If the source cannot be determined return ("", -1) rN)rBrr) _pytest._code_codeCoderq firstlineno TypeErrorrr~r getsourcefilegetfilerr findsourceIOErrorrr2)obj_pytestcodefspathrRr_s r getfslinenors?"}!!#&& !!     &...s33.&.((-- B   666 ) b))1T   &sOO 66     " fc " """" 6>sW/ C#ABC# B C#B*C#=CC# CC#CC#"C#c tjj|\}}n#tjj$rYdSxYwt }d|D|_||fS)N)Nrc6g|]}|SrFr)rrHs rrzfindsource.. s :::dDKKMM:::r)rr~rrr_sysexrr)r sourcelinesrRrAs rrrsw fn77<< VV :  xx XXF::k:::FL 6>s '*Ac ddl}|j|} tj|}n#t $rd}YnwxYwt |tsJt|fi|S)Nrz<"Buggy python version consider upgrading, cannot get source") rr getrawcoderr"IndentationErrorrr-r)rr&rstrsrcs rr"r" s - " "3 ' 'CR"3'' RRRQR fc " """" & # #F # ##s5 AAcZ |R|D]M}|}|}|r!t|t|z }nNd}|dkrt|Sg}d}|| t j fdD]\}}\}}\}}}|t|krn|t|krl||dz }|r&|d|r ||d}||t||D]} ||| n#ttj f$rYnwxYw| |t|d|S)Nrc3*K|D] }|dzV dV)NrTrBrF)rrHs rreadline_generatorz$deindent..readline_generator#s;  D+     HHH rc"tSr;nextitsrr_zdeindent..,sPTUWPXPXrr1) expandtabslstripr<rtokenizegenerate_tokensisspacerranger TokenErrorr#) rrYrHrnewlinesrrslineelineirs @rr r s ~  D??$$D A T3q66) F {{E{{H  E " "B /7/GHXHXHXHX/Y/Y * * +Aq*5!jua!s5zz!!s8}}$$UQY'2244;;==)T'6']%:%:%<%<)=D%%%5%(( * *a)))) * h1 2      OOE#h--..)*** Os=E>cddl}g}||D]}t|tjst|tjr[||jdz dD];}t||d}|r&||djdz dz <| t||}||dz }|t|krd} n||} || fS)Nrr1) finalbodyorelse) rWwalkrrstmt ExceptHandlerrrRgetattrsortrr<) rRnoderWlrnameval insert_indexr5r@s rget_statement_startend2r@s JJJ A XXd^^44 a # # 4z!T5G'H'H 4 HHQX\ " " "- 4 4at,,4HHSV]Q.2333FFHHH6**L lQ Es1vv o #:rFc |at|}tjdkr|dz } t|ddd}n)#t$rt |||\}}d||fcYSwxYwt ||\}}|t|j}||dzkrtj }|j|d |_ d|j||D tj fd D] }|j| n1#tjt"f$r |j|z}Ynt&$rYnwxYw|rB|j|dz } | d s| s|dz}nn|B|||fS) N)rrAr]ir1rc3 K|] }|dzV dS)rNrFrs r z(getstatementrange_ast..os& : :Qq4x : : : : : :rc"tSr;rrsrr_z'getstatementrange_ast..qs Rr#)r-rtr}r^ ValueErrorgetstatementrange_oldrr<rr BlockFinderrstartedrr tokeneater EndOfBlockrlastrdr startswith) rRrArSastnodecontentr5r@ block_findertokrHrs @rrVrVWsf++  e # # tOG $gx>>GG $ $ $.vvyIIJE3# # # # $)99JE3  {&, UQY*,, %|E215==??  : :&,uSy"9 : : : /0@0@0@0@AA . .' '--- ."$45 , , ,#e+CCC    D  |C!G$++-- ??3   t  1HCC   E3 s'<#A"!A".$D E5 EEc(ddlm}t|ddD]}|r1|j|}d|vrd|vrd|vrt dd|vrd |vr5|j||d z}|dd d |} ||n#tttf$rYwxYwt|d zt|d zD](}|||}| r||fccS)td |fz)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_commandrsuperr$r*zlikely a subclassassertraiser1z def xxx():z z%no valid source range around line %d ) codeoprrrr4insertrhr OverflowErrorrr<rf) rRrArSrr5rHtrylines trysourcer@s rrrs'&&&&&vr2&&""  <&D$6T>>jD6H6H !4555t##t(;(;<fQh/ <(((JJx((   OI & & & &]J7    H 3v;;q=11 " "CuSy)I$$&& "cz!!!!! " " = I J JJs BB10B1r;)FNr) __future__rbisectrrtrrrtypesrr^r{rrr|rbobjectrrrrrr"r rrVrrFrrrs!!!!!!   KKK///////I DDD zzzzzVzzz@#  $1     >   $$$''''T.((((V K K K K K Ks 1 ==