y`? d Z ddlZddlmZ ddlmZ G d d Z G d d Z G d d
e Z G d d Z G d
de Z dS )zjutilities for analyzing expressions and blocks of Python
code, as well as generating Python from AST nodes N)
exceptions)pyparserc e Zd ZdZd ZdS )
PythonCodez$++--LL;KLLDDD $T > >-= > >
N__name__
__module__ __qualname____doc__r r r r r s) FF r r c e Zd ZdZd ZdS ) ArgumentListzBparses a fragment of code as a comma-separated list of expressionsc | g | _ g | _ t | _ t | _ t |t rCt j d| rt j d| s|dz
}t j
|dfi |}n|}t j | t fi |}|
| d S )Nz\Sz,\s*$,r )codeargsargsr
r r r
r rematchr r FindTupler r r s r r zArgumentList.__init__6 s
$'EE!&)ee# dC xt$$
RXh-E-E
>$CC2BCCDDD tZ D D3C D D
r Nr r r r r" r" 2 s) LL r r" c " e Zd ZdZ fdZ xZS )PythonFragmentzextends PythonCode to provide identifier lookups in partial control
statements
e.g.::
for x in 5:
elif y==9:
except (MyException, e):
c
t j d| t j }|st j d|z fi || d r |d | d }| dd \ }}|dv r|dz }nI|dk r|d z }n=|d
v r d |z dz }n0|d k r d
|z dz }n!|dk r|dz }nt j d|z fi | t j |fi | d S )Nz ^(\w+)(?:\s+(.*?))?:\s*(#|$)z0Fragment '%s' is not a partial control statement )forifwhilepasstryzpass
except:pass)elifelsezif False:pass
exceptz try:pass
withz!Unsupported control keyword: '%s')
r' r( stripSr CompileExceptiongroupstartsuperr )r r r mkeywordr __class__s r r zPythonFragment.__init__U sd H4djjllBD I I -BTI "
771:: & !''!** %D''!Q--$ , , ,&=DD
--DD
(
(
($t+f4DD
$&/DD
&=DD -3g= "
22!122222r r r r r r
__classcell__rA s @r r+ r+ H sB 3 3 3 3 3 3 3 3 3r r+ c 8 e Zd ZdZddZd dZed ZdS )
FunctionDeclzfunction declarationTc ( || _ t j |dfi |}t j | fi |}| | t | d st
j d|z fi ||s'| j r"t
j d| j d z fi |d S d S )Nr funcnamez'Code '%s' is not a function declarationz('**%s' keyword argument not allowed here)
r r r ParseFuncr hasattrr r; kwargs
kwargnames)r r allow_kwargsr r r s r r zFunctionDecl.__init__u s ~dF??.>?? t 8 8'7 8 8
tZ(( -9D@ "
-:/"%& #
r Fc g }| j ddd }| j ddd }| j ddd }| j ddd }| j r+| d| d z |D ]}|r| |d| |rl| d }|| | N| |dt j | | | | j
r+| d| d z |D ]o}|s|s| | | d }| |dt j | p| |S )aE Return 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).
NrI z**r =*) argnamesrM defaults
kwdefaultsrL appendpopr ExpressionGeneratorvaluevarargsreverse) r as_call namedeclsrR rM rS rT namedefaults r get_argument_expressionsz%FunctionDecl.get_argument_expressions s( =2&_TTrT*
=2&_TTrT*
; 7 TJNN1$5$5 5 6 6 6 ' 'D
' DDD$$!78888
'$..++? $$T**** $$ 44!=g!F!F!L!L!N!N!NP
&&&& < 4 S8<<?? 2 3 3 3 D
h
&&&&",,q// ttX 9' B B H H J J JL
r c T t | j t | j z S )N)tuplerR rM )r s r allargnameszFunctionDecl.allargnames s! T]##eDO&<&<< r )r r rL rA s r r zFunctionArgs.__init__ s. ,t3>>v>>>>>r rB rD s @r re re s> 88? ? ? ? ? ? ? ? ?r re )
r r' makor r r r" r+ rF re r r r rh s 5 5
D ,&3 &3 &3 &3 &3Z &3 &3 &3RQ= Q= Q= Q= Q= Q= Q= Q=h? ? ? ? ?< ? ? ? ? ?r