shell bypass 403
o
�\O� � @ s d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
dd lmZ dd
lmZmZmZ G d d � d e�Zd
d� ZdDdd�ZdDdd�ZdDdd�ZdDdd�ZG dd� de�Z G dd� de �Z G d d � d e �Z G d d � d e �Z G d d!� d!e�Z G d"d#� d#e �Z!G d$d%� d%e �Z"G d&d'� d'e �Z#G d(d)� d)e �Z$G d*d+� d+e �Z%G d,d-� d-e �Z&G d.d/� d/e �Z'G d0d1� d1e �Z(G d2d3� d3e �Z)G d4d5� d5e�Z*G d6d7� d7e �Z+G d8d9� d9e+�Z,G d:d;� d;e �Z-G d<d=� d=e�Z.G d>d?� d?e.�Z/G d@dA� dAe/�Z0G dBdC� dCe�Z1dS )Ez Implementation of Statements.� N� )�ProgrammingError)�
ExprParser)� STRING_TYPES)�
Algorithms�
Securities)�DbDoc)�mysqlx_crud_pb2)� SqlResult�Result�
ColumnTypec @ s e Zd Zdd� ZdS )�Exprc C �
|| _ d S �N)�expr)�selfr � r �G/usr/local/python-3.10/lib/python3.10/site-packages/mysqlx/statement.py�__init__& �
z
Expr.__init__N)�__name__�
__module__� __qualname__r r r r r r
% s r
c G s* t | �dkrt| d ttf�r| d S | S )Nr r )�len�
isinstance�list�tuple)�valuesr r r �flexible_params* s r � c C sP d|v r | d dkr| d dkp| d dko| d dkS | d dko'| d dkS )z�Check if the given identifier is quoted.
Args:
identifier (string): Identifier to check.
sql_mode (Optional[string]): SQL mode.
Returns:
`True` if the identifier has backtick quotes, and False otherwise.
� ANSI_QUOTESr �`����"r ��
identifier�sql_moder r r �is_quoted_identifier0 s
�r' c C s: t | |�r| S d|v rd�| �dd��S d�| �dd��S )a� Quote the given identifier with backticks, converting backticks (`) in
the identifier name with the correct escape sequence (``) unless the
identifier is quoted (") as in sql_mode set to ANSI_QUOTES.
Args:
identifier (string): Identifier to quote.
sql_mode (Optional[string]): SQL mode.
Returns:
A string with the identifier quoted with backticks.
r z"{0}"r# z""z`{0}`r! z``)r' �format�replacer$ r r r �quote_identifierA s
r* c s d� � fdd�| D ��S )a Quote the given multi-part identifier with backticks.
Args:
identifiers (iterable): List of identifiers to quote.
sql_mode (Optional[string]): SQL mode.
Returns:
A string with the multi-part identifier quoted with backticks.
�.c s g | ]}t |� ��qS r �r* )�.0r% �r& r r �
<listcomp>_ s �z.quote_multipart_identifier.<locals>.<listcomp>)�join)Z identifiersr& r r. r �quote_multipart_identifierU s
�r1 c C s^ d|v rdnd}||v rd� |�nd}|�|d�}t|�du r!| n|d �|�|d �|�fS ) Nr r# r! z.{0}r+ r r r" )r( �splitr �strip)Zdefault_schema�
table_namer& �quote� delimiter�tempr r r �parse_table_namec s �r8 c @ s: e Zd ZdZd dd�Zedd� �Zedd� �Zd d
� Zd S )
� Statementa Provides base functionality for statement objects.
Args:
target (object): The target database object, it can be
:class:`mysqlx.Collection` or :class:`mysqlx.Table`.
doc_based (bool): `True` if it is document based.
Tc C s& || _ || _|r|j| _d S d | _d S r )�_target�
_doc_based� _connection)r �target� doc_basedr r r r s s zStatement.__init__c C s | j S )z,object: The database object target.
)r: �r r r r r= x s zStatement.targetc C s | j jS )z3:class:`mysqlx.Schema`: The Schema object.
)r: �schemar? r r r r@ ~ s zStatement.schemac C � t ��zqExecute the statement.
Raises:
NotImplementedError: This method must be implemented.
��NotImplementedErrorr? r r r �execute� � zStatement.executeN)T) r r r �__doc__r �propertyr= r@ rE r r r r r9 k s
r9 c sl e Zd ZdZd� fdd� Zdd� Zdd � Zdd d �Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
� ZS )�FilterableStatementa� A statement to be used with filterable statements.
Args:
target (object): The target database object, it can be
:class:`mysqlx.Collection` or :class:`mysqlx.Table`.
doc_based (Optional[bool]): `True` if it is document based
(default: `True`).
condition (Optional[str]): Sets the search condition to filter
documents or records.
TNc sd t t| �j||d� d| _d| _d| _d| _d| _d| _d| _ i | _
g | _ |d ur0| � |� d S d S )N�r= r> F)
�superrI r �_has_projection�
_has_where�
_has_limit� _has_sort�
_has_group_by� _has_having�
_has_bindings� _binding_map� _bindings�where)r r= r> � condition�� __class__r r r � s ��z FilterableStatement.__init__c C s0 d| _ || _t|| j �}|�� | _|j| _| S )a Sets the search condition to filter.
Args:
condition (str): Sets the search condition to filter documents or
records.
Returns:
mysqlx.FilterableStatement: FilterableStatement object.
T)rM �_wherer r; r Z _where_exprZ placeholder_name_to_positionrS )r rV r r r r rU � s
zFilterableStatement.wherec G �4 t |� }d| _d�|�| _t| j| j ��� | _| S �NT�,)r rL r0 �_projection_strr r; Z parse_table_select_projectionZ_projection_expr�r �fieldsr r r � _projection� s ��z FilterableStatement._projectionr c C s d| _ || _|| _| S )aE Sets the maximum number of records or documents to be returned.
Args:
row_count (int): The maximum number of records or documents.
offset (Optional[int]) The number of records or documents to skip.
Returns:
mysqlx.FilterableStatement: FilterableStatement object.
T)rN �
_limit_offset�_limit_row_count)r Z row_count�offsetr r r �limit� s
zFilterableStatement.limitc G rZ )z�Sets the sorting criteria.
Args:
*sort_clauses: The expression strings defining the sort criteria.
Returns:
mysqlx.FilterableStatement: FilterableStatement object.
Tr\ )r rO r0 � _sort_strr r; Zparse_order_specZ
_sort_expr)r Z sort_clausesr r r �sort� s ��zFilterableStatement.sortc G s4 t |� }d| _d�|�| _t| j| j ��� | _d S r[ )r rP r0 �
_grouping_strr r; Zparse_expr_listZ _groupingr^ r r r � _group_by� s ��z FilterableStatement._group_byc C s d| _ t|| j ��� | _d S �NT)rQ r r; r �_having�r rV r r r rj � s zFilterableStatement._havingc G sV d| _ t|�}|dkr| �|d � | S |dkr td��| j�|d |d d�� | S )a� Binds a value to a specific placeholder.
Args:
*args: The name of the placeholder and the value to bind.
A :class:`mysqlx.DbDoc` object or a JSON string
representation can be used.
Returns:
mysqlx.FilterableStatement: FilterableStatement object.
Raises:
ProgrammingError: If the number of arguments is invalid.
Tr r � z#Invalid number of arguments to bind)�name�value)rR r � _bind_singler rT �append)r �args�countr r r �bind� s �zFilterableStatement.bindc C sV t |t�r| �t|�� d S t |t�r't�|�}|�� D ] }| �||| � q d S d S r )r r rs �strr �json�loads�keys)r �object�dict�keyr r r ro s
�z FilterableStatement._bind_singlec C rA rB rC r? r r r rE
rF zFilterableStatement.execute)TN)r )r r r rG r rU r` rd rf rh rj rs ro rE �
__classcell__r r rW r rI � s
rI c �( e Zd ZdZ� fdd�Zdd� Z� ZS )� SqlStatementz�A statement for SQL execution.
Args:
connection (mysqlx.connection.Connection): Connection object.
sql (string): The sql statement to be executed.
c s$ t t| �jd dd� || _|| _d S �NFrJ )rK r} r r< �_sql)r Z
connection�sqlrW r r r s
zSqlStatement.__init__c C s | j �| j� t| j �S )zaExecute the statement.
Returns:
mysqlx.SqlResult: SqlResult object.
)r< Zsend_sqlr r
r? r r r rE s
zSqlStatement.execute�r r r rG r rE r{ r r rW r r} s r} c �0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )� AddStatementz�A statement for document addition on a collection.
Args:
collection (mysqlx.Collection): The Collection object.
c s" t t| �j|d� g | _g | _d S �N)r= )rK r� r �_values�_ids�r �
collectionrW r r r / s
zAddStatement.__init__c G s: t |� D ]}t|t�r| j�|� q| j�t|�� q| S )z�Adds a list of documents into a collection.
Args:
*values: The documents to be added into the collection.
Returns:
mysqlx.AddStatement: AddStatement object.
)r r r r� rp )r r �valr r r �add4 s
zAddStatement.addc C s< t | j�dkr
t� S | jD ]
}| j�|�� � q
| j�| �S )�[Execute the statement.
Returns:
mysqlx.Result: Result object.
r )r r� r r� rp Z ensure_idr< � send_insert)r �docr r r rE D s
zAddStatement.execute)r r r rG r r� rE r{ r r rW r r� ) s
r� c @ s e Zd Zddd�Zdd� ZdS )�
UpdateSpecNc C sl |t jjkr| �||� d S || _|| _t|�dkr'|d dkr'|dd � | _t| jd��� j | _|| _
d S )Nr �$r F) �
MySQLxCrud�UpdateOperation�SET�
_table_set� update_type�sourcer r Zdocument_fieldr% rn )r r� r� rn r r r r T s ��
zUpdateSpec.__init__c C s$ t jj| _t|d��� | _|| _d S ri )r� r� r� r� r Zparse_table_update_fieldr� rn )r r� rn r r r r� ` s
zUpdateSpec._table_setr )r r r r r� r r r r r� S s
r� c sR e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�ModifyStatementa A statement for document update operations on a Collection.
Args:
collection (mysqlx.Collection): The Collection object.
condition (Optional[str]): Sets the search condition to identify the
documents to be updated.
Nc s t t| �j||d� g | _d S )N)r= rV )rK r� r � _update_ops�r r� rV rW r r r n s �
zModifyStatement.__init__c C � | j �ttjj||�� | S )a2 Sets or updates attributes on documents in a collection.
Args:
doc_path (string): The document path of the item to be set.
value (string): The value to be set on the specified attribute.
Returns:
mysqlx.ModifyStatement: ModifyStatement object.
)r� rp r� r� r� ZITEM_SET�r Zdoc_pathrn r r r �sets �
�zModifyStatement.setc C r� )aj Add an update to the statement setting the field, if it exists at
the document path, to the given value.
Args:
doc_path (string): The document path of the item to be set.
value (object): The value to be set on the specified attribute.
Returns:
mysqlx.ModifyStatement: ModifyStatement object.
)r� rp r� r� r� Z ITEM_REPLACEr� r r r �change� s
��zModifyStatement.changec G s | j �dd� t|� D �� | S )a Removes attributes from documents in a collection.
Args:
doc_path (string): The document path of the attribute to be
removed.
Returns:
mysqlx.ModifyStatement: ModifyStatement object.
c S s g | ] }t tjj|��qS r )r� r� r� Z ITEM_REMOVE)r- �xr r r r/ � s ��z)ModifyStatement.unset.<locals>.<listcomp>)r� �extendr )r Z doc_pathsr r r �unset� s
�zModifyStatement.unsetc C r� )a� Insert a value into the specified array in documents of a
collection.
Args:
field (string): A document path that identifies the array attribute
and position where the value will be inserted.
value (object): The value to be inserted.
Returns:
mysqlx.ModifyStatement: ModifyStatement object.
)r� rp r� r� r� Z ARRAY_INSERT�r �fieldrn r r r � array_insert� s �z ModifyStatement.array_insertc C r� )a� Inserts a value into a specific position in an array attribute in
documents of a collection.
Args:
doc_path (string): A document path that identifies the array
attribute and position where the value will be
inserted.
value (object): The value to be inserted.
Returns:
mysqlx.ModifyStatement: ModifyStatement object.
)r� rp r� r� r� Z ARRAY_APPENDr� r r r � array_append� s
��z ModifyStatement.array_appendc C � | j �| �S �r� �r< �updater? r r r rE � � zModifyStatement.executer ) r r r rG r r� r� r� r� r� rE r{ r r rW r r� f s r� c sB e Zd ZdZd
� fdd� Zdd� Zdd� Zd d
� Zd d � Z� Z S )�
FindStatementa� A statement document selection on a Collection.
Args:
collection (mysqlx.Collection): The Collection object.
condition (Optional[str]): An optional expression to identify the
documents to be retrieved. If not specified
all the documents will be included on the
result unless a limit is set.
Nc s t t| ��|d|� d S ri )rK r� r r� rW r r r � s zFindStatement.__init__c G s
| j |� S )z�Sets a document field filter.
Args:
*fields: The string expressions identifying the fields to be
extracted.
Returns:
mysqlx.FindStatement: FindStatement object.
)r` r^ r r r r_ � s
zFindStatement.fieldsc G � | j |� | S )z�Sets a grouping criteria for the resultset.
Args:
*fields: The string expressions identifying the grouping criteria.
Returns:
mysqlx.FindStatement: FindStatement object.
�rh r^ r r r �group_by� �
zFindStatement.group_byc C � | � |� | S )a= Sets a condition for records to be considered in agregate function
operations.
Args:
condition (string): A condition on the agregate functions used on
the grouping criteria.
Returns:
mysqlx.FindStatement: FindStatement object.
�rj rk r r r �having� �
zFindStatement.havingc C r� )zaExecute the statement.
Returns:
mysqlx.DocResult: DocResult object.
�r< �findr? r r r rE � r� zFindStatement.executer )
r r r rG r r_ r� r� rE r{ r r rW r r� � s r� c s@ e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Z� Z S ) �SelectStatementz�A statement for record retrieval operations on a Table.
Args:
table (mysqlx.Table): The Table object.
*fields: The fields to be retrieved.
c s t t| ��|d� | j|� d S �NF)rK r� r r` �r �tabler_ rW r r r s zSelectStatement.__init__c G r� )z�Sets a grouping criteria for the resultset.
Args:
*fields: The fields identifying the grouping criteria.
Returns:
mysqlx.SelectStatement: SelectStatement object.
r� r^ r r r r� r� zSelectStatement.group_byc C r� )a; Sets a condition for records to be considered in agregate function
operations.
Args:
condition (str): A condition on the agregate functions used on the
grouping criteria.
Returns:
mysqlx.SelectStatement: SelectStatement object.
r� rk r r r r� r� zSelectStatement.havingc C r� )zaExecute the statement.
Returns:
mysqlx.RowResult: RowResult object.
r� r? r r r rE , r� zSelectStatement.executec
C s� | j r d�| j�nd}| jrd�| j�nd}| jr d�| j�nd}| jr*d�| j�nd}| j r7d�| j
| j �nd}djt | dd �| j
j| jj|||||d
�}|S ) Nz
WHERE {0}r z
GROUP BY {0}z HAVING {0}z
ORDER BY {0}z LIMIT {0} OFFSET {1}zISELECT {select} FROM {schema}.{table}{where}{group}{having}{order}{limit}r] �*)�selectr@ r� rd rU �groupr� �order)rM r( rY rP rg rQ rj rO re rN rb ra �getattrr@ rm r= )r rU r� r� Zorder_byrd �stmtr r r �get_sql4 s, ���
���
�zSelectStatement.get_sql)
r r r rG r r� r� rE r� r{ r r rW r r� s r� c r� )�InsertStatementz�A statement for insert operations on Table.
Args:
table (mysqlx.Table): The Table object.
*fields: The fields to be inserted.
c s( t t| �j|dd� t|� | _g | _d S r~ )rK r� r r �_fieldsr� r� rW r r r M s
zInsertStatement.__init__c G s | j �tt|� �� | S )z�Set the values to be inserted.
Args:
*values: The values of the columns to be inserted.
Returns:
mysqlx.InsertStatement: InsertStatement object.
)r� rp r r �r r r r r r R s zInsertStatement.valuesc C r� r� )r< r� r? r r r rE ^ r� zInsertStatement.execute)r r r rG r r rE r{ r r rW r r� F s
r� c r� )�UpdateStatementz�A statement for record update operations on a Table.
Args:
table (mysqlx.Table): The Table object.
*fields: The fields to be updated.
c s t t| �j|dd� g | _d S r~ )rK r� r r� r� rW r r r n s
zUpdateStatement.__init__c C r� )a Updates the column value on records in a table.
Args:
field (string): The column name to be updated.
value (object): The value to be set on the specified column.
Returns:
mysqlx.UpdateStatement: UpdateStatement object.
)r� rp r� r� r� r� r� r r r r� r r� zUpdateStatement.setc C r� )zZExecute the statement.
Returns:
mysqlx.Result: Result object
r� r? r r r rE � r� zUpdateStatement.execute)r r r rG r r� rE r{ r r rW r r� g s
r� c r| )�RemoveStatementz�A statement for document removal from a collection.
Args:
collection (mysqlx.Collection): The Collection object.
c s t t| �j|d� d S r� )rK r� r r� rW r r r � s zRemoveStatement.__init__c C r� r� �r< �deleter? r r r rE � r� zRemoveStatement.executer� r r rW r r� � � r� c s* e Zd ZdZd� fdd� Zdd� Z� ZS )�DeleteStatementz�A statement that drops a table.
Args:
table (mysqlx.Table): The Table object.
condition (Optional[str]): The string with the filter expression of
the rows to be deleted.
Nc s t t| �j||dd� d S )NF)r= rV r> )rK r� r )r r� rV rW r r r � s
�zDeleteStatement.__init__c C r� r� r� r? r r r rE � r� zDeleteStatement.executer r� r r rW r r� � s r� c r� )� CreateCollectionIndexStatementz�A statement that creates an index on a collection.
Args:
collection (mysqlx.Collection): Collection.
index_name (string): Index name.
is_unique (bool): `True` if the index is unique.
c s( t t| �j|d� || _|| _g | _d S r� )rK r� r � _index_name�
_is_uniquer� )r r� �
index_nameZ is_uniquerW r r r � s
z'CreateCollectionIndexStatement.__init__c C s | j �|||f� | S )a� Add the field specification to this index creation statement.
Args:
document_path (string): The document path.
column_type (string): The column type.
is_required (bool): `True` if the field is required.
Returns:
mysqlx.CreateCollectionIndexStatement: CreateCollectionIndexStatement object.
)r� rp )r Z
document_pathZ column_typeZ is_requiredr r r r� � s z$CreateCollectionIndexStatement.fieldc C s> dd� | j D �}| jjddd| jjj| jj| j| jg|�R � S )r� c S s g | ] }|D ]}|�qqS r r )r- Zsublist�itemr r r r/ � s z:CreateCollectionIndexStatement.execute.<locals>.<listcomp>�xpluginZcreate_collection_indexT)r� r< �execute_nonqueryr: r@ rm r� r� r^ r r r rE � s ��z&CreateCollectionIndexStatement.execute)r r r rG r r� rE r{ r r rW r r� � s
r� c r| )� DropCollectionIndexStatementz�A statement that drops an index on a collection.
Args:
collection (mysqlx.Collection): The Collection object.
index_name (string): The index name.
c s t t| �j|d� || _d S r� )rK r� r r� )r r� r� rW r r r � s
z%DropCollectionIndexStatement.__init__c C s" | j �ddd| jjj| jj| j�S )r� r� Zdrop_collection_indexT)r< r� r: r@ rm r� r? r r r rE � s �z$DropCollectionIndexStatement.executer� r r rW r r� � s r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
TableIndexr rl c C s || _ || _|| _d S r )�_name� _index_type�_columns)r rm Z
index_type�columnsr r r r � s
zTableIndex.__init__c C s6 d}| j tju r |d7 }|d7 }|�| jd�| j��S )Nr zUNIQUE zINDEX {0} ({1})r\ )r� r� � UNIQUE_INDEXr( r� r0 r� )r r� r r r r� � s
zTableIndex.get_sqlN)r r r r� �INDEXr r� r r r r r� � s
r� c sZ e Zd ZdZd� fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z � Z S )�CreateViewStatementz�A statement for creating views.
Args:
view (mysqlx.View): The View object.
replace (Optional[bool]): `True` to add replace.
Fc s\ t t| �j|dd� || _|j| _|j| _|| _g | _ t
j | _ t
j| _d | _d | _d | _d S r~ )rK r� r �_viewr@ �_schemarm r� �_replacer� r Z UNDEFINED�
_algorithmr ZDEFINER� _security�_definer� _defined_as�
_check_option)r �viewr) rW r r r s
z CreateViewStatement.__init__c C s dd� |D �| _ | S )z�Sets the column names.
Args:
columns (list): The list of column names.
Returns:
mysqlx.CreateViewStatement: CreateViewStatement object.
c S s g | ]}t |��qS r r, )r- �colr r r r/ s z/CreateViewStatement.columns.<locals>.<listcomp>)r� )r r� r r r r� � zCreateViewStatement.columnsc C �
|| _ | S )z�Sets the algorithm.
Args:
mysqlx.constants.ALGORITHMS: The algorithm.
Returns:
mysqlx.CreateViewStatement: CreateViewStatement object.
)r� )r � algorithmr r r r� ! � z CreateViewStatement.algorithmc C r� )z�Sets the SQL security mode.
Args:
mysqlx.constants.SECURITIES: The SQL security mode.
Returns:
mysqlx.CreateViewStatement: CreateViewStatement object.
)r� )r �securityr r r r� - r� z CreateViewStatement.securityc C r� )z�Sets the definer.
Args:
definer (string): The definer.
Returns:
mysqlx.CreateViewStatement: CreateViewStatement object.
)r� )r �definerr r r r� 9 r� zCreateViewStatement.definerc C r� )z�Sets the SelectStatement statement for describing the view.
Args:
mysqlx.SelectStatement: SelectStatement object.
Returns:
mysqlx.CreateViewStatement: CreateViewStatement object.
)r� )r � statementr r r �
defined_asE r� z CreateViewStatement.defined_asc C r� )z�Sets the check option.
Args:
mysqlx.constants.CHECK_OPTIONS: The check option.
Returns:
mysqlx.CreateViewStatement: CreateViewStatement object.
)r� )r � check_optionr r r �with_check_optionQ r� z%CreateViewStatement.with_check_optionc
C s� | j rdnd}| jrd�| j�nd}| jr d�d�| j��nd}t| jj| jf�}| j r2d�| j �nd}dj|| j
|| j ||| j |d�}| j
�d |� | jS )
zhExecute the statement to create a view.
Returns:
mysqlx.View: View object.
z OR REPLACEr � DEFINER = {0}� ({0})�, � WITH {0} CHECK OPTIONz�CREATE{replace} ALGORITHM = {algorithm}{definer} SQL SECURITY {security} VIEW {view_name}{columns} AS {defined_as}{check_option})r) r� r� r� � view_namer� r� r� r� )r� r� r( r� r0 r1 r� rm r� r� r� r� r� r< r� r� )r r) r� r� r� r� r� r r r rE ] s. ������� zCreateViewStatement.execute)F)
r r r rG r r� r� r� r� r� r� rE r{ r r rW r r� s
r� c r| )�AlterViewStatementzYA statement for alter views.
Args:
view (mysqlx.View): The View object.
c s t t| ��|� d S r )rK r� r )r r� rW r r r ~ s zAlterViewStatement.__init__c C s� | j r d�| j �nd}| jrd�d�| j��nd}t| jj| jf�}| jr+d�| j�nd}dj| j || j
||| j |d�}| j �
d|� | jS ) zgExecute the statement to alter a view.
Returns:
mysqlx.View: View object.
r� r r� r� r� zvALTER ALGORITHM = {algorithm}{definer} SQL SECURITY {security} VIEW {view_name}{columns} AS {defined_as}{check_option})r� r� r� r� r� r� r� r� )r� r( r� r0 r1 r� rm r� r� r� r� r� r< r� r� )r r� r� r� r� r� r r r rE � s* �������zAlterViewStatement.executer� r r rW r r� x r� r� c s� e Zd ZdZe�dej�Z� fdd�Ze dd� �Z
dd� Z d d
� Z d d � Z
d
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Z� ZS )%�CreateTableStatementz�A statement that creates a new table if it doesn't exist already.
Args:
collection (mysqlx.Schema): The Schema object.
table_name (string): The name for the new table.
z$(from\s+)([`\"].+[`\"]|[^\.]+)(\s|$)c st t t| ��|� d | _d | _d | _d | _d | _d| _g | _ g | _
g | _ g | _ g | _
d| _|| _d�| j�� �| _d S )NFr z
\1{0}.\2\3)rK r� r �_charset�
_collation�_comment�_as�_like�_tempr� �_f_keys�_indices�_p_keys�
_u_indices� _auto_incr� r( r@ �get_name� _tbl_repl)r r@ r4 rW r r r � s z CreateTableStatement.__init__c C s t t| jj| j��S )z7string: The fully qualified name of the Table.
)r1 r8 r@ rm r� r? r r r r4 � s
�z CreateTableStatement.table_namec C sb g }|� d� | jr|� d� | jr|� d� | jr |� d� d�|�}|j| j| j| j| jd�S )NzAUTO_INCREMENT = {inc}z!DEFAULT CHARACTER SET = {charset}z DEFAULT COLLATE = {collation}zCOMMENT = '{comment}'r\ )�inc�charset� collation�comment)rp r� r� r� r0 r( r� )r �options�
table_optsr r r �_get_table_opts� s
�z$CreateTableStatement._get_table_optsc C s� g }| j r|�d�d�| j ��� | jD ] }|�|�� � q| jD ] }|�|�� � q!| jD ] }|�|�� � q.| jD ] }|�|�� � q;d�|�S )NzPRIMARY KEY ({0})r\ ) r� rp r( r0 r� r� r� r� r� )r Zdefsr� rz �indexr r r �_get_create_def� s
z$CreateTableStatement._get_create_defc C s t t| jj|��| _| S )z�Create table with the definition of another existing Table.
Args:
table_name (string): Name of the source table.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
)r1 r8 r@ rm r� )r r4 r r r �like� s �zCreateTableStatement.likec C s8 t |t�rtj�| j|�| _| S t |t�r|�� | _| S )a. Create the Table and fill it with values from a Select Statement.
Args:
select (object): Select Statement. Can be a string or an instance of
:class`mysqlx.SelectStatement`.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
) r r r� �tbl_frmt�subr r� r� r� )r r� r r r � as_select� s
�
z CreateTableStatement.as_selectc C s |� | j�� � | j�|� | S )z�Add a Column to the Table.
Args:
column_def (MySQLx.ColumnDef): Column Definition object.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
)�
set_schemar@ r� r� rp )r Z
column_defr r r �
add_column� s z CreateTableStatement.add_columnc G s t |� }| j�|� | S )z�Add multiple Primary Keys to the Table.
Args:
*keys: Fields to be used as Primary Keys.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
)r r� r� )r rw r r r �add_primary_key s z$CreateTableStatement.add_primary_keyc G � | j �t|tjt|� �� | S )z�Adds an Index to the Table.
Args:
index_name (string): Name of the Index.
*cols: Fields to be used as an Index.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
)r� rp r� r� r �r r� �colsr r r � add_index �
�z CreateTableStatement.add_indexc G r )a Adds a Unique Index to the Table.
Args:
index_name (string): Name of the Unique Index.
*cols: Fields to be used as a Unique Index.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
)r� rp r� r� r r r r r �add_unique_index r z%CreateTableStatement.add_unique_indexc C s* |� | j�� � |�|� | j�|� | S )z�Adds a Foreign Key to the Table.
Args:
key (MySQLx.ForeignKeyDef): The Foreign Key Definition object.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
)r r@ r� �set_namer� rp )r rm rz r r r �add_foreign_key* s
z$CreateTableStatement.add_foreign_keyc C r� )z�Set the initial Auto Increment value for the table.
Args:
inc (int): The initial AUTO_INCREMENT value for the table.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
�r� )r r r r r �set_initial_auto_increment8 r� z/CreateTableStatement.set_initial_auto_incrementc C r� )z�Sets the default Charset type for the Table.
Args:
charset (string): Charset type.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
�r� �r r r r r �set_default_charsetD r� z(CreateTableStatement.set_default_charsetc C r� )z�Sets the default Collation type for the Table.
Args:
collation (string): Collation type.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
�r� �r r r r r �set_default_collationP r� z*CreateTableStatement.set_default_collationc C r� )z�Add a comment to the Table.
Args:
comment (string): Comment to be added to the Table.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
�r� �r r r r r � set_comment\ r� z CreateTableStatement.set_commentc C �
d| _ | S )zSet the Table to be Temporary.
Returns:
mysqlx.CreateTableStatement: CreateTableStatement object.
T)r� r? r r r � temporaryh � z CreateTableStatement.temporaryc C sn dj | j| jr dndd�}| jrd}nd}|j || jp | jp d| �� | �� d�}| j�d |d
� | j �
| j �S ) zYExecute the statement.
Returns:
mysqlx.Table: Table object.
zCREATE {table_type} {name}zTEMPORARY TABLEZTABLE)rm Z
table_typez{create} LIKE {query}z,{create} ({create_def}) {table_opts} {query}r )�create�queryZ
create_defr r� F) r( r4 r� r� r� r r r< r� r@ Z get_tabler� )r r' r� r r r rE q s ��z CreateTableStatement.execute)r r r rG �re�compile�
IGNORECASEr r rH r4 r r r
r
r r r r r r r r r# r% rE r{ r r rW r r� � s*
r� c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�
ColumnDefBasez�A Base class defining the basic parameters required to define a column.
Args:
name (string): Name of the column.
type (MySQLx.ColumnType): Type of the column.
size (int): Size of the column.
c C s4 d | _ d| _d| _d| _|| _|| _d| _|| _d S )NFr )�_default_schema� _not_null�_p_key�_u_indexr� �_sizer� �_type�r rm �type�sizer r r r � s
zColumnDefBase.__init__c C r$ )zwDisable NULL values for this column.
Returns:
mysqlx.ColumnDefBase: ColumnDefBase object.
T)r. r? r r r �not_null� r& zColumnDefBase.not_nullc C r$ )zwSet current column as a Unique Index.
Returns:
mysqlx.ColumnDefBase: ColumnDefBase object.
T)r0 r? r r r � unique_index� r& zColumnDefBase.unique_indexc C r� )z�Add a comment to the column.
Args:
comment (string): Comment to be added to the column.
Returns:
mysqlx.ColumnDefBase: ColumnDefBase object.
r! r" r r r r � r� zColumnDefBase.commentc C r$ )ztSets the Column as a Primary Key.
Returns:
mysqlx.ColumnDefBase: ColumnDefBase object.
T)r/ r? r r r �primary� r& zColumnDefBase.primaryc C r r �r- �r r@ r r r r � r zColumnDefBase.set_schemaN)
r r r rG r r6 r7 r r8 r r r r r r, � s
r, c s� e Zd ZdZd � fdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Z� ZS ) � ColumnDefz�Class containing the complete definition of the Column.
Args:
name (string): Name of the column.
type (MySQL.ColumnType): Type of the column.
size (int): Size of the column.
Nc sZ t t| ��|||� d | _d | _d | _d | _d| _d| _d| _ g | _
g | _ d | _ d | _
d S r� )rK r; r Z_ref�_default� _decimals�
_ref_table�_binaryr� � _unsignedr� � _ref_fieldsr� r� r3 rW r r r � s
zColumnDef.__init__c C s� d}| j r t�| j�st�| j�st�| j�r d�| j �}n#t�| j�r2| j r2d�| j | jp/d�}nt� | j�rAd�d�
| j ��}| j rId�|�}| j
rQd�|�}| jr[d�|| j�}| jred �|| j�}d
�t�| j�|�S ) Nr z({0})z
({0}, {1})r r\ z {0} UNSIGNEDz
{0} BINARYz{0} CHARACTER SET {1}z{0} COLLATE {1}�{0} {1})r1 r Z
is_numericr2 Zis_charZ is_binaryr( Z is_decimalsr= Z
is_finite_setr0 r� r@ r? r� r� Z to_string)r Ztype_defr r r �
_data_type� s(
�
�
zColumnDef._data_typec C s� | j rdnd}| jr dnd}| jrdnd}| jrdnd}d�| �� ||||�}| jr0d�|�}n| jr8d �|�}| jrR| j rRt
t | j | j��}d
�||d �
| j ��}|j| j| jd �S )
Nz NOT NULLz NULLz AUTO_INCREMENTr z DEFAULT {default}z COMMENT '{comment}'z{0}{1}{2}{3}{4}z{0} PRIMARY KEYz{0} UNIQUE KEYz{0} REFERENCES {1} ({2})r\ )�defaultr )r. r� r< r� r( rC r/ r0 r> rA r1 r8 r- r0 )r �nullZauto_incrD r �defnZ ref_tabler r r �_col_definition� s& �
�
�zColumnDef._col_definitionc C s6 t |t�r |j| _| S |du rd| _| S t|�| _| S )z�Sets the default value of this Column.
Args:
default_val (object): The default value of the Column. Can be a
string, number or :class`MySQLx.Expr`.
Returns:
mysqlx.ColumnDef: ColumnDef object.
NZNULL)r r
r r< �repr)r Z default_valr r r � set_default s
�
�zColumnDef.set_defaultc C r$ )zlSet the Column to Auto Increment.
Returns:
mysqlx.ColumnDef: ColumnDef object.
Tr r? r r r �auto_increment r& zColumnDef.auto_incrementc G � t |� | _|| _| S )z�Sets the Column as a Foreign Key.
Args:
name (string): Name of the referenced Table.
*refs: Fields this Column references.
Returns:
mysqlx.ColumnDef: ColumnDef object.
)r rA r> �r rm �refsr r r � foreign_key$ �
zColumnDef.foreign_keyc C r$ )zfSet the Column as unsigned.
Returns:
mysqlx.ColumnDef: ColumnDef object.
T)r@ r? r r r �unsigned2 r& zColumnDef.unsignedc C r� )z�Set the size of the decimal Column.
Args:
size (int): Size of the decimal.
Returns:
mysqlx.ColumnDef: ColumnDef object.
)r= )r r5 r r r �decimals; r� zColumnDef.decimalsc C r� )z�Set the Charset type of the Column.
Args:
charset (string): Charset type.
Returns:
mysqlx.ColumnDef: ColumnDef object.
r r r r r r G r� zColumnDef.charsetc C r� )z�Set the Collation type of the Column.
Args:
collation (string): Collation type.
Returns:
mysqlx.ColumnDef: ColumnDef object.
r r r r r r S r� zColumnDef.collationc C r$ )zqSet the current column to binary type.
Returns:
mysqlx.ColumnDef: ColumnDef object.
T)r? r? r r r �binary_ r& zColumnDef.binaryc G s t tt|� �| _| S )z�Set the Enum/Set values.
Args:
*values: Values for Enum/Set type Column.
Returns:
mysqlx.ColumnDef: ColumnDef object.
)�maprH r r� r� r r r r h r� zColumnDef.valuesc C s d� | j| �� �S )NrB )r( r� rG r? r r r r� t s zColumnDef.get_sqlr )r r r rG r rC rG rI rJ rN rP rQ r r rR r r� r{ r r rW r r; � s r; c s4 e Zd ZdZ� fdd�Zdd� Z� fdd�Z� ZS )�GeneratedColumnDefz�Class used to describe a Generated Column.
Args:
name: Name of the column.
col_type: Type of the column.
expr: The Expression used to generate the value of this column.
c s2 t t| ��||� t|t�sJ �d| _|j| _d S r� )rK rT r r r
�_storedr �_expr)r rm Zcol_typer rW r r r � s zGeneratedColumnDef.__init__c C r$ )z�Set the Generated Column to be stored.
Returns:
mysqlx.GeneratedColumnDef: GeneratedColumnDef object.
T)rU r? r r r �stored� r& zGeneratedColumnDef.storedc s& d� tt| ��� | j| jrd�S d�S )Nz {0} GENERATED ALWAYS AS ({1}){2}z STOREDr )r( rK rT r� rV rU r? rW r r r� � s ��zGeneratedColumnDef.get_sql)r r r rG r rW r� r{ r r rW r rT x s
rT c @ sh e Zd ZdZdZdZdZdZdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS )�
ForeignKeyDefz Class describing a Foreign Key.r rl � � c C s> g | _ g | _d | _d | _d | _| �tj�| _| �tj�| _ d S r )
r� � _f_fieldsr� �_f_tabler- �_actionrX � NO_ACTION�_update_action�_delete_actionr? r r r r � s zForeignKeyDef.__init__c C s. |t ju rdS |t ju rdS |t ju rdS dS )N�RESTRICT�CASCADEzSET NULLz NO ACTION)rX ra rb �SET_NULL�r �actionr r r r] � s
zForeignKeyDef._actionc C r r )r� )r rm r r r r � r zForeignKeyDef.set_namec C r r r9 r: r r r r � r zForeignKeyDef.set_schemac G s t |� | _| S )z�Add a list of fields in the parent table.
Args:
*fields: Fields in the given table which constitute the Foreign Key.
Returns:
mysqlx.ForeignKeyDef: ForeignKeyDef object.
)r r� r^ r r r r_ � r� zForeignKeyDef.fieldsc G rK )z�Add the child table name and the fields.
Args:
name (string): Name of the referenced table.
*refs: A list fields in the referenced table.
Returns:
mysqlx.ForeignKeyDef: ForeignKeyDef object.
)r r[ r\ rL r r r � refers_to� rO zForeignKeyDef.refers_toc C � | � |�| _| S )a� Define the action on updating a Foreign Key.
Args:
action (int): Action to be performed on updating the reference.
Can be any of the following values:
1. ForeignKeyDef.NO_ACTION
2. ForeignKeyDef.RESTRICT
3. ForeignKeyDef.CASCADE
4. ForeignKeyDef.SET_NULL
Returns:
mysqlx.ForeignKeyDef: ForeignKeyDef object.
)r] r_ rd r r r � on_update� s zForeignKeyDef.on_updatec C rg )a� Define the action on deleting a Foreign Key.
Args:
action (int): Action to be performed on updating the reference.
Can be any of the following values:
1. ForeignKeyDef.NO_ACTION
2. ForeignKeyDef.RESTRICT
3. ForeignKeyDef.CASCADE
4. ForeignKeyDef.SET_NULL
Returns:
mysqlx.ForeignKeyDef: ForeignKeyDef object.
)r] r` rd r r r � on_delete� s zForeignKeyDef.on_deletec C sV d� | j�}d� | j�}d� | jd�| j�tt| j| j ��d�| j
��}d� |||�S )Nz
ON UPDATE {0}z
ON DELETE {0}z)FOREIGN KEY {0}({1}) REFERENCES {2} ({3})r\ z {0} {1} {2}) r( r_ r` r� r0 r� r1 r8 r- r\ r[ )r r� r� rz r r r r� � s �
�zForeignKeyDef.get_sqlN)r r r rG r^ ra rb rc r r] r r r_ rf rh ri r� r r r r rX � s rX )r )2rG ru r) �errorsr r r �compatr � constantsr r Zdbdocr Zprotobufr r� �resultr
r r rx r
r r' r* r1 r8 r9 rI r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r, r; rT rX r r r r �<module> sR
" *e<?!"*w! p= 4