관리-도구
편집 파일: statement.cpython-38.pyc
U )�eO� � @ 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�ZdEdd�ZdFdd�ZdGdd�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 )Hz Implementation of Statements.� N� )�ProgrammingError)� ExprParser)� STRING_TYPES)� Algorithms� Securities)�DbDoc)�mysqlx_crud_pb2)� SqlResult�Result� ColumnTypec @ s e Zd Zdd� ZdS )�Exprc C s || _ d S �N)�expr)�selfr � r �e/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysqlx/statement.py�__init__& s z Expr.__init__N)�__name__� __module__� __qualname__r r r r r r % s r c G s* t | �dkr&t| d ttf�r&| d S | S )Nr r )�len� isinstance�list�tuple)�valuesr r r �flexible_params* s r � c C sT d|kr8| d dkr | d dkp6| d dko6| d dkS | d dkoN| d dkS dS )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 �`����"Nr �� identifier�sql_moder r r �is_quoted_identifier0 s �r% c C s> t | |�r| S d|kr(d�| �dd��S d�| �dd��S dS ) 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``N)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 �r/ c C s^ d|kr dnd}||kr"d� |�nd}|�|d�}t|�dkrB| 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 �r6 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|jnd | _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. )r8 �r r r r r; x s zStatement.targetc C s | j jS )z3:class:`mysqlx.Schema`: The Schema object. )r8 �schemar= r r r r> ~ s zStatement.schemac C s t �dS �zqExecute the statement. Raises: NotImplementedError: This method must be implemented. N��NotImplementedErrorr= r r r �execute� s zStatement.executeN)T) r r r �__doc__r �propertyr; r> rB r r r r r7 k s r7 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 s` t t| �j||d� d| _d| _d| _d| _d| _d| _d| _ i | _ g | _ |d k r\| � |� d S )N�r; r< F) �superrE 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)rI �_wherer r9 r Z _where_exprZ placeholder_name_to_positionrO )r rR r r r r rQ � s zFilterableStatement.wherec G s4 t |� }d| _d�|�| _t| j| j ��� | _| S �NT�,)r rH r. �_projection_strr r9 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)rJ � _limit_offset�_limit_row_count)r � row_count�offsetr r r �limit� s zFilterableStatement.limitc G s4 t |� }d| _d�|�| _t| j| j ��� | _| S )z�Sets the sorting criteria. Args: *sort_clauses: The expression strings defining the sort criteria. Returns: mysqlx.FilterableStatement: FilterableStatement object. TrW )r rK r. � _sort_strr r9 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 rV )r rL r. � _grouping_strr r9 Zparse_expr_listZ _groupingrY r r r � _group_by� s �z FilterableStatement._group_byc C s d| _ t|| j ��� | _d S �NT)rM r r9 r �_having�r rR r r r rf � s zFilterableStatement._havingc G sV d| _ t|�}|dkr&| �|d � n,|dkr8td��n| 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)rN r � _bind_singler rP �append)r �args�countr r r �bind� s zFilterableStatement.bindc C sP t |t�r| �t|�� n2t |t�rLt�|�}|�� D ]}| �||| � q6d S r )r r ro �strr �json�loads�keys)r �object�dict�keyr r r rk s z FilterableStatement._bind_singlec C s t �dS r? r@ r= r r r rB s zFilterableStatement.execute)TN)r )r r r rC r rQ r[ r` rb rd rf ro rk rB � __classcell__r r rS r rE � s rE c s( 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 �NFrF )rG rx r r: �_sql)r � connection�sqlrS r r r s zSqlStatement.__init__c C s | j �| j� t| j �S )zaExecute the statement. Returns: mysqlx.SqlResult: SqlResult object. )r: Zsend_sqlrz r r= r r r rB s zSqlStatement.execute�r r r rC r rB rw r r rS r rx s rx c s0 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; )rG r~ r �_values�_ids�r � collectionrS 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� rl )r r �valr r r �add4 s zAddStatement.addc C s< t | j�dkrt� S | jD ]}| j�|�� � q| j�| �S )�[Execute the statement. Returns: mysqlx.Result: Result object. r )r r� r r� rl Z ensure_idr: � send_insert)r �docr r r rB D s zAddStatement.execute)r r r rC r r� rB rw r r rS r r~ ) s r~ c @ s e Zd Zddd�Zdd� ZdS )� UpdateSpecNc C sj |t jjkr| �||� nL|| _|| _t|�dkrL|d dkrL|dd � | _t| jd��� j | _|| _ d S )Nr �$r F) � MySQLxCrud�UpdateOperation�SET� _table_set� update_type�sourcer r Zdocument_fieldr# rj )r r� r� rj r r r r T s � zUpdateSpec.__init__c C s$ t jj| _t|d��� | _|| _d S re )r� r� r� r� r Zparse_table_update_fieldr� rj )r r� rj r r r r� ` s zUpdateSpec._table_set)N)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; rR )rG r� r � _update_ops�r r� rR rS r r r n s �zModifyStatement.__init__c C s | 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� rl r� r� r� ZITEM_SET�r Zdoc_pathrj r r r �sets s �zModifyStatement.setc C s | j �ttjj||�� | S )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� rl 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 s | j �ttjj||�� | S )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� rl r� r� r� Z ARRAY_INSERT�r �fieldrj r r r � array_insert� s �z ModifyStatement.array_insertc C s | j �ttjj||�� | S )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� rl r� r� r� Z ARRAY_APPENDr� r r r � array_append� s ��z ModifyStatement.array_appendc C s | j �| �S �r� �r: �updater= r r r rB � s zModifyStatement.execute)N) r r r rC r r� r� r� r� r� rB rw r r rS 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 re )rG r� r r� rS 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[ rY r r r rZ � s zFindStatement.fieldsc G s | j |� | S )z�Sets a grouping criteria for the resultset. Args: *fields: The string expressions identifying the grouping criteria. Returns: mysqlx.FindStatement: FindStatement object. �rd rY r r r �group_by� s zFindStatement.group_byc C s | � |� | 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. �rf rg r r r �having� s zFindStatement.havingc C s | j �| �S )zaExecute the statement. Returns: mysqlx.DocResult: DocResult object. �r: �findr= r r r rB � s zFindStatement.execute)N) r r r rC r rZ r� r� rB rw r r rS 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)rG r� r r[ �r �tablerZ rS r r r s zSelectStatement.__init__c G s | j |� | S )z�Sets a grouping criteria for the resultset. Args: *fields: The fields identifying the grouping criteria. Returns: mysqlx.SelectStatement: SelectStatement object. r� rY r r r r� s zSelectStatement.group_byc C s | � |� | S )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� rg r r r r� s zSelectStatement.havingc C s | j �| �S )zaExecute the statement. Returns: mysqlx.RowResult: RowResult object. r� r= r r r rB , s zSelectStatement.executec C s� | j rd�| j�nd}| jr(d�| j�nd}| jr>d�| j�nd}| jrTd�| j�nd}| j rnd�| 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}rX �*)�selectr> r� r` rQ �groupr� �order)rI r& rU rL rc rM rf rK ra rJ r] r\ �getattrr> ri r; )r rQ r� r� Zorder_byr` �stmtr r r �get_sql4 s4 ���� �� �zSelectStatement.get_sql) r r r rC r r� r� rB r� rw r r rS r r� s r� c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�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 ry )rG r� r r �_fieldsr� r� rS 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� rl r r �r r r r r r R s zInsertStatement.valuesc C s | j �| �S r� )r: r� r= r r r rB ^ s zInsertStatement.execute)r r r rC r r rB rw r r rS r r� F s r� c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )�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 ry )rG r� r r� r� rS r r r n s zUpdateStatement.__init__c C s | j �ttjj||�� | S )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� rl r� r� r� r� r� r r r r� r s �zUpdateStatement.setc C s | j �| �S )zZExecute the statement. Returns: mysqlx.Result: Result object r� r= r r r rB � s zUpdateStatement.execute)r r r rC r r� rB rw r r rS r r� g s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�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 )rG r� r r� rS r r r � s zRemoveStatement.__init__c C s | j �| �S r� �r: �deleter= r r r rB � s zRemoveStatement.executer} r r rS r r� � s 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; rR r< )rG r� r )r r� rR rS r r r � s �zDeleteStatement.__init__c C s | j �| �S r� r� r= r r r rB � s zDeleteStatement.execute)Nr} r r rS r r� � s r� c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )� 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 )rG r� r � _index_name� _is_uniquer� )r r� � index_nameZ is_uniquerS 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� rl )r Z document_pathZ column_typeZ is_requiredr r r r� � s z$CreateCollectionIndexStatement.fieldc C s< dd� | j D �}| jjddd| jjj| jj| j| jf|�� S )r� c S s g | ]}|D ]}|�q qS r r )r+ Zsublist�itemr r r r- � s z:CreateCollectionIndexStatement.execute.<locals>.<listcomp>�xpluginZcreate_collection_indexT)r� r: �execute_nonqueryr8 r> ri r� r� rY r r r rB � s ��z&CreateCollectionIndexStatement.execute)r r r rC r r� rB rw r r rS r r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )� 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 )rG r� r r� )r r� r� rS r r r � s z%DropCollectionIndexStatement.__init__c C s" | j �ddd| jjj| jj| j�S )r� r� Zdrop_collection_indexT)r: r� r8 r> ri r� r= r r r rB � s �z$DropCollectionIndexStatement.executer} r r rS r r� � s r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )� TableIndexr rh c C s || _ || _|| _d S r )�_name� _index_type�_columns)r ri Z index_type�columnsr r r r � s zTableIndex.__init__c C s6 d}| j tjkr|d7 }|d7 }|�| jd�| j��S )Nr zUNIQUE zINDEX {0} ({1})rW )r� r� � UNIQUE_INDEXr& r� r. 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 ry )rG r� r �_viewr> �_schemari r� �_replacer� r � UNDEFINED� _algorithmr ZDEFINER� _security�_definer� _defined_as� _check_option)r �viewr' rS 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� s zCreateViewStatement.columnsc C s || _ | S )z�Sets the algorithm. Args: mysqlx.constants.ALGORITHMS: The algorithm. Returns: mysqlx.CreateViewStatement: CreateViewStatement object. )r� )r � algorithmr r r r� ! s z CreateViewStatement.algorithmc C s || _ | S )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� - s z CreateViewStatement.securityc C s || _ | S )z�Sets the definer. Args: definer (string): The definer. Returns: mysqlx.CreateViewStatement: CreateViewStatement object. )r� )r �definerr r r r� 9 s zCreateViewStatement.definerc C s || _ | S )z�Sets the SelectStatement statement for describing the view. Args: mysqlx.SelectStatement: SelectStatement object. Returns: mysqlx.CreateViewStatement: CreateViewStatement object. )r� )r Z statementr r r � defined_asE s z CreateViewStatement.defined_asc C s || _ | S )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 s z%CreateViewStatement.with_check_optionc C s� | j r dnd}| jr d�| j�nd}| jr<d�d�| j��nd}t| jj| jf�}| j rdd�| 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� r. r/ r� ri r� r� r� r� r� r: r� r� )r r' r� r� r� r� r| r r r rB ] s4 ������ � zCreateViewStatement.execute)F) r r r rC r r� r� r� r� r� r� rB rw r r rS r r� s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�AlterViewStatementzYA statement for alter views. Args: view (mysqlx.View): The View object. c s t t| ��|� d S r )rG r� r )r r� rS r r r ~ s zAlterViewStatement.__init__c C s� | j rd�| j �nd}| jr.d�d�| j��nd}t| jj| jf�}| jrVd�| 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� r. r/ r� ri r� r� r� r� r� r: r� r� )r r� r� r� r� r| r r r rB � s0 ������ �zAlterViewStatement.executer} r r rS r r� x s 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)rG 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> r2 rS r r r � s z CreateTableStatement.__init__c C s t t| jj| j��S )z7string: The fully qualified name of the Table. )r/ r6 r> ri r� r= r r r r2 � 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}'rW )�inc�charset� collation�comment)rl r� r� r� r. 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 ]}|�|�� � qB| jD ]}|�|�� � q\| jD ]}|�|�� � qvd�|�S )NzPRIMARY KEY ({0})rW ) r� rl r& r. r� r� r� r� r� )r Zdefsr� rv �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. )r/ r6 r> ri r� )r r2 r r r �like� s �zCreateTableStatement.likec C s6 t |t�r tj�| j|�| _nt |t�r2|�� | _| 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� rl )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 rs r r r �add_primary_key s z$CreateTableStatement.add_primary_keyc G s | 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� rl r� r� r �r r� �colsr r r � add_index s �z CreateTableStatement.add_indexc G s | j �t|tjt|� �� | S )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� rl r� r� r r r r r �add_unique_index s �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� rl )r ri rv r r r �add_foreign_key* s z$CreateTableStatement.add_foreign_keyc C s || _ | S )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 s z/CreateTableStatement.set_initial_auto_incrementc C s || _ | S )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 s z(CreateTableStatement.set_default_charsetc C s || _ | S )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 s z*CreateTableStatement.set_default_collationc C s || _ | S )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\ s z CreateTableStatement.set_commentc C s d| _ | S )zSet the Table to be Temporary. Returns: mysqlx.CreateTableStatement: CreateTableStatement object. T)r� r= r r r � temporaryh s z CreateTableStatement.temporaryc C sn dj | j| jrdndd�}| jr&d}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)ri Z table_typez{create} LIKE {query}z,{create} ({create_def}) {table_opts} {query}r )�create�queryZ create_defr� r| F) r& r2 r� r� r� r� r� r: r� r> Z get_tabler� )r r r� r r r rB q s ��z CreateTableStatement.execute)r r r rC �re�compile� IGNORECASEr� r rD r2 r� r� r� r� r� r� r r r r r r r r rB rw r r rS 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 ri �type�sizer r r r � s zColumnDefBase.__init__c C s d| _ | S )zwDisable NULL values for this column. Returns: mysqlx.ColumnDefBase: ColumnDefBase object. T)r r= r r r �not_null� s zColumnDefBase.not_nullc C s d| _ | S )zwSet current column as a Unique Index. Returns: mysqlx.ColumnDefBase: ColumnDefBase object. T)r r= r r r � unique_index� s zColumnDefBase.unique_indexc C s || _ | S )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� � s zColumnDefBase.commentc C s d| _ | S )ztSets the Column as a Primary Key. Returns: mysqlx.ColumnDefBase: ColumnDefBase object. T)r r= r r r �primary� s zColumnDefBase.primaryc C s || _ d S r �r �r r> r r r r� � s zColumnDefBase.set_schemaN) r r r rC r r! r" r� r# 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� )rG r& r Z_ref�_default� _decimals� _ref_table�_binaryr� � _unsignedr� � _ref_fieldsr� r� r rS r r r � s zColumnDef.__init__c C s� d}| j r<t�| j�s.t�| j�s.t�| j�r<d�| j �}nFt�| j�rd| j rdd�| j | jp^d�}n t� | j�r�d�d� | j ��}| j r�d�|�}| j r�d�|�}| jr�d�|| j�}| jr�d �|| j�}d �t�| j�|�S ) Nr z({0})z ({0}, {1})r rW z {0} UNSIGNEDz {0} BINARYz{0} CHARACTER SET {1}z{0} COLLATE {1}�{0} {1})r r Z is_numericr �is_charZ is_binaryr& Z is_decimalsr( Z is_finite_setr. r� r+ r* r� r� Z to_string)r Ztype_defr r r � _data_type� s( � � zColumnDef._data_typec C s� | j r dnd}| jrdnd}| jr&dnd}| jr4dnd}d�| �� ||||�}| jr`d�|�}n| jrpd �|�}| jr�| j r�t 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})rW )�defaultr� )r r� r' r� r& r/ r r r) r, r/ r6 r r. )r �nullZauto_incr0 r� ZdefnZ ref_tabler r r �_col_definition� s* � � �zColumnDef._col_definitionc C s2 t |t�r|j| _n|dkr$d| _n 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 s d| _ | S )zlSet the Column to Auto Increment. Returns: mysqlx.ColumnDef: ColumnDef object. Tr r= r r r �auto_increment s zColumnDef.auto_incrementc G s 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 r, r) �r ri �refsr r r � foreign_key$ s zColumnDef.foreign_keyc C s d| _ | S )zfSet the Column as unsigned. Returns: mysqlx.ColumnDef: ColumnDef object. T)r+ r= r r r �unsigned2 s zColumnDef.unsignedc C s || _ | S )z�Set the size of the decimal Column. Args: size (int): Size of the decimal. Returns: mysqlx.ColumnDef: ColumnDef object. )r( )r r r r r �decimals; s zColumnDef.decimalsc C s || _ | S )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 s zColumnDef.charsetc C s || _ | S )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 s zColumnDef.collationc C s d| _ | S )zqSet the current column to binary type. Returns: mysqlx.ColumnDef: ColumnDef object. T)r* r= r r r �binary_ s 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. )�mapr3 r r� r� r r r r h s zColumnDef.valuesc C s d� | j| �� �S )Nr- )r& r� r2 r= r r r r� t s zColumnDef.get_sql)N)r r r rC r r/ r2 r4 r5 r8 r9 r: r� r� r; r r� rw r r rS 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�s t�d| _|j| _d S r� ) rG r= r r r �AssertionError�_storedr �_expr)r ri Zcol_typer rS r r r � s zGeneratedColumnDef.__init__c C s d| _ | S )z�Set the Generated Column to be stored. Returns: mysqlx.GeneratedColumnDef: GeneratedColumnDef object. T)r? r= r r r �stored� s zGeneratedColumnDef.storedc s$ d� tt| ��� | j| jr dnd�S )Nz {0} GENERATED ALWAYS AS ({1}){2}z STOREDr )r& rG r= r� r@ r? r= rS r r r� � s �zGeneratedColumnDef.get_sql)r r r rC r rA r� rw r r rS r r= x s r= 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 rh � � c C s> g | _ g | _d | _d | _d | _| �tj�| _| �tj�| _ d S r ) r� � _f_fieldsr� �_f_tabler �_actionrB � NO_ACTION�_update_action�_delete_actionr= r r r r � s zForeignKeyDef.__init__c C s. |t jkrdS |t jkr dS |t jkr*dS dS )N�RESTRICT�CASCADEzSET NULLz NO ACTION)rB rK rL �SET_NULL�r �actionr r r rG � s zForeignKeyDef._actionc C s || _ d S r )r� )r ri r r r r � s zForeignKeyDef.set_namec C s || _ d S r r$ r% r r r r� � s 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� rY r r r rZ � s zForeignKeyDef.fieldsc G s t |� | _|| _| S )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 rE rF r6 r r r � refers_to� s zForeignKeyDef.refers_toc C s | � |�| _| 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. )rG rI rN r r r � on_update� s zForeignKeyDef.on_updatec C s | � |�| _| S )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. )rG rJ rN 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})rW z {0} {1} {2}) r& rI rJ r� r. r� r/ r6 r rF rE )r r� r� rv r r r r� � s � �zForeignKeyDef.get_sqlN)r r r rC rH rK rL rM r rG r r� rZ rP rQ rR r� r r r r rB � s rB )r )r )r )r )2rC rq r �errorsr r r �compatr � constantsr r Zdbdocr �protobufr r� �resultr r r rt r r r% r( r/ r6 r7 rE rx r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r& r= rB r r r r �<module> sP " *e<?!"*w! p= 4