OwlCyberSecurity - MANAGER
Edit File: sql.cpython-39.pyc
a gH�f0R � @ 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 �ZG d*d+� d+e e �Z G d,d-� d-e �Z G d.d/� d/e �Z G d0d1� d1e �Z G d2d3� d3e �Z dS )4zFThis 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 �N/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/sqlparse/sql.py� get_real_name s z NameAliasMixin.get_real_namec C sh | j tjdfd�\}}|dur0| j|d dd�S | j tjd�\}}t| j�dkrd|durd| jdd �S dS ) �2Returns the alias for this identifier or ``None``.ZASr N� T)�keywords��t� ��reverse)r r �Keywordr � Whitespace�lenr )r Zkw_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_whitespace� is_newlinec C s` t |�}|| _|| _d | _d| _|tjv | _| jtjv | _ | jtj v | _ | jrV|� � n|| _ d S )NF)�strr* r+ r, r/ r r r. r r0 ZNewliner1 �upperr- )r r+ r* r r r �__init__3 s zToken.__init__c C s | j S �N�r* �r r r r �__str__> s z Token.__str__c C sF | � � }| �� }|�d�r(|�d�r(dnd}djf dt| �it� ��S )N�'�"z"<{cls} {q}{value}{q} at 0x{id:2X}>�id)�_get_repr_name�_get_repr_value� startswith�endswith�formatr; �locals)r �clsr* �qr r r �__repr__E s ��zToken.__repr__c C s t | j��d�d S )Nr ���)r2 r+ �splitr7 r r r r< M s zToken._get_repr_namec C s2 t | �}t|�dkr$|d d� d }t�dd|�S )N� � z...z\s+� )r2 r! �re�sub)r �rawr r r r= P s zToken._get_repr_valuec c s | V dS )zResolve subgroups.Nr r7 r r r �flattenV s z Token.flattenFc s� | j |u }|r|du r|S t|t�r*|f}|rp| jr:tjnd� � fdd�|D �}|D ]}|�| j�rT dS qTdS | jr�dd� |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 r5 )rJ �compile��.0�v��flagr r � <genexpr>p � z Token.match.<locals>.<genexpr>TFc s s | ]}|� � V qd S r5 )r3 rO r r r rT x rU )r+ � isinstancer2 r. rJ � IGNORECASE�searchr- )r r+ �values�regexZ type_matched�patternr rR r �matchZ s z Token.matchc C s$ | j }|r t||�rdS |j }qdS )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, rV )r Z 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� s zToken.is_child_ofc C s" | j }|r ||krdS |j }qdS )z7Returns ``True`` if *other* is in this tokens ancestry.TFr^ )r r_ r, r r r � has_ancestor� s zToken.has_ancestorN)F)r% r& r' r( � __slots__r4 r8 rD r< r= rM r\ r] r` ra 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&� ZdDd'd(�ZdEd)d*�ZdFd+d,�ZdGd-d.�Zd/d0� ZdHd1d2�Zd3d4� Zd5d6� Z d7d8� Z d9d:� Z d;d<� Z dId=d>�Z � Z!S )J� 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 r^ )�setattr�rP �tokenr7 r r � <listcomp>� rU z&TokenList.__init__.<locals>.<listcomp>T)r �superr4 r2 r/ )r r �� __class__r7 r r4 � s zTokenList.__init__c C s d� dd� | �� D ��S )N� c s s | ] }|j V qd S r5 r6 re r r r rT � rU z$TokenList.__str__.<locals>.<genexpr>)�joinrM r7 r r r r8 � s zTokenList.__str__c C s t | j�S r5 )�iterr r7 r r r �__iter__� s zTokenList.__iter__c C s | j | S r5 r )r �itemr r r � __getitem__� s zTokenList.__getitem__c C s t | �jS r5 )�typer% r7 r r r r< � s zTokenList._get_repr_namer rk c C s� t | j�}t| j�D ]�\}}|�� }|�� } ||d k} | r@dnd} | �d�r\| �d�r\dnd} tdjf i t � ��|d� |j r|du s�||k r| r�d nd } |� ||d ||| � qdS ) z Pretty-print the object tree.r z`- z|- r9 r: z${_pre}{pre}{idx} {cls} {q}{value}{q})�fileNz z| ) r! r � enumerater< r= r> r? �printr@ rA r/ � _pprint_tree)r � max_depth�depth�f�_preZ token_count�idxrf rB r* �last�prerC Z parent_prer r r ru � s �� zTokenList._pprint_treec C sH d}| � � D ]6}|t|j� }|| kr2|k r>n n| S |}q dS )z-Returns the token that is on position offset.r N)rM r! r* )r �offsetrz rf �endr r r �get_token_at_offset� s z TokenList.get_token_at_offsetc c s` z,| j D ] }|jr"|�� E dH q|V qW n. tyZ } ztd�|�W Y d}~n d}~0 0 dS )znGenerator yielding ungrouped tokens. This method is recursively called for all child tokens. Nz Maximum recursion depth exceeded)r r/ rM �RecursionErrorr )r rf �errr r r rM � s zTokenList.flattenc c s | j D ]}|jr|V qd S r5 )r r/ �r rf r r r � get_sublists� s zTokenList.get_sublistsc C s | j S r5 r r7 r r r �_groupable_tokens� s zTokenList._groupable_tokensFc C s� |du r dS t |ttf�s |f}|rB|du s0J �t|d dd�}n |du rTt| j�}t||�}|D ]0}| j| }|D ] }||�rt||f S qtqbdS )z next token that match functionsNr rE �NN)rV �list�tuple�ranger! r ) r �funcs�startr~ r Zindexesrz rf �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 s �r | j p � o t| tjtd� S �N)r �i�r0 r r �Comment�Ztk��skip_cm�skip_wsr r �matcher s �z&TokenList.token_first.<locals>.matcherr �r� )r r� r� r� r r� r � token_first� s zTokenList.token_firstrE c s" |d7 }| � � ��fdd�||�S )Nr c s t | � ���S r5 )r r� �r� r r r r �<lambda> rU z)TokenList.token_next_by.<locals>.<lambda>r� )r r� r r rz 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 r5 r r� �r� r r r� rU z9TokenList.token_not_matching.<locals>.<listcomp>.<lambda>r )rP r r� r rg rU z0TokenList.token_not_matching.<locals>.<listcomp>)rV r� r� r� �r r� rz 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 rz r� r� r r r � token_prev s zTokenList.token_prevc s2 |du r dS |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 s �r | j p � o t| tjtd� S r� r� r� r� r r r� . s �z%TokenList.token_next.<locals>.matcherr r� )r rz 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)rV �int� token_indexr �index)r rf r� r r r r� 3 s zTokenList.token_indexc C s� |}| j | }|| }|rbt||�rb| j |d |� }|} | j �|� | j |d |�= t|�| _n,| j ||� }||�} | g| j ||�< | | _|D ] } | | _q�| S )z+Replace tokens by an instance of *grp_cls*.r )r rV �extendr2 r* r, ) r Zgrp_clsr� r~ Z include_endr� Z start_idxZend_idxZ subtokens�grprf r r r � group_tokens8 s zTokenList.group_tokensc C s, t |t�s| �|�}| |_| j�||� dS )z Inserts *token* before *where*.N)rV r� r� r, r �insert)r �whererf r r r � insert_beforeV s zTokenList.insert_beforec C sT t |t�s| �|�}| j||d�\}}| |_|du rB| j�|� n| j�||� dS )z Inserts *token* after *where*.�r� N)rV r� r� r� r, r �appendr� )r r� rf 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$ r7 r r r � has_aliash s zTokenList.has_aliasc C s dS )r Nr r7 r r r r$ l s 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 r7 r r r �get_namep s zTokenList.get_namec C s dS )r Nr r7 r r r r y s zTokenList.get_real_namec C s: | j tjdfd�\}}| �|�\}}|dur6t|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 Zprev_r r r �get_parent_name} s zTokenList.get_parent_namec C s� |r| j |d� n| j }|r$t|�n|}tjtjtjjg}|rJ|�tj� |D ]B}|j |v rjt |j � S t |t tf�rN|r�|�� n|�� S qNdS )z/Returns the name of the first token with a nameN)r �reversedr �Name�Wildcard�String�Symbolr� r r+ r r* rV � Identifier�Functionr r� )r rz r r r r �typesrf r r r r � s zTokenList._get_first_name)N)Nr Nrk )r NF)TF)NNNrE N)TF)TFF)r )TF)T)NFFF)"r% r& r' r( rb r4 r8 rn rp r< ru r rM r� �propertyr� r� r� r r� r� r� r� r� r� r� r� r� r$ r� r r� r � __classcell__r r ri r rc � sD � �rc c @ s e Zd ZdZdd� ZdS )� StatementzRepresents a SQL statement.c C s� | j dd�}|du rdS |jtjjtjjfv r4|jS |jtjjkr�| �|�}|dur�| j |dd�\}}t |t t f�rL| j |dd�\}}|durL|jtjjkrL|jS qLdS )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 ZDMLZDDLr- ZCTEr� r� rV r� �IdentifierList)r rf Ztidxr r r �get_type� s �zStatement.get_typeN)r% r& r' r( r� r r r r r� � s 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 rf r r r � is_wildcard� s zIdentifier.is_wildcardc C s6 | j tjdfd�\}}| j|dd�\}}|r2|jS dS )z<Returns the typecast or ``None`` of this object as a string.z::r Fr� N)r r r r� r* )r Zmidx�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 ZOrderr- )r r Zorderingr 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 rE N)r rV �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 @ s e Zd ZdZdd� ZdS )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 r0 r\ r r r� r r r �get_identifiers� s z IdentifierList.get_identifiersN)r% r& r' r( r� r r r r r� � s 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)ZDAYZHOURZMINUTEZMONTHZSECONDZYEAR) r% r& r' r( r r� �Builtinr �M_OPENr� �Single�M_CLOSEZM_EXTENDr r r r r� � s r� c @ s0 e Zd ZdZejdfZejdfZedd� �Z dS )� ParenthesiszTokens between parenthesis.�(�)c C s | j dd� S �Nr rE r r7 r r r r� � s z Parenthesis._groupable_tokensN� r% r& r' r( r r r� r� r� r� r r r r r� � s r� c @ s0 e Zd ZdZejdfZejdfZedd� �Z dS )r� z Tokens between square brackets�[�]c C s | j dd� S r� r r7 r r r r� s z SquareBrackets._groupable_tokensNr� r r r r r� � s r� c @ s e Zd ZdZdS )� Assignmentz An assignment like 'var := val;'N�r% r& r' r( r r r r r� s r� c @ s$ e Zd ZdZejdfZejdfZdS )�Ifz7An 'if' clause with possible 'else if' or 'else' parts.ZIFzEND IFN�r% r& r' r( r r r� r� r r r r r� s r� c @ s$ e Zd ZdZejdfZejdfZdS )�Forz A 'FOR' loop.)ZFORZFOREACHzEND LOOPNr� r r r r r� s r� c @ s( e Zd ZdZedd� �Zedd� �ZdS )� Comparisonz/A comparison used for example in WHERE clauses.c C s | j d S �Nr r r7 r r r �left s zComparison.leftc C s | j d S )NrE r r7 r r r �right s zComparison.rightN)r% r& r' r( r� r� r� r r r r r� s r� c @ s e Zd ZdZdd� ZdS )r� z A comment.c C s | j o| j d jtjjkS r� )r r+ r r� � Multiliner7 r r r � is_multiline& s zComment.is_multilineN)r% r& r' r( r� r r r r r� # s r� c @ s$ e Zd ZdZejdfZejdfZdS )�WherezA WHERE clause.ZWHERE) �ORDER BYzGROUP BY�LIMITZUNIONz UNION ALLZEXCEPT�HAVINGZ RETURNINGZINTONr� r r r r r� * s r� c @ s e Zd ZdZejdfZdS )�OverzAn OVER clause.ZOVERN)r% r& r' r( r r r� r r r r r� 2 s r� c @ s$ e Zd ZdZejdfZejdfZdS )�HavingzA HAVING clause.r� )r� r� Nr� r r r r r� 7 s 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 ]�}|�tjd�r,qn~|r@|jtjv r@qnj|�tjd�rb|�g g f� |}nH|�tjd�rv|}n4|�tjd�r�|�dg f� |}n|�tjd�r�d}|r�|s�|�g g f� ||kr�|d d �|� q||kr|d d �|� q|S ) zeReturns a list of 2-tuples (condition, value). If an ELSE exists condition is None. r r r� ZWHENZTHEN�ELSENr� rE r )r r\ r r r+ r r� )r r� Z CONDITIONZVALUE�ret�moderf r r r � get_casesB s4 zCase.get_casesN)F) r% r& r' r( r r r� r� r� r r r r r� = s r� c @ s e Zd ZdZdd� Zdd� ZdS )r� z A function or procedure call.c C sZ | j td�d }g }|jD ]:}t|t�r4|�� S t|ttt ft j d�r|� |� q|S )z Return a list of parameters.�r� r )r� r ) r r� r rV r� r� r r� r� r� r �Literalr� )r Z parenthesis�resultrf r r r �get_parametersu s � zFunction.get_parametersc C s" | j td�}|sdS |d jd S )z Return the window if it exists.r� Nr rE )r r� r )r Z over_clauser r r � get_window� s zFunction.get_windowN)r% r& r' r( r� r� r r r r r� r s r� c @ s$ e Zd ZdZejdfZejdfZdS )�BeginzA BEGIN/END block.ZBEGINr� Nr� r r r r r� � s r� c @ s e Zd ZdZdS )� OperationzGrouping of operationsNr� r r r r r� � s r� c @ s e Zd ZdZdS )�ValueszGrouping of valuesNr� r r r r r� � s r� c @ s e Zd ZdZdS )�CommandzGrouping of CLI commands.Nr� r r r r r� � s r� )!r( rJ Zsqlparser r Zsqlparse.exceptionsr Zsqlparse.utilsr r r r) rc r� 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: o ( 5