File: //usr/lib/python3/dist-packages/sqlparse/engine/__pycache__/grouping.cpython-310.pyc
o
�e}_�4 � @ s� d dl mZ d dl mZ d dlmZmZ ejejjejj fZ
ej ej j ej j
fZejejjfZdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Z dd� Z d d � Z d d � Z eej �d d!� �Z!d"d#� Z"d$d%� Z#d&d'� Z$eej%�d(d)� �Z&eej'�d*d+� �Z(e� d,d-� �Z)eej*�d.d/� �Z+d0d1� Z,e� d2d3� �Z-d4d5� Z.d6d7� Z/d8d9� d:d9� d;d<d<fd=d>�Z0d;S )?� )�sql)�tokens)�recurse�imtc C s� g }d}t t| ��D ]K\}}|| }|jrq
|jr$t||�s$t||� q
|j|j� r0|�|� q
|j|j � rUz|�
� }W n t yE Y q
w |}| � |||� ||| 7 }q
dS )z*Groups Tokens that have beginning and end.r N)
� enumerate�list�
is_whitespace�is_group�
isinstance�_group_matching�match�M_OPEN�append�M_CLOSE�pop�
IndexError� group_tokens) �tlist�cls�opens� tidx_offset�idx�token�tidx�open_idx� close_idx� r �:/usr/lib/python3/dist-packages/sqlparse/engine/grouping.pyr s,
� ��r c C � t | tj� d S �N)r r �SquareBrackets�r r r r �group_brackets4 � r" c C r r )r r � Parenthesisr! r r r �group_parenthesis8 r# r% c C r r )r r �Caser! r r r �
group_case< r# r' c C r r )r r �Ifr! r r r �group_if@ r# r) c C r r )r r �Forr! r r r � group_forD r# r+ c C r r )r r �Beginr! r r r � group_beginH r# r- c C �8 dd� }dd� }dd� }| }}t | tj||||� d S )Nc S � | � tjd�S )Nz::�r �T� Punctuation�r r r r r M � z group_typecasts.<locals>.matchc S � | d uS r r r3 r r r �validP � z group_typecasts.<locals>.validc S � ||fS r r �r �pidxr �nidxr r r �postS r7 z group_typecasts.<locals>.post��_groupr �
Identifier�r r r6 r< �
valid_prev�
valid_nextr r r �group_typecastsL s
rC c C s0 dd� }dd� }dd� }t | tj||||� d S )Nc S � | j tjjkS r )�ttyper1 �Keyword�TZCastr3 r r r r [ r4 z group_tzcasts.<locals>.matchc S r5 r r r3 r r r r6 ^ r7 z group_tzcasts.<locals>.validc S r8 r r r9 r r r r< a r7 zgroup_tzcasts.<locals>.postr= )r r r6 r< r r r �
group_tzcastsZ s rH c C sd dd� }dd� }dd� }dd� }d d
� }d d � }t | tj||||d
d� t | tj||||dd� d S )Nc S s t | tjjd�S )N��m)r r � TypedLiteralr
r3 r r r r m r# z"group_typed_literal.<locals>.matchc S � t | tj�S r )r
r rK r3 r r r �match_to_extendp � z,group_typed_literal.<locals>.match_to_extendc S r5 r r r3 r r r rA s r7 z'group_typed_literal.<locals>.valid_prevc S � | d uo
| j tjj� S r )r r rK r r3 r r r rB v � z'group_typed_literal.<locals>.valid_nextc S rO r )r r rK �M_EXTENDr3 r r r � valid_finaly rP z(group_typed_literal.<locals>.valid_finalc S s ||fS r r r9 r r r r< | r7 z!group_typed_literal.<locals>.postF��extendT)r> r rK )r r rM rA rB rR r< r r r �group_typed_literalg s �
�rU c C �8 dd� }dd� }dd� }dd� }t | tj||||� d S ) Nc S r/ )N�.r0 r3 r r r r � r4 zgroup_period.<locals>.matchc S s( t jt jf}tjtjjf}t| ||d�S �N��i�t)r r r? r1 �Name�String�Symbolr )r �sqlcls�ttypesr r r rA � s z group_period.<locals>.valid_prevc S � dS �NTr r3 r r r rB � s z group_period.<locals>.valid_nextc S sT t jt jf}tjtjjtjf}|d ur| | nd }t|||d�}|r&||fS ||fS rX ) r r �Functionr1 r\ r] r^ �Wildcardr )r r: r r; r_ r` �next_rB r r r r< � s
zgroup_period.<locals>.postr= �r r rA rB r< r r r � group_period� s
rg c C rV ) Nc S s | j o| jdkS )N�AS)�
is_keyword�
normalizedr3 r r r r � r# zgroup_as.<locals>.matchc S s | j dkp| j S )N�NULL)rj ri r3 r r r rA � s z group_as.<locals>.valid_prevc S s& t jt jt jf}t| |d� o| d uS �N�r[ )r1 �DML�DDL�CTEr )r r` r r r rB � s z group_as.<locals>.valid_nextc S r8 r r r9 r r r r< � r7 zgroup_as.<locals>.postr= rf r r r �group_as� s
rq c C r. )Nc S r/ )Nz:=)r r1 �
Assignmentr3 r r r r � r4 z group_assignment.<locals>.matchc S s | d uo | j tjvS r )rE r1 rF r3 r r r r6 � � z group_assignment.<locals>.validc S s, t jdf}| j||d�\}}|p|}||fS )N�;�rJ r )r1 r2 �
token_next_by)r r: r r; � m_semicolon�snidx�_r r r r< � s
z group_assignment.<locals>.post)r> r rr r@ r r r �group_assignment� s
rz c sf t jt jt jt jt jf� tt t �dd� }� �fdd�}dd� }| }}t | t j
||||dd� d S ) Nc S rD r )rE r1 �Operator�
Comparisonr3 r r r r � r4 z group_comparison.<locals>.matchc s. t | �� d�r dS | r| jr| jdkrdS dS )N)r[ rZ Trk F)r ri rj r3 �r_ r` r r r6 � s
z group_comparison.<locals>.validc S r8 r r r9 r r r r< � r7 z group_comparison.<locals>.postFrS ) r r$ rc r? � OperationrK � T_NUMERICAL�T_STRING�T_NAMEr> r| r@ r r} r �group_comparison� s �
�r� c C sP t jjt jf}| j|d�\}}|r&| �tj||� | j||d�\}}|sd S d S )Nrm �r[ r )r1 r] r^ r\ rv r r r? )r r` r r r r r �group_identifier� s �r� c
sb t jt jt jf� tjtjjf�dd� }� �fdd�}dd� }dd� }t| t j||||d d
d � d S ) Nc S rL r )r
r r r3 r r r r � rN zgroup_arrays.<locals>.matchc s t | � �d�S rX �r r3 r} r r rA � r4 z group_arrays.<locals>.valid_prevc S ra rb r r3 r r r rB � s z group_arrays.<locals>.valid_nextc S s ||fS r r r9 r r r r< � r7 zgroup_arrays.<locals>.postTF)rT r ) r r r? rc r1 r\ r] r^ r> rf r r} r � group_arrays� s
�r� c sj t t t �tjtjtjtjtjtj f� dd� }� �fdd�}dd� }| }}t
| tj||||dd� d S ) Nc S s t | tjtjfd�S rl )r r1 r{ rd r3 r r r r � rs z group_operator.<locals>.matchc s t | � �d�p| o| �tjd�S )NrY )� CURRENT_DATE� CURRENT_TIME�CURRENT_TIMESTAMP)r r r1 rF r3 r} r r r6 � s ��z group_operator.<locals>.validc S s t j| | _||fS r )r1 r{ rE r9 r r r r< s z group_operator.<locals>.postFrS ) r r� r� r r r$ rc r? r~ rK r> r@ r r} r �group_operator� s �
�r� c s� t jdf� tjtjtjtjtjtjf�t t
t t jt j t j
f �dd� }� ��fdd�}dd� }| }}t| tj||||dd � d S )
N)�null�rolec S r/ )N�,r0 r3 r r r r r4 z$group_identifier_list.<locals>.matchc s t | �� �d�S )N)rZ rJ r[ r� r3 ��m_roler_ r` r r r6 r# z$group_identifier_list.<locals>.validc S r8 r r r9 r r r r< r7 z#group_identifier_list.<locals>.postTrS )r1 rF r rc r& r? r| �IdentifierListr~ r r� r� �Commentrd r> r@ r r� r �group_identifier_list
s
�
�
�r� c C sv | j tjd�\}}|r9| jdd� |d�\}}|d ur+| j|dd�\}}| �tj||� | j tj|d�\}}|s d S d S )Nrm c S s t | tjd�p | jS rl )r r1 r� r )�tkr r r �<lambda>' s z group_comments.<locals>.<lambda>)r F)�skip_wsr� )rv r1 r� �token_not_matching�
token_prevr r �r r r �eidx�endr r r �group_comments" s
� �r� c C s� | j tjjd�\}}|rD| j tjj|d�\}}|d u r!| jd }n| j|d }| �|�}| �tj||� | j tjj|d�\}}|s d S d S )NrI ru ���� ) rv r �Wherer
r �_groupable_tokensr � token_indexr r� r r r � group_where/ s
�r� c C s� t jt jt jt jt jt jf}| j|tj d�\}}|r@| �
|�\}}t |t j�r1| j t j||dd� | j|tj |d�\}}|sd S d S )NrY TrS )rZ r[ r )
r r$ rc r& r? r~ r| rv r1 �Number�
token_nextr
r )r �I_ALIASr r r; re r r r �
group_aliased@ s � �r� c C s� d}d}| j D ]}|jdkrd}|jdkrd}q|r |r d S | jtjd�\}}|rL| �|�\}}t|tj�r>| � tj
||� | jtj|d�\}}|s)d S d S )NF�CREATET�TABLErm r� ) r �valuerv r1 r\ r� r
r r$ r rc )r �
has_create� has_table� tmp_tokenr r r; re r r r �group_functionsM s"
� �r� c C sn | j tjjd�\}}|r5| �|�\}}t|tjtjd�r&| � tj||� |}| j tjj|d�\}}|s dS dS )z,Group together Identifier and Asc/Desc tokenrm rY r� N)
rv r1 rF �Orderr� r r r? r� r �r r r r: �prev_r r r � group_ordera s �r� c C sh | j tjd�\}}|r2| �|�\}}t|tj�r$| jtj||dd� |}| j tj|d�\}}|s d S d S )N)rZ TrS )rZ r )rv r r� r� r
� TokenListr r� r r r �align_commentsl s �r� c C sh | j tjdfd�\}}|}d}|r"t|tj�r|}| �|�\}}|s|dkr2| jtj||dd� d S d S )N�VALUESrI r� TrS ) rv r1 rF r
r r$ r� r �Values)r r r � start_idx�end_idxr r r � group_valuesw s ��r� c C sF t ttttttttt t
t t t
ttttttttttfD ]}|| � q| S r )r� r" r% r' r) r+ r- r� r� rg r� r� r� rC rH rU r� r� rq r� rz r� r� r� )�stmt�funcr r r �group� s6 �
r� c C ra rb r rm r r r r� � � r� c C ra rb r rm r r r r� � r� NTc C s� d}d\} }
t t| ��D ]]\} } | | }
|
dk rq | jr q |r1| jr1t| |�s1t| ||||||� || �rd| �|
�\}}|
rd||
�rd||�rd|| | |
|�\}}| j||||d�}||| 7 }||} }
q |
| } }
q dS )zDGroups together tokens that are joined by a middle token. i.e. x < yr )NNrS N)r r r r r
r> r� r )r r r rA rB r< rT r r r: r� r r r r; re �from_idx�to_idx�grpr r r r> � s(
�r> )1�sqlparser r r1 �sqlparse.utilsr r r� �Integer�Floatr r] �Singler^ r� r\ � Placeholderr� r r" r% r' r) r+ r- rC rH rU rg rq rz r� r? r� r� r� r� r� r� r� r� r� rc r� r� r� r� r� r> r r r r �<module> sX #
$�