OwlCyberSecurity - MANAGER
Edit File: statement_splitter.cpython-39.pyc
a ���e� � @ s" d dl mZmZ G dd� d�ZdS )� )�sql�tokensc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �StatementSplitterz1Filter that split stream at individual statementsc C s | � � d S �N)�_reset��self� r �c/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/sqlparse/engine/statement_splitter.py�__init__ s zStatementSplitter.__init__c C s( d| _ d| _d| _d| _g | _d| _dS )z/Set the filter attributes to its default valuesFr N)� _in_declare� _is_create� _begin_depth� consume_wsr �levelr r r r r s zStatementSplitter._resetc C s� |t ju r|dkrdS |t ju r,|dkr,dS |t jvr:dS |�� }|t jju rb|�d�rbd| _dS |dkr�| jr�| jdkr�d| _dS |d kr�| jd7 _| jr�dS dS |d kr�t d| jd �| _dS |d v r�| jr�| jdkr�dS |d v r�dS dS ) z<Get the new split level (increase, decrease or remain equal)�(� �)���r ZCREATETZDECLAREZBEGINZEND)ZIFZFORZWHILEZCASE)zEND IFzEND FORz END WHILE) �T� Punctuation�Keyword�upperZDDL� startswithr r r �max)r �ttype�valueZunifiedr r r �_change_splitlevel s< ��z$StatementSplitter._change_splitlevelc c s� t jt jjf}|D ]x\}}| jr>||vr>t�| j�V | �� | j | � ||�7 _ | j� t� ||�� | j dkr|t j u r|dkrd| _q| jr�tdd� | jD ��s�t�| j�V dS )zProcess the streamr �;Tc s s | ] }|j V qd S r )Z is_whitespace)�.0�tr r r � <genexpr>j � z,StatementSplitter.process.<locals>.<genexpr>N)r � Whitespace�Comment�Singler r Z Statementr r r r �append�Tokenr �all)r �streamZ EOS_TTYPEr r r r r �processO s zStatementSplitter.processN)�__name__� __module__� __qualname__�__doc__r r r r* r r r r r s 4r N)Zsqlparser r r r r r r r �<module> s