OwlCyberSecurity - MANAGER
Edit File: grouping.cpython-39.pyc
a gH�f 9 � @ 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!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,�d0d1� �Z-eej �d2d3� �Z.e� d4d5� �Z/d6d7� Z0d8d9� Z1d:d;� d<d;� d=d>d>fd?d@�Z2d=S )A� )�sql)�tokens)�recurse�imtc C s� g }d}t t| ��D ]�\}}|| }|jr,q|jrHt||�sHt||� q|j|j� r`|�|� q|j|j � rz |� � }W n t y� Y qY n0 |}| � |||� ||| 7 }qdS )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�clsZopens� tidx_offset�idx�token�tidxZopen_idxZ close_idx� r �Z/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/sqlparse/engine/grouping.pyr s& r c C s t | tj� d S �N)r r �SquareBrackets�r r r r �group_brackets4 s r c C s t | tj� d S r )r r � Parenthesisr r r r �group_parenthesis8 s r c C s t | tj� d S r )r r �Caser r r r � group_case< s r" c C s t | tj� d S r )r r �Ifr r r r �group_if@ s r$ c C s t | tj� d S r )r r �Forr r r r � group_forD s r& c C s t | tj� d S r )r r ZBeginr r r r � group_beginH s r' c C s8 dd� }dd� }dd� }| }}t | tj||||� d S )Nc S s | � tjd�S )Nz::�r �T� Punctuation�r r r r r M s z group_typecasts.<locals>.matchc S s | d uS r r r+ r r r �validP s z group_typecasts.<locals>.validc S s ||fS r r �r �pidxr �nidxr r r �postS s z group_typecasts.<locals>.post��_groupr � Identifier�r r r, r0 � valid_prev� valid_nextr r r �group_typecastsL s r7 c C s8 dd� }dd� }dd� }dd� }t | tj||||� d S ) Nc S s | j tjjkS r )�ttyper) �KeywordZTZCastr+ r r r r [ s z group_tzcasts.<locals>.matchc S s | d uS r r r+ r r r r5 ^ s z!group_tzcasts.<locals>.valid_prevc S s* | d uo(| j p(| �tjd�p(| jtjj� S �N�AS)r r r) r9 r � TypedLiteralr r+ r r r r6 a s � �z!group_tzcasts.<locals>.valid_nextc S s ||fS r r r- r r r r0 h s zgroup_tzcasts.<locals>.postr1 �r r r5 r6 r0 r r r � group_tzcastsZ s r> 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 r< r r+ r r r r t s z"group_typed_literal.<locals>.matchc S s t | tj�S r )r r r< r+ r r r �match_to_extendw s z,group_typed_literal.<locals>.match_to_extendc S s | d uS r r r+ r r r r5 z s z'group_typed_literal.<locals>.valid_prevc S s | d uo| j tjj� S r )r r r< r r+ r r r r6 } s z'group_typed_literal.<locals>.valid_nextc S s | d uo| j tjj� S r )r r r< ZM_EXTENDr+ r r r � valid_final� s z(group_typed_literal.<locals>.valid_finalc S s ||fS r r r- r r r r0 � s z!group_typed_literal.<locals>.postF��extendT)r2 r r< )r r rA r5 r6 rB r0 r r r �group_typed_literaln s ��rE c C s8 dd� }dd� }dd� }dd� }t | tj||||� d S ) Nc S s< t jdft jdft jdffD ]\}}| �||�r dS q dS )N�.z->z->>TF)r) r* �Operatorr )r r8 �valuer r r r � s � zgroup_period.<locals>.matchc S s( t jt jf}tjtjjf}t| ||d�S �N��i�t)r r r3 r) �Name�String�Symbolr )r �sqlcls�ttypesr r r r5 � s z group_period.<locals>.valid_prevc S s dS �NTr r+ r r r r6 � s z group_period.<locals>.valid_nextc S sZ t jt jf}tjtjjtjtjjf}|d ur4| | nd }t |||d�}|rR||fS ||fS rI ) r r �Functionr) rM rN rO �Wildcard�Singler )r r. r r/ rP rQ �next_r6 r r r r0 � s zgroup_period.<locals>.postr1 r= r r r � group_period� s rW c C s8 dd� }dd� }dd� }dd� }t | tj||||� d S ) Nc S s | j o| jdkS r: )� is_keyword� normalizedr+ r r r r � s zgroup_as.<locals>.matchc S s | j dkp| j S )N�NULL)rY rX r+ r r r r5 � s z group_as.<locals>.valid_prevc S s& t jt jt jf}t| |d� o$| d uS �N�rL )r) ZDMLZDDLZCTEr )r rQ r r r r6 � s z group_as.<locals>.valid_nextc S s ||fS r r r- r r r r0 � s zgroup_as.<locals>.postr1 r= r r r �group_as� s r] c C s8 dd� }dd� }dd� }| }}t | tj||||� d S )Nc S s | � tjd�S )Nz:=)r r) � Assignmentr+ r r r r � s z group_assignment.<locals>.matchc S s | d uo| j tjfvS r )r8 r) r9 r+ r r r r, � s z group_assignment.<locals>.validc S s, t jdf}| j||d�\}}|p"|}||fS )N�;�r@ r )r) r* � token_next_by)r r. r r/ Z m_semicolonZsnidx�_r r r r0 � s z group_assignment.<locals>.post)r2 r r^ r4 r r r �group_assignment� s rc c sf t jt jt jt jt jf� tt t �dd� }� �fdd�}dd� }| }}t | t j ||||dd� d S ) Nc S s | j tjjkS r )r8 r) rG � Comparisonr+ r r r r � s z group_comparison.<locals>.matchc s2 t | �� d�rdS | r*| jr*| jdkr*dS dS d S )N)rL rK TrZ F)r rX rY r+ �rP rQ r r r, � s z group_comparison.<locals>.validc S s ||fS r r r- r r r r0 � s z group_comparison.<locals>.postFrC ) r r rS r3 � Operationr< � T_NUMERICAL�T_STRING�T_NAMEr2 rd r4 r re r �group_comparison� s � �rj c C sJ t jjt jf}| j|d�\}}|rF| �tj||� | j||d�\}}q d S )Nr\ �rL r )r) rN rO rM ra r r r3 )r rQ r r r r r �group_identifier� s rl c C sd | j tjjd�\}}|r`| �|�\}}t|tjtjd�rH| � tj||� | j tjj|d�\}}qd S )Nr? rJ r` ) ra r �Overr � token_nextr r r) rM r )r r r r/ rV r r r � group_over� s ro c sb t jt jt jf� tjtjjf�dd� }� �fdd�}dd� }dd� }t| t j||||d d d � d S ) Nc S s t | tj�S r )r r r r+ r r r r � s zgroup_arrays.<locals>.matchc s t | � �d�S rI �r r+ re r r r5 � s z group_arrays.<locals>.valid_prevc S s dS rR r r+ r r r r6 s z group_arrays.<locals>.valid_nextc S s ||fS r r r- r r r r0 s zgroup_arrays.<locals>.postTF)rD r ) r r r3 rS r) rM rN rO r2 r= r re r � group_arrays� s �rq c sj t t t �tjtjtjtjtjtj f� dd� }� �fdd�}dd� }| }}t | tj||||dd� d S ) Nc S s t | tjtjfd�S r[ )r r) rG rT r+ r r r r s z group_operator.<locals>.matchc s t | � �d�p | o | �tjd�S )NrJ )Z CURRENT_DATEZ CURRENT_TIMEZCURRENT_TIMESTAMP)r r r) r9 r+ re r r r, s ��z group_operator.<locals>.validc S s t j| | _||fS r )r) rG r8 r- r r r r0 s z group_operator.<locals>.postFrC ) rg rh ri r r r rS r3 rf r< r2 r4 r re r �group_operator s � �rr c s� t jdf� tjtjtjtjtjtjf�t t t t jt j t j f �dd� }� ��fdd�}dd� }| }}t| tj||||dd � d S ) N)�nullZrolec S s | � tjd�S )N�,r( r+ r r r r * s z$group_identifier_list.<locals>.matchc s t | �� �d�S )N)rK r@ rL rp r+ �Zm_rolerP rQ r r r, - s z$group_identifier_list.<locals>.validc S s ||fS r r r- r r r r0 0 s z#group_identifier_list.<locals>.postTrC )r) r9 r rS r! r3 rd ZIdentifierListrf rg rh ri �CommentrT r2 r4 r ru r �group_identifier_list# s � � �rw c C sp | j tjd�\}}|rl| jdd� |d�\}}|d urV| j|dd�\}}| �tj||� | j tj|d�\}}qd S )Nr\ c S s t | tjd�p| jS r[ )r r) rv Z is_newline)Ztkr r r �<lambda>= � z group_comments.<locals>.<lambda>)r F)Zskip_wsrk )ra r) rv Ztoken_not_matching� token_prevr r �r r r �eidx�endr r r �group_comments8 s � r~ c C s� | j tjjd�\}}|r�| j tjj|d�\}}|d u rB| jd }n| j|d }| �|�}| �tj||� | j tjj|d�\}}qd S )Nr? r` ���� ) ra r �Wherer r Z_groupable_tokensr Z token_indexr r{ r r r � group_whereE s r� c C s~ t jt jt jt jt jt jf}| j|tj d�\}}|rz| � |�\}}t |t j�rb| j t j||dd� | j|tj |d�\}}q0d S )NrJ TrC )rK rL r ) r r rS r! r3 rf rd ra r) �Numberrn r r )r ZI_ALIASr r r/ rV r r r � group_aliasedV s � r� c C s� d}d}d}| j D ]6}|j�� dkr(d}|j�� dkr:d}|jdkrd}q|rZ|rZ|sZd S | jtjd�\}}|r�| �|�\}}t|tj �r�| �|�\} } | r�t| tj �r�| } n|} | � tj || � | jtj|d�\}}qld S )NFZCREATETZTABLEr; r\ rk ) r rH �upperra r) rM rn r r r rm r rS ) r Z has_createZ has_tableZhas_asZ tmp_tokenr r r/ rV Zover_idxZoverr| r r r �group_functionsc s, r� c C sh | j tjjd�\}}|rd| �|�\}}t|tjtjd�rL| � tj||� |}| j tjj|d�\}}qdS )z,Group together Identifier and Asc/Desc tokenr\ rJ rk N) ra r) r9 ZOrderrz r r r3 r� r �r r r r. �prev_r r r � group_order s r� c C sb | j tjd�\}}|r^| �|�\}}t|tj�rH| jtj||dd� |}| j tj|d�\}}qd S )N)rK TrC )rK r )ra r rv rz r Z TokenListr r� r r r �align_comments� s r� c C sb | j tjdfd�\}}|}d}|rBt|tj�r2|}| �|�\}}q |dkr^| jtj||dd� d S )NZVALUESr? r TrC ) ra r) r9 r r r rn r �Values)r r r Z start_idxZend_idxr r r � group_values� s r� c C sH t ttttttttt t t t t tttttttttttfD ] }|| � q6| S r )r~ r r r" r$ r& r' ro r� r� rW rq rl r� r7 r> rE rr rj r] r� rc r� rw r� )�stmt�funcr r r �group� s8 � r� c C s dS rR r r\ r r r rx � ry rx c C s dS rR r r\ r r r rx � ry NTc C s� d}d\} } t t| ��D ]�\} } | | } | dk r2q| jr:q|rb| jrbt| |�sbt| ||||||� || �r�| �| �\}}| r�|| �r�||�r�|| | | |�\}}| j||||d�}||| 7 }|| } } q| | } } qdS )zDGroups together tokens that are joined by a middle token. i.e. x < yr )NNrC N)r r r r r r2 rn r )r r r r5 r6 r0 rD r r r. r� r r r r/ rV Zfrom_idxZto_idx�grpr r r r2 � s& r2 )3Zsqlparser r r) Zsqlparse.utilsr r r� �Integer�Floatrg rN rU rO rh rM Z Placeholderri r r r r" r$ r& r' r7 r> rE rW r] rc rj r3 rl rm ro rq rr rw rv r~ r� r� r� rS r� r� r� r� r� r2 r r r r �<module> s\ # %�