File: //home/arjun/projects/env/lib64/python3.10/site-packages/mako/__pycache__/ast.cpython-310.pyc
o
%we� � @ sr d Z ddlZddlmZ ddlmZ G dd� d�ZG dd� d�ZG d d
� d
e�ZG d d � d �ZG d
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 ZdZdd� ZdS )�
PythonCodez<represents information about a string containing Python codec K s^ || _ t� | _t� | _t|t�r tj|�� dfi |��}n|}tj | fi |��}|�
|� d S )N�exec) �code�set�declared_identifiers�undeclared_identifiers�
isinstance�strr �parse�lstrip�FindIdentifiers�visit��selfr �exception_kwargs�expr�f� r �A/home/arjun/projects/env/lib/python3.10/site-packages/mako/ast.py�__init__ s
zPythonCode.__init__N��__name__�
__module__� __qualname__�__doc__r r r r r r � r c @ r )� ArgumentListzBparses a fragment of code as a comma-separated list of expressionsc K s� g | _ g | _t� | _t� | _t|t�r.t�d|�r#t�d|�s#|d7 }t j
|dfi |��}n|}t j | t fi |��}|�
|� d S )Nz\Sz,\s*$�,r )�codeargs�argsr r r
r r �re�matchr r
� FindTupler r r r r r r 6 s
zArgumentList.__init__Nr r r r r r 2 r r c � e Zd ZdZ� fdd�Z� ZS )�PythonFragmentz�extends PythonCode to provide identifier lookups in partial control
statements
e.g.::
for x in 5:
elif y==9:
except (MyException, e):
c s� t �d|�� t j�}|stjd| fi |���|�d�r%|d |�d�� }|�dd�\}}|dv r6|d }n3|dkr?|d }n*|d
v rJd | d }n |d krUd
| d }n|dkr^|d }n tjd| fi |���t� j |fi |�� d S )Nz ^(\w+)(?:\s+(.*?))?:\s*(#|$)z0Fragment '%s' is not a partial control statement� � � )�for�if�while�pass�tryzpass
except:pass)�elif�elsezif False:pass
�exceptz try:pass
�withz!Unsupported control keyword: '%s')
r# r$ �strip�Sr �CompileException�group�start�superr )r r r �m�keywordr �� __class__r r r U s4 ��
��zPythonFragment.__init__�r r r r r �
__classcell__r r r< r r' H s r' c @ s0 e Zd ZdZd dd�Zd dd�Zedd � �Zd
S )
� FunctionDeclzfunction declarationTc K s� || _ tj|dfi |��}tj| fi |��}|�|� t| d�s+tjd| fi |���|s>| jr@tjd| j d 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 r6 �kwargs�
kwargnames)r r � allow_kwargsr r r r r r r u s&
��
����zFunctionDecl.__init__Fc C s4 g }| j ddd� }| jddd� }| jddd� }| jddd� }| jr/|�d|�d� � |D ]3}|r?|�d||f � q1|r_|�d�}|du rP|�|� q1|�d|t�|�� � f � q1|�|� q1| j
rr|�d|�d� � |D ] }|sz|s�|�|� qt|�d�}|�d|t�|�� � f � qt|� � |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).
NrB z**r z%s=%s�*) �argnamesrF �defaults�
kwdefaultsrE �append�popr �ExpressionGenerator�value�varargs�reverse) r �as_call� namedeclsrI rF rJ rK �name�defaultr r r �get_argument_expressions� sB
��
��z%FunctionDecl.get_argument_expressionsc C s t | j�t | j� S )N)�tuplerI rF )r r r r � allargnames� s zFunctionDecl.allargnamesN)T)F)r r r r r rV �propertyrX r r r r r@ q s
9r@ c r& )� FunctionArgsz.the argument portion of a function declarationc s t � jd| fi |�� d S )Nzdef ANON(%s):pass)r9 r )r r rE r< r r r � s zFunctionArgs.__init__r> r r r< r rZ � s rZ )
r r# �makor r r r r' r@ rZ r r r r �<module> s ")T