File: //home/arjun/projects/env/lib/python3.10/site-packages/mako/__pycache__/pygen.cpython-310.pyc
o
%we�( � @ s2 d Z ddlZddlmZ G dd� d�Zdd� ZdS )z<utilities for generating and formatting literal Python code.� N)�
exceptionsc @ sx e Zd Zdd� Zdd� Zdd� Zdd� Zd d
d �Zd d
� Zdd� Z dd� Z
dd� Z d dd�Z dd� Z
dd� Zdd � Zd S ) �
PythonPrinterc C s� d| _ g | _d| _|| _d| _g | _d| _| �� i | _t �
d�| _ t �
d�| _ t �
d�| _
t �
d�| _t �
d �| _t �
d
�| _d S ) Nr z � Fz^\s*#z^\s*$z:[ \t]*(?:#.*)?$z ^\s*(if|try|elif|while|for|with)z(^\s*(def|class|else|elif|except|finally)z"^\s*(else|elif|except|finally).*\:)�indent�
indent_detail� indentstring�stream�lineno� line_buffer�in_indent_lines�_reset_multi_line_flags�
source_map�re�compile�_re_space_comment� _re_space�
_re_indent� _re_compound�_re_indent_keyword�_re_unindentor)�selfr � r �C/home/arjun/projects/env/lib/python3.10/site-packages/mako/pygen.py�__init__ s" �zPythonPrinter.__init__c C s | j |7 _ d S �N)r �r �numr r r �_update_lineno7 � z PythonPrinter._update_linenoc C s | j | jvr|| j| j < d S d S r )r r
)r r r r r � start_source: s �zPythonPrinter.start_sourcec C s | j �d| � | �|� d S )N�
)r �writer r r r r � write_blanks> s zPythonPrinter.write_blanksNc C sP d| _ tt�d|��D ]\}}| j�|� |dur | �|| � | �d� q dS )z�print a line or lines of python which already contain indentation.
The indentation of the total block of lines will be adjusted to that of
the current indent level.F�\r?\nNr )r � enumerater �splitr
�appendr r )r �block�starting_lineno�i�lr r r �write_indented_blockB s �z"PythonPrinter.write_indented_blockc G s |D ]}| � |� qdS )z"print a series of lines of python.N)� writeline)r �lines�liner r r �
writelinesN s �zPythonPrinter.writelinesc C sV | j s
| �� d| _ |du s| j�|�s| j�|�r d}nd}|o*t|�o*|d dk}|sQ|r4| �|�rQ| jdkrQ| jd8 _t| j�dkrLt �
d��| j� � |du rWdS | j �
| �|�d � | �t|�d��� | j�|�r�| j�|�}|r�|�d�}| jd7 _| j�|� dS d}| j�|�}|r�| jd7 _| j�|� dS dS dS ) z�print a line of python, indenting it according to the current
indent level.
this also adjusts the indentation counter according to the
content of the line.
TNFr �#r z Too many whitespace closuresr )r �_flush_adjusted_linesr �matchr �len�_is_unindentorr r r �
MakoException�popr r! � _indent_liner r% r �searchr �groupr& r )r r. �hastext�
is_commentr2 �indentor�m2r r r r, S sN
�
����
�zPythonPrinter.writelinec C s | � � dS )z1close this printer, flushing any remaining lines.N)r1 �r r r r �close� s zPythonPrinter.closec C s< t | j�dkr dS | jd }|du rdS | j�|�}t|�S )zqreturn true if the given line is an 'unindentor',
relative to the last 'indent' event received.
r F���N)r3 r r r2 �bool)r r. r< r2 r r r r4 � s
z PythonPrinter._is_unindentor� c C s2 |dkr | j | j | S t�d| | j | j |�S )z�indent the given line according to the current indent level.
stripspace is a string of space that will be truncated from the
start of the line before indenting.rB �^%s)r r r �sub)r r. �
stripspacer r r r7 � s
�zPythonPrinter._indent_linec C s d\| _ | _dS )z^reset the flags which would indicate we are in a backslashed
or triple-quoted section.)FFN)� backslashed� triplequotedr> r r r r � s z%PythonPrinter._reset_multi_line_flagsc C sP | j p| j}tt�d|��| _ tt�d|��}|dks!|d dkr&| j | _|S )zcreturn true if the given line is part of a multi-line block,
via backslash or triple-quote.�\\$�
\"\"\"|\'\'\'r � r )rF rG rA r r8 r3 �findall)r r. �
current_state�triplesr r r �_in_multi_line� s
z PythonPrinter._in_multi_linec C s� d }| � � | jD ]3}| �|�r| j�|d � q |�� }|d u r0t�d|�r0t�d|�� d�}| j�| �
||�d � q g | _| � � d S )Nr �^[ \t]*[^# \t]� ^([ \t]*)r ) r r
rN r r! �
expandtabsr r8 r2 r9 r7 )r rE �entryr r r r1 � s
z#PythonPrinter._flush_adjusted_linesr �rB )�__name__�
__module__� __qualname__r r r r" r+ r/ r, r? r4 r7 r rN r1 r r r r r s (
G
(
r c s� ddg�d\� �� ��fdd�}ddd�}g }d}t �d | �D ]+}||�r+|�|� q |�� }|du rBt �d
|�rBt �d |��d �}|�|||�� q d
�|�S )z<remove the left-whitespace margin of a block of Python code.F)r r c s� �� p�� }t �d| �rd�� < nd�� < dd� }| rd�� r?|d�� | �\}} |r3d��< n/|d�� | �\}} n#|d| �\}} |rJ|S |d | �\}} |r[|�d
���< q|d | �\}} | s |S ) NrH TFc S s2 t �| |�}|r||t|�d��d � fS d |fS )Nr )r r2 r3 r9 )�reg�t�mr r r r2 s z7adjust_whitespace.<locals>.in_multi_line.<locals>.matchz%sz .*?(?=%s|$)r0 rI r z.*?(?=\"\"\"|\'\'\'|#|$))r r8 r9 )r. � start_stater2 rY �rF �staterG r r �
in_multi_line s*
�z(adjust_whitespace.<locals>.in_multi_linerB c S s t �d| d| �S )NrC rB )r rD )r. rE r r r r7 ' r z'adjust_whitespace.<locals>._indent_lineNr# rO rP r r rS )r r% r& rQ r8 r2 r9 �join)�textr] r7 r- rE r. r r[ r �adjust_whitespace� s
$
r` )�__doc__r �makor r r` r r r r �<module> s p