^jdZddlZddlmZddlmZGddZGddZGd d eZGd d ZGd deZ y)zjutilities for analyzing expressions and blocks of Python code, as well as generating Python from AST nodesN) exceptions)pyparserceZdZdZddZy) PythonCodezDDTJ JM>>,# DD  $ $T >-= >  N)r__name__ __module__ __qualname____doc__rrrrrs F'rrceZdZdZdZy) ArgumentListzBparses a fragment of code as a comma-separated list of expressionsc jg|_g|_t|_t|_t |t rItjd|rtjd|s|dz }tj|dfi|}n|}tj|tfi|}|j|y)Nz\Sz,\s*$,r )codeargsargsr r rrrrematchrr FindTuplerr)rr rrrs rrzArgumentList.__init__@s  $'E!&)e# dC xxt$RXXh-E >>$C2BCDD   tZ D3C D  rNrr$rrr&r&=s Lrr&c"eZdZdZfdZxZS)PythonFragmentzextends PythonCode to provide identifier lookups in partial control statements e.g.:: for x in 5: elif y==9: except (MyException, e): c tjd|jtj}|st j d|zfi||j dr|d|jd}|j dd\}}d}|dvr|dz}nN|d k(r|d z}nC|d vr d |zdz}d }n4|dk(r d|zdz}d }n$|dk(r|dz}nt j d|zfi|t|$|fd|i|y)Nz^(\w+)(?:\s+(.*?))?:\s*(#|$)z0Fragment '%s' is not a partial control statementr)forifwhilepasstryzpass except:pass)elifelsezif False:pass exceptz try:pass withz!Unsupported control keyword: '%s'r ) r+r,stripSrCompileExceptiongroupstartsuperr)rr rmkeywordrr __class__s rrzPythonFragment.__init__^s- HH4djjlBDD I--BTI"  771: !''!*%D1   , ,&=D  --D ( ($t+f4DM  $&/DM  &=D--3g="  O]O>NOrr r!r"r#r __classcell__rFs@rr/r/Rs !P!Prr/c0eZdZdZddZddZedZy) FunctionDeclzfunction declarationc B||_tj|dfi|}tj|fi|}|j |t |dst jd|zfi||s3|jr&t jd|jdzfi|yy)Nr funcnamez'Code '%s' is not a function declarationz('**%s' keyword argument not allowed herer;) r rr ParseFuncrhasattrrr@kwargs kwargnames)rr allow_kwargsrrrs rrzFunctionDecl.__init__s ~~dF?.>?   t 8'7 8  tZ(--9D@"  --://"%&# !,|rc`g}|jddd}|jddd}|jddd}|jddd}|jr#|j d|j dz|D]}|r|j |d||r]|j d}||j |C|j |dtj|j{|j ||jr#|j d|j dz|D]`}|s|s|j ||j d}|j |dtj|jb|j|S)aEReturn the argument declarations of this FunctionDecl as a printable list. By default the return value is appropriate for writing in a ``def``; set `as_call` to true to build arguments to be passed to the function instead (assuming locals with the same names as the arguments exist). Nr;z**r=*) argnamesrQdefaults kwdefaultsrPappendpoprExpressionGeneratorvaluevarargsreverse) ras_call namedeclsrVrQrWrXnamedefaults rget_argument_expressionsz%FunctionDecl.get_argument_expressionss ==2&__TrT* ==2&__TrT*  ;;   TJNN1$55 6 'D  D$!78$..+?$$T*$$!=!=g!F!L!L!NP   &# '( <<   S8<<?2 3 Dh  &",,q/  X99'BHHJL   rcXt|jt|jzS)N)tuplerVrQ)rs r allargnameszFunctionDecl.allargnamessT]]#eDOO&<<v>rrGrIs@rriris8??rri) r#r+makorrrr&r/rKrir$rrrlsR5 **Z*-PZ-P`P=P=f?<?r