|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/sqlparse/__pycache__/ |
Upload File : |
o
�{;f�P � @ s� d Z ddlZddlmZ ddlmZ ddlmZm Z G dd� d�Z
G dd � d �Z G d
d � d e �Z G d d
� d
e �Z
G dd� de
e �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG d d � d e �ZG d d � d e �ZG d d!� d!e �ZG d"d#� d#e �ZG d$d%� d%e �ZG d&d'� d'e �ZG d(d)� d)e
e �ZG d*d+� d+e �Z G d,d-� d-e �Z G d.d/� d/e �Z G d0d1� d1e �Z dS )2zFThis module contains classes representing syntactical elements of SQL.� N��tokens)�
SQLParseError)�imt�
remove_quotesc @ s e Zd ZdZdd� Zdd� ZdS )�NameAliasMixinz'Implements get_real_name and get_alias.c C s$ | j tjdfd�\}}| j|dd�S )�7Returns the real name (object name) of this identifier.�.��mT)� real_name)�
token_next_by�T� Punctuation�_get_first_name)�self�dot_idx�_� r �./usr/lib/python3/dist-packages/sqlparse/sql.py�
get_real_name s z NameAliasMixin.get_real_namec C sl | j tjdfd�\}}|dur| j|d dd�S | j tjd�\}}t| j�dkr2|dur4| jdd �S dS dS )
�2Returns the alias for this identifier or ``None``.�ASr
N� T)�keywords��t� ��reverse)r
r �Keywordr �
Whitespace�lenr )r �kw_idx�kwr �wsr r r � get_alias s �zNameAliasMixin.get_aliasN)�__name__�
__module__� __qualname__�__doc__r r& r r r r r s r c @ sf e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
ddd�Z dd� Z dd� Z
dd� ZdS )�Tokenz�Base class for all other classes in this module.
It represents a single token and has two instance attributes:
``value`` is the unchanged value of the token and ``ttype`` is
the type of the token.
)�value�ttype�parent�
normalized�
is_keyword�is_group�
is_whitespacec C sX t |�}|| _|| _d | _d| _|tjv | _| jtjv | _ | jr'|�
� | _ d S || _ d S )NF) �strr, r- r. r1 r r r0 r! r2 �upperr/ )r r- r, r r r �__init__3 s zToken.__init__c C � | j S �N�r, �r r r r �__str__= s z
Token.__str__c C sF | � � }| �� }|�d�r|�d�rdnd}djddt| �it� ��S )N�'�"z"<{cls} {q}{value}{q} at 0x{id:2X}>�idr )�_get_repr_name�_get_repr_value�
startswith�endswith�formatr= �locals)r �clsr, �qr r r �__repr__D s ��zToken.__repr__c C s t | j��d�d S )Nr ���)r3 r- �splitr9 r r r r> L s zToken._get_repr_namec C s2 t | �}t|�dkr|d d� d }t�dd|�S )N� � z...z\s+� )r3 r" �re�sub)r �rawr r r r? O s zToken._get_repr_valuec c s � | V dS )zResolve subgroups.Nr r9 r r r �flattenU s �
z
Token.flattenFc s� | j |u }|r |du r
|S t|t�r|f}|r8| jr tjnd� � fdd�|D �}|D ] }|�| j�r5 dS q*dS | jrBdd� |D �}| j|v S )a, Checks whether the token matches the given arguments.
*ttype* is a token type. If this token doesn't match the given token
type.
*values* is a list of possible values for this token. The values
are OR'ed together so if only one of the values matches ``True``
is returned. Except for keyword tokens the comparison is
case-sensitive. For convenience it's OK to pass in a single string.
If *regex* is ``True`` (default is ``False``) the given values are
treated as regular expressions.
Nr c 3 s � | ] }t �|� �V qd S r7 )rL �compile��.0�v��flagr r � <genexpr>o s � z Token.match.<locals>.<genexpr>TFc s s � | ]}|� � V qd S r7 )r4 rQ r r r rV w s � )r- �
isinstancer3 r0 rL �
IGNORECASE�searchr/ )r r- �values�regex� type_matched�patternr rT r �matchY s
�
z Token.matchc C s& | j }|rt||�r dS |j }|sdS )z�Returns ``True`` if this token is within *group_cls*.
Use this method for example to check if an identifier is within
a function: ``t.within(sql.Function)``.
TF)r. rW )r � group_clsr. r r r �within{ s
�z Token.withinc C s
| j |kS )z<Returns ``True`` if this token is a direct child of *other*.�r. )r �otherr r r � is_child_of� �
zToken.is_child_ofc C s$ | j }|r||kr dS |j }|sdS )z7Returns ``True`` if *other* is in this tokens ancestry.TFra )r rb r. r r r � has_ancestor� s �zToken.has_ancestorN�F)r' r( r) r* � __slots__r5 r: rF r> r? rO r^ r` rc re r r r r r+ ( s
"
r+ c s e Zd ZdZdZd?� fdd� Zdd� Zdd � Zd
d � Zd d
� Z d@dd�Z
dd� Z dd� Z dd� Z
edd� �ZdAdd �ZdBd d �ZdCd!d"�Zd#d$� Zd%d&� ZdBd'd(�ZdDd)d*�ZdEd+d,�Z dBd-d.�Zd/d0� ZdFd1d2�Zd3d4� Zd5d6� Z d7d8� Z d9d:� Z d;d<� Z dGd=d>�Z � Z!S )H� TokenListzwA group of tokens.
It has an additional instance attribute ``tokens`` which holds a
list of child-tokens.
r Nc s: |pg � _ � fdd�� j D � t� �d t� �� d� _d S )Nc s g | ]}t |d � ��qS ra )�setattr�rR �tokenr9 r r �
<listcomp>� � z&TokenList.__init__.<locals>.<listcomp>T)r �superr5 r3 r1 )r r �� __class__r9 r r5 � s
zTokenList.__init__c C s d� dd� | �� D ��S )N� c s s � | ]}|j V qd S r7 r8 rj r r r rV � s � z$TokenList.__str__.<locals>.<genexpr>)�joinrO r9 r r r r: � s zTokenList.__str__c C s
t | j�S r7 )�iterr r9 r r r �__iter__� �
zTokenList.__iter__c C s
| j | S r7 r )r �itemr r r � __getitem__� ru zTokenList.__getitem__c C s
t | �jS r7 )�typer' r9 r r r r> � ru zTokenList._get_repr_namer rq c C s� t | j�}t| j�D ]P\}}|�� }|�� } ||d k}
|
r dnd} | �d�r.| �d�r.dnd} tdjd i t � ��|d� |j
rZ|du sH||k rZ|
rLd nd
}
|� ||d |||
� q
dS ) z Pretty-print the object tree.r z`- z|- r; r< z${_pre}{pre}{idx} {cls} {q}{value}{q})�fileNz z| r ) r" r � enumerater> r? r@ rA �printrB rC r1 � _pprint_tree)r � max_depth�depth�f�_pre� token_count�idxrk rD r, �last�prerE �
parent_prer r r r| � s&
��� ��zTokenList._pprint_treec C sD d}| � � D ]}|t|j� }|| kr |k r | S |}qdS )z-Returns the token that is on position offset.r N)rO r" r, )r �offsetr� rk �endr r r �get_token_at_offset� s ��z TokenList.get_token_at_offsetc
c sT � z| j D ]}|jr|�� E dH q|V qW dS ty) } ztd�|�d}~ww )znGenerator yielding ungrouped tokens.
This method is recursively called for all child tokens.
Nz Maximum recursion depth exceeded)r r1 rO �RecursionErrorr )r rk �errr r r rO � s �
�
��zTokenList.flattenc c s � | j D ]}|jr |V qd S r7 )r r1 �r rk r r r � get_sublists� s �
��zTokenList.get_sublistsc C r6 r7 r r9 r r r �_groupable_tokens� s zTokenList._groupable_tokensFc C s� |du rdS t |ttf�s|f}|r;|du sJ �t|d dd�D ]}| j| }|D ]}||�r7||f S q)q dS t| j||� |d�D ]\}}|D ]}||�rZ||f S qLqFdS )z next token that match functionsNr rG )�start�NN)rW �list�tuple�ranger rz )r �funcsr� r� r r� rk �funcr r r �_token_matching� s*
�� ���zTokenList._token_matchingTc s � �fdd�}| � |�d S )z�Returns the first child token.
If *skip_ws* is ``True`` (the default), whitespace
tokens are ignored.
if *skip_cm* is ``True`` (default: ``False``), comments are
ignored too.
c � �r| j p� ot| tjtd� S �N)r �i�r2 r r �Comment��tk��skip_cm�skip_wsr r �matcher �
�z&TokenList.token_first.<locals>.matcherr �r� )r r� r� r� r r� r � token_first� s
zTokenList.token_firstrG c s" |d7 }| � � ��fdd�||�S )Nr c s t | � ���S r7 )r r� �r� r r r r �<lambda> s z)TokenList.token_next_by.<locals>.<lambda>r� )r r� r r r� r� r r� r r
s zTokenList.token_next_byc C s2 t |ttf�s
|fn|}dd� |D �}| �||�S )Nc s g | ]� � fd d��qS )c s
� | � S r7 r r� �r� r r r� s
z9TokenList.token_not_matching.<locals>.<listcomp>.<lambda>r )rR r r� r rl rm z0TokenList.token_not_matching.<locals>.<listcomp>)rW r� r� r� �r r� r� r r r �token_not_matching s z TokenList.token_not_matchingc C s | � ||�d S )Nr r� r� r r r �token_matching s zTokenList.token_matchingc C s | j |||dd�S )z�Returns the previous token relative to *idx*.
If *skip_ws* is ``True`` (the default) whitespace tokens are ignored.
If *skip_cm* is ``True`` comments are ignored.
``None`` is returned if there's no previous token.
T)�_reverse)�
token_next)r r� r� r� r r r �
token_prev s zTokenList.token_prevc s2 |du rdS |d7 }� �fdd�}| j |||d�S )z�Returns the next token relative to *idx*.
If *skip_ws* is ``True`` (the default) whitespace tokens are ignored.
If *skip_cm* is ``True`` comments are ignored.
``None`` is returned if there's no next token.
Nr� r c r� r� r� r� r� r r r� - r� z%TokenList.token_next.<locals>.matcherr r� )r r� r� r� r� r� r r� r r� " s
zTokenList.token_nextc C s0 t |t�r|n| �|�}|| j|d� �|� S )zReturn list index of token.N)rW �int� token_indexr �index)r rk r� r r r r� 2 s zTokenList.token_indexc C s� |}| j | }|| }|r1t||�r1| j |d |� }|} | j �|� | j |d |�= t|�| _n| j ||� }||�} | g| j ||�< | | _|D ]}
| |
_qI| S )z+Replace tokens by an instance of *grp_cls*.r )r rW �extendr3 r, r. ) r �grp_clsr� r� � include_endr� � start_idx�end_idx� subtokens�grprk r r r � group_tokens7 s
zTokenList.group_tokensc C s, t |t�s
| �|�}| |_| j�||� dS )z Inserts *token* before *where*.N)rW r� r� r. r �insert)r �whererk r r r �
insert_beforeU s
zTokenList.insert_beforec C sV t |t�s
| �|�}| j||d�\}}| |_|du r"| j�|� dS | j�||� dS )z Inserts *token* after *where*.�r� N)rW r� r� r� r. r �appendr� )r r� rk r� �nidx�next_r r r � insert_after\ s
zTokenList.insert_afterc C s | � � duS )z(Returns ``True`` if an alias is present.N)r& r9 r r r � has_aliasg s zTokenList.has_aliasc C � dS )r Nr r9 r r r r& k � zTokenList.get_aliasc C s | � � p| �� S )a Returns the name of this identifier.
This is either it's alias or it's real name. The returned valued can
be considered as the name under which the object corresponding to
this identifier is known within the current statement.
)r& r r9 r r r �get_nameo s zTokenList.get_namec C r� )r Nr r9 r r r r x r� zTokenList.get_real_namec C s: | j tjdfd�\}}| �|�\}}|durt|j�S dS )ztReturn name of the parent object if any.
A parent object is identified by the first occurring dot.
r r
N)r
r r r� r r, )r r r �prev_r r r �get_parent_name| s zTokenList.get_parent_namec C s� |r | j |d� n| j }|rt|�n|}tjtjtjjg}|r%|�tj� |D ]#}|j |v r5t
|j � S t |t
tf�rJ|rD|�� S |�� S q'dS )z/Returns the name of the first token with a nameN)r �reversedr �Name�Wildcard�String�Symbolr� r r- r r, rW �
Identifier�Functionr r� )r r� r r r r �typesrk r r r r � s
��zTokenList._get_first_namer7 )Nr Nrq )r NF)TF)NNNrG N)TFF)r )T)NFFF)"r' r( r) r* rg r5 r: rt rw r> r| r� rO r� �propertyr� r� r� r
r� r� r� r� r� r� r� r� r� r& r� r r� r �
__classcell__r r ro r rh � sF
�
�rh c @ � e Zd ZdZdd� ZdS )� StatementzRepresents a SQL statement.c C s� | j dd�}|du r dS |jtjjtjjfv r|jS |jtjjkrM| �|�}| j |dd�\}}t
|t t f�rM| j |dd�\}}|durM|jtjjkrM|jS dS )aM Returns the type of a statement.
The returned value is a string holding an upper-cased reprint of
the first DML or DDL keyword. If the first token in this group
isn't a DML or DDL keyword "UNKNOWN" is returned.
Whitespaces and comments at the beginning of the statement
are ignored.
T)r� N�UNKNOWNr� )
r� r- r r �DML�DDLr/ �CTEr� r� rW r� �IdentifierList)r � first_token�fidx�tidxrk r � dml_keywordr r r �get_type� s
zStatement.get_typeN)r' r( r) r* r� r r r r r� � � r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) r� zNRepresents an identifier.
Identifiers may have aliases or typecasts.
c C s | j tjd�\}}|duS )z7Return ``True`` if this identifier contains a wildcard.r N)r
r r� )r r rk r r r � is_wildcard� s zIdentifier.is_wildcardc C s6 | j tjdfd�\}}| j|dd�\}}|r|jS dS )z<Returns the typecast or ``None`` of this object as a string.z::r
Fr� N)r
r r r� r, )r �midx�markerr� r� r r r � get_typecast� s zIdentifier.get_typecastc C s" | j tjjd�\}}|r|jS dS )z5Returns the ordering or ``None`` as uppercase string.r N)r
r r �Orderr/ )r r �orderingr r r � get_ordering� s zIdentifier.get_orderingc c s, � | j D ]}t|t�r|j dd� V qdS )z(Returns an iterator of index token listsr rG N)r rW �SquareBracketsr� r r r �get_array_indices� s �
��z Identifier.get_array_indicesN)r' r( r) r* r� r� r� r� r r r r r� � s r� c @ r� )r� z.A list of :class:`~sqlparse.sql.Identifier`'s.c c s, � | j D ]}|js|�tjd�s|V qdS )zkReturns the identifiers.
Whitespaces and punctuations are not included in this generator.
�,N)r r2 r^ r r r� r r r �get_identifiers� s �
��z IdentifierList.get_identifiersN)r' r( r) r* r� r r r r r� � r� r� c @ s< e Zd ZdZejjdfejdfgZej j
dfZ ejdfZ dS )� TypedLiteralzEA typed literal, such as "date '2001-09-28'" or "interval '2 hours'".N� TIMESTAMP)�DAY�HOUR�MINUTE�MONTH�SECOND�YEAR)
r' r( r) r* r r� �Builtinr �M_OPENr� �Single�M_CLOSE�M_EXTENDr r r r r� � s
r� c @ �0 e Zd ZdZejdfZejdfZedd� �Z dS )� ParenthesiszTokens between parenthesis.�(�)c C � | j dd� S �Nr rG r r9 r r r r� � � z Parenthesis._groupable_tokensN�
r' r( r) r* r r r� r� r� r� r r r r r� � �
r� c @ r� )r� z Tokens between square brackets�[�]c C r� r� r r9 r r r r� r� z SquareBrackets._groupable_tokensNr� r r r r r� � r� r� c @ � e Zd ZdZdS )�
Assignmentz An assignment like 'var := val;'N�r' r( r) r* r r r r r � r c @ �$ e Zd ZdZejdfZejdfZdS )�Ifz7An 'if' clause with possible 'else if' or 'else' parts.�IFzEND IFN�r' r( r) r* r r r� r� r r r r r �
r c @ r )�Forz
A 'FOR' loop.)�FOR�FOREACHzEND LOOPNr r r r r r r r c @ s( e Zd ZdZedd� �Zedd� �ZdS )�
Comparisonz/A comparison used for example in WHERE clauses.c C �
| j d S �Nr r r9 r r r �left rd zComparison.leftc C r
)NrG r r9 r r r �right rd zComparison.rightN)r' r( r) r* r� r r r r r r r s
r c @ r� )r� z
A comment.c C s | j o | j d jtjjkS r )r r- r r� � Multiliner9 r r r � is_multiline$ s zComment.is_multilineN)r' r( r) r* r r r r r r� ! r� r� c @ r )�WherezA WHERE clause.�WHERE) �ORDER BYzGROUP BY�LIMIT�UNIONz UNION ALL�EXCEPT�HAVING� RETURNING�INTONr r r r r r ( r r c @ r )�HavingzA HAVING clause.r )r r Nr r r r r r 0 r r c @ s. e Zd ZdZejdfZejdfZddd�ZdS ) �CasezAA CASE statement with one or more WHEN and possibly an ELSE part.�CASE�ENDFc C s� d}d}g }|}| j D ]n}|�tjd�rq |r |jtjv r q |�tjd�r/|�g g f� |}n$|�tjd�r9|}n|�tjd�rJ|�dg f� |}n |�tjd�rSd}|r^|s^|�g g f� ||krl|d d
�|� q ||kry|d d �|� q |S ) zeReturns a list of 2-tuples (condition, value).
If an ELSE exists condition is None.
r r r �WHEN�THEN�ELSENr rG r )r r^ r r r- r! r� )r r� � CONDITION�VALUE�ret�moderk r r r � get_cases; s6
�zCase.get_casesNrf ) r' r( r) r* r r r� r� r' r r r r r 6 s
r c @ r� )r� z A function or procedure call.c C sN | j d }|j D ] }t|t�r|�� S t|ttftjd�r$|g S qg S )z Return a list of parameters.rG )r� r ) r rW r� r� r r� r� r �Literal)r � parenthesisrk r r r �get_parametersn s
�zFunction.get_parametersN)r' r( r) r* r* r r r r r� k r� r� c @ r )�BeginzA BEGIN/END block.�BEGINr Nr r r r r r+ y r r+ c @ r )� OperationzGrouping of operationsNr r r r r r- r r- c @ r )�ValueszGrouping of valuesNr r r r r r. � r r. c @ r )�CommandzGrouping of CLI commands.Nr r r r r r/ � r r/ ) r* rL �sqlparser r �sqlparse.exceptionsr �sqlparse.utilsr r r r+ rh r� r� r� r� r� r� r r r r r� r r r r� r+ r- r. r/ r r r r �<module> s: n '
5