OwlCyberSecurity - MANAGER
Edit File: grouping.cpython-39.pyc
a ���e6 � @ 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 ]�\}}|| }|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 �Y/home/gouroczh/virtualenv/pat/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 | � tjd�S )N�.r( r+ r 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 sT t jt jf}tjtjjtjf}|d ur.| | nd }t|||d�}|rL||fS ||fS rG ) r r �Functionr) rK rL rM �Wildcardr )r r. r r/ rN rO �next_r6 r r r r0 � s zgroup_period.<locals>.postr1 r= r r r � group_period� s rT 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)rV rU r+ r r r r5 � s z group_as.<locals>.valid_prevc S s& t jt jt jf}t| |d� o$| d uS �N�rJ )r) ZDMLZDDLZCTEr )r rO 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 rZ 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 tjvS 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 r` 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) �Operator� Comparisonr+ r r r r � s z group_comparison.<locals>.matchc s2 t | �� d�rdS | r*| jr*| jdkr*dS dS d S )N)rJ rI TrW F)r rU rV r+ �rN rO 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 rQ r3 � Operationr< � T_NUMERICAL�T_STRING�T_NAMEr2 rb r4 r rc r �group_comparison� s � �rh c C sJ t jjt jf}| j|d�\}}|rF| �tj||� | j||d�\}}q d S )NrY �rJ r )r) rL rM rK r^ r r r3 )r rO r r r r r �group_identifier� s rj 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 rG �r r+ rc r r r5 � s z group_arrays.<locals>.valid_prevc S s dS rP 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 rQ r) rK rL rM r2 r= r rc r � group_arrays� s �rl c sj t t t �tjtjtjtjtjtj f� dd� }� �fdd�}dd� }| }}t | tj||||dd� d S ) Nc S s t | tjtjfd�S rX )r r) ra rR r+ r r r r s z group_operator.<locals>.matchc s t | � �d�p | o | �tjd�S )NrH )Z CURRENT_DATEZ CURRENT_TIMEZCURRENT_TIMESTAMP)r r r) r9 r+ rc r r r, s ��z group_operator.<locals>.validc S s t j| | _||fS r )r) ra r8 r- r r r r0 s z group_operator.<locals>.postFrC ) re rf rg r r r rQ r3 rd r< r2 r4 r rc r �group_operator� s � �rm 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)rI r@ rJ rk r+ �Zm_rolerN rO r r r, s z$group_identifier_list.<locals>.validc S s ||fS r r r- r r r r0 ! s z#group_identifier_list.<locals>.postTrC )r) r9 r rQ r! r3 rb ZIdentifierListrd re rf rg �CommentrR r2 r4 r rp r �group_identifier_list s � � �rr c C sp | j tjd�\}}|rl| jdd� |d�\}}|d urV| j|dd�\}}| �tj||� | j tj|d�\}}qd S )NrY c S s t | tjd�p| jS rX )r r) rq r )Ztkr r r �<lambda>. � z group_comments.<locals>.<lambda>)r F)Zskip_wsri )r^ r) rq Ztoken_not_matching� token_prevr r �r r r Zeidx�endr r r �group_comments) s � rx c C s� | j tjjd�\}}|r�| j tjj|d�\}}|d u rB| jd }n| j|d }| �|�}| �tj||� | j tjj|d�\}}qd S )Nr? r] ���� ) r^ r �Wherer r Z_groupable_tokensr Z token_indexr rv r r r � group_where6 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 )NrH TrC )rI rJ r ) r r rQ r! r3 rd rb r^ r) �Number� token_nextr r )r ZI_ALIASr r r/ rS r r r � group_aliasedG 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�| � tj ||� | jtj|d�\}}qld S )NFZCREATETZTABLEr; rY ri ) r �value�upperr^ r) rK r~ r r r r rQ ) r Z has_createZ has_tableZhas_asZ tmp_tokenr r r/ rS r r r �group_functionsT s$ r� c C sh | j tjjd�\}}|rd| �|�\}}t|tjtjd�rL| � tj||� |}| j tjj|d�\}}qdS )z,Group together Identifier and Asc/Desc tokenrY rH ri N) r^ r) r9 ZOrderru r r r3 r} r �r r r r. �prev_r r r � group_orderk s r� c C sb | j tjd�\}}|r^| �|�\}}t|tj�rH| jtj||dd� |}| j tj|d�\}}qd S )N)rI TrC )rI r )r^ r rq ru r Z TokenListr r� r r r �align_commentsv s r� c C sb | j tjdfd�\}}|}d}|rBt|tj�r2|}| �|�\}}q |dkr^| jtj||dd� d S )NZVALUESr? ry TrC ) r^ r) r9 r r r r~ r �Values)r r r Z start_idxZend_idxr r r � group_values� s r� c C sF t ttttttttt t t t t ttttttttttfD ] }|| � q4| S r )rx r r r" r$ r& r' r� r| rT rl rj r� r7 r> rE rm rh rZ r r` r� rr r� )�stmt�funcr r r �group� s6 � r� c C s dS rP r rY r r r rs � rt rs c C s dS rP r rY r r r rs � rt 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 r~ r )r r r r5 r6 r0 rD r r r. r� r r r r/ rS Zfrom_idxZto_idx�grpr r r r2 � s& r2 )1Zsqlparser r r) Zsqlparse.utilsr r r} �Integer�Floatre rL �SinglerM rf rK Z Placeholderrg r r r r" r$ r& r' r7 r> rE rT rZ r` rh r3 rj rl rm rr rq rx r{ r| r rQ r� r� r� r� r� r2 r r r r �<module> sV # $�