a e g@srdZddlZddlmZddlmZGdddZGdddZGd d d eZGd d d ZGd ddeZ dS)zjutilities for analyzing expressions and blocks of Python code, as well as generating Python from AST nodesN) exceptions)pyparserc@seZdZdZddZdS) PythonCodez|d krd |d}n(|dkr|d}ntjd|fi|tj |fi|dS)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') rr stripSrCompileExceptiongroupstartsuperr)rrrmkeywordr __class__rrrUs4    zPythonFragment.__init__rrrrr __classcell__rrr6rr!Hs r!c@s0eZdZdZd ddZd ddZedd Zd S) FunctionDeclzfunction declarationTcKs||_tj|dfi|}tj|fi|}||t|dsVtjd|fi||s||jr|tjd|j dfi|dS)Nrfuncnamez'Code '%s' is not a function declarationz('**%s' keyword argument not allowed here) rrr Z ParseFuncrhasattrrr0kwargs kwargnames)rrZ allow_kwargsrrrrrrrus$   zFunctionDecl.__init__Fc Cs6g}|jddd}|jddd}|jddd}|jddd}|jr^|d|d|D]f}|r~|d||fqb|r|d}|dur||q|d|t| fqb||qb|j r|d|d|D]@}|s|s||q|d}|d|t| fq| |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**rz%s=%s*) argnamesr?defaults kwdefaultsr>appendpoprZExpressionGeneratorvaluevarargsreverse) rZas_callZ namedeclsrAr?rBrCnamedefaultrrrget_argument_expressionssB       z%FunctionDecl.get_argument_expressionscCst|jt|jS)N)tuplerAr?)rrrr allargnamesszFunctionDecl.allargnamesN)T)F)rrrrrrKpropertyrMrrrrr:qs   9r:cs eZdZdZfddZZS) FunctionArgsz.the argument portion of a function declarationc stjd|fi|dS)Nzdef ANON(%s):pass)r3r)rrr>r6rrrszFunctionArgs.__init__r8rrr6rrOsrO) rrZmakorrrrr!r:rOrrrrs  ")T