File: //home/arjun/projects/env/lib/python3.10/site-packages/mako/__pycache__/pyparser.cpython-310.pyc
o
%weu � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ h d�Ze� d�Z
e� e� dd
d �Z G d d
� d
ej
�ZG dd� dej
�ZG dd� dej
�ZG dd� d�ZdS )zyHandles parsing of Python code.
Parsing to AST is done via _ast on Python > 2.5, otherwise the compiler
module is used.
� N)� _ast_util)�compat)�
exceptions)�util> �None�True�False�print�arg�execc
K s\ zt �| d|�W S ty- } ztjdt�� jjt�� | dd� f fi |��|�d}~ww )z Parse an expression into ASTz <unknown>z (%s) %s (%r)r �2 N) r �parse� Exceptionr �SyntaxExceptionr � exception_as� __class__�__name__)�code�mode�exception_kwargs�e� r �F/home/arjun/projects/env/lib/python3.10/site-packages/mako/pyparser.pyr
s"
�������r
c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �FindIdentifiersc K s$ d| _ d| _t� | _|| _|| _d S �NF)� in_function�in_assign_targets�set�local_ident_stack�listenerr ��selfr r r r r �__init__2 s
zFindIdentifiers.__init__c C s( | j s | jj�|� d S | j�|� d S �N)r r �declared_identifiers�addr )r! �namer r r �
_add_declared9 s z FindIdentifiers._add_declaredc C s | � |j� d S r# )r' r& �r! �noder r r �visit_ClassDef? � z FindIdentifiers.visit_ClassDefc C s8 | � |j� | j}d| _|jD ]}| � |� q|| _d S �NT)�visit�valuer �targets)r! r) �in_a�nr r r � visit_AssignB s
z FindIdentifiers.visit_Assignc C sF |j d ur | �|j � |jd ur| �|j� |jD ]}| �|� qd S r# )r& r' �typer- �body�r! r) � statementr r r �visit_ExceptHandlerN s
�z#FindIdentifiers.visit_ExceptHandlerc G s | � |d� d S r, )�_visit_function)r! r) �argsr r r � visit_LambdaV r+ z FindIdentifiers.visit_Lambdac C s | � |j� | �|d� d S r )r' r& r8 r( r r r �visit_FunctionDefY � z!FindIdentifiers.visit_FunctionDefc c s0 � |D ]}t |tj�r|jE d H q|V qd S r# )�
isinstance�_ast�Tuple�elts)r! r9 r
r r r �_expand_tuples] s � �z FindIdentifiers._expand_tuplesc C sj | j }d| _ | j}|�dd� | �|jj�D ��| _|r"| �|j� n |jD ]}| �|� q%|| _ || _d S )NTc S � g | ]}t |��qS r ��arg_id��.0r
r r r �
<listcomp>p � z3FindIdentifiers._visit_function.<locals>.<listcomp>)r r �unionrA r9 r- r4 )r! r) �islambda�infr r1 r r r r8 d s �
z FindIdentifiers._visit_functionc C sH | � |j� | � |j� |jD ]}| � |� q|jD ]}| � |� qd S r# )r- �iter�targetr4 �orelser5 r r r � visit_Forz s
�zFindIdentifiers.visit_Forc C sb t |jtj�r| �|j� d S |jtvr+|j| jjvr-|j| j vr/| jj
� |j� d S d S d S d S r# ) r= �ctxr> �Storer' �id�reservedr r$ r �undeclared_identifiersr% r( r r r �
visit_Name� s
�zFindIdentifiers.visit_Namec C s> |j D ]}|jd ur| �|j� q| �|j�d�d � qd S )N�.r )�names�asnamer' r& �split�r! r) r& r r r � visit_Import� s
�z FindIdentifiers.visit_Importc C sR |j D ]#}|jd ur| �|j� q|jdkr tj di | j���| �|j� qd S )N�*��'import *' is not supported, since all identifier names must be explicitly declared. Please use the form 'from <modulename> import <name1>, <name2>, ...' instead.)r] )rW rX r' r& r �CompileExceptionr rZ r r r �visit_ImportFrom� s
���z FindIdentifiers.visit_ImportFromN)r �
__module__� __qualname__r" r' r* r2 r7 r: r; rA r8 rO rU r[ r_ r r r r r 1 s
r c @ � e Zd Zdd� Zdd� ZdS )� FindTuplec K s || _ || _|| _d S r# )r r � code_factory)r! r rd r r r r r" � s
zFindTuple.__init__c C sx |j D ]6}| j|fi | j��}| jj�|� | jj�t|��� � | jj }|�
|j �| j_ | jj }|�
|j �| j_ qd S r# ) r@ rd r r �codeargs�appendr9 �ExpressionGeneratorr. r$ rI rT )r! r) r1 �p�ldi�luir r r � visit_Tuple� s
�
��zFindTuple.visit_TupleN)r r` ra r" rk r r r r rc � s rc c @ rb )� ParseFuncc K s || _ || _d S r# )r r r r r r r" � s
zParseFunc.__init__c C s� |j | j_dd� |jjD �}|jjr|�|jjj� dd� |jjD �}|jjr/|�|jjj� || j_ |jj
| j_
|| j_ |jj | j_
|jj| j_|jj| j_d S )Nc S rB r rC rE r r r rG � rH z/ParseFunc.visit_FunctionDef.<locals>.<listcomp>c S rB r rC rE r r r rG � rH )r& r �funcnamer9 �varargrf r
�
kwonlyargs�kwarg�argnames�defaults�
kwargnames� kw_defaults�
kwdefaults�varargs�kwargs)r! r) rq rs r r r r; � s
zParseFunc.visit_FunctionDefN)r r` ra r" r; r r r r rl � � rl c @ rb )rg c C s t �d�| _| j�|� d S )Nz )r �SourceGenerator� generatorr- )r! �astnoder r r r" � r< z ExpressionGenerator.__init__c C s d� | jj�S )N� )�joinrz �result)r! r r r r. � s zExpressionGenerator.valueN)r r` ra r" r. r r r r rg � rx rg )r )�__doc__�operatorr> �makor r r r rS �
attrgetterrD � restore__astr
� NodeVisitorr rc rl rg r r r r �<module> s
v