U 4Jec@s4dZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZmZmZmZdd l m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&dZ'GdddZ(dddddZ)dDddddddZ*dEddddddZ+dFddddd d!Z,dGdddd"d#d$d%Z-Gd&d'd'Z.Gd(d)d)e.Z/Gd*d+d+e.Z0Gd,d-d-e.Z1Gd.d/d/e1Z2Gd0d1d1Z3Gd2d3d3e/Z4Gd4d5d5e/Z5Gd6d7d7e5Z6Gd8d9d9e5Z7Gd:d;d;e1Z8Gdd?d?e/Z:Gd@dAdAe/Z;GdBdCdCe.Zzsz.quote_multipart_identifier..)join)r=r4r"r@r#quote_multipart_identifieros rCzTuple[str, str])default_schema table_namer4rcCs^d|kr dnd}||kr"d|nd}||d}t|dkrB|n |d||d|fS)zParse table name. Args: default_schema (str): The default schema. table_name (str): The table name. sql_mode(Optional[str]): The SQL mode. Returns: str: The parsed table name. r5r8r6r>r rr7)splitr+strip)rDrEr4quote delimitertempr"r"r#parse_table_name~s   rKc@sDeZdZdZd+ddddddZedd d d Zed d d dZedd ddZej dddddZedd ddZ edd ddZ e j dddddZ edd ddZ e j dddddZ edd ddZ edd ddZej dddd dZdd d!d"Zdd d#d$Zdd d%d&Zd'd d(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. Trr2rtarget doc_basedrcCs@||_||_|r|nd|_d|_d|_d|_d|_d|_dS)NrTF) _target _doc_basedget_connection _connection_stmt_id _exec_counter_changed _prepared_deallocate_prepare_executer!rNrOr"r"r#r$szStatement.__init__rcCs|jS)z#object: The database object target.)rPr!r"r"r#rNszStatement.targetrcCs|jjS)z*:class:`mysqlx.Schema`: The Schema object.)rPschemar[r"r"r#r\szStatement.schemaintcCs|jS)zXReturns this statement ID. Returns: int: The statement ID. rTr[r"r"r#stmt_idszStatement.stmt_idvaluercCs ||_dSr r^r!rar"r"r#r_scCs|jS)z5int: The number of times this statement was executed.rUr[r"r"r# exec_counterszStatement.exec_countercCs|jS)z+bool: `True` if this statement has changes.rVr[r"r"r#changedszStatement.changedcCs ||_dSr rerbr"r"r#rfscCs|jS)z1bool: `True` if this statement has been prepared.rWr[r"r"r#preparedszStatement.preparedcCs ||_dSr rgrbr"r"r#rhscCs |jdkS)z;bool: `True` if this statement was executed more than once.r rcr[r"r"r#repeatedszStatement.repeatedcCs|jS)z9bool: `True` to deallocate + prepare + execute statement.rXr[r"r"r#deallocate_prepare_executesz$Statement.deallocate_prepare_executecCs ||_dSr rjrbr"r"r#rkscCs|jS)zkCheck if it is document based. Returns: bool: `True` if it is document based. )rQr[r"r"r# is_doc_basedszStatement.is_doc_basedcCs|jd7_dS)z@Increments the number of times this statement has been executed.r Nrcr[r"r"r#increment_exec_countersz Statement.increment_exec_countercCs d|_dS)zd?d@ZdAddBdCdDZdAddEdFZZS)IFilterableStatementaA 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. TNrr2 Optional[str]rrNrO conditionrcstj||di|_i|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_d|_d|_d|_d|_|r||dS)NrNrOr0rF)superr$ _binding_map _bindings_having _grouping_str _grouping _limit_offset_limit_row_count_projection_str_projection_expr _sort_str _sort_expr _where_str _where_expr has_bindings has_limit has_group_by has_havinghas_projectionhas_sort has_where _set_wherer!rNrOrx __class__r"r#r$ s4zFilterableStatement.__init__z!Union[DbDoc, Dict[str, Any], str])objrc Cst|tr|t|nt|tr8||n|t|trzt|}t|ts\tWn,tk r}zt d|W5d}~XYnX| D]}||||qnt ddS)aBind single object. Args: obj (:class:`mysqlx.DbDoc` or str): DbDoc or JSON string object. Raises: :class:`mysqlx.ProgrammingError`: If invalid JSON string to bind. ValueError: If JSON loaded is not a dictionary. zInvalid JSON string to bindNz%Invalid JSON string or object to bind) r,dictbindr Zas_strr1jsonloads ValueErrorrkeys)r!rreserrkeyr"r"r# _bind_single.s      z FilterableStatement._bind_singler1clausesrcGs6d|_dt||_t|j|j |_d|_|S)zSets the sorting criteria. Args: *clauses: The expression strings defining the sort criteria. Returns: mysqlx.FilterableStatement: FilterableStatement object. T,) rrBr/rrrQZparse_order_specrrVr!rr"r"r#_sortHs  zFilterableStatement._sortrxrc Cshd|_||_zt||j }||_Wn,tk rT}ztd|W5d}~XYnX|j|_ d|_ |S)aSets the search condition to filter. Args: condition (str): Sets the search condition to filter documents or records. Returns: mysqlx.FilterableStatement: FilterableStatement object. TzInvalid conditionN) rrrrQrrrrZplaceholder_name_to_positionr{rV)r!rxrrr"r"r#rYs zFilterableStatement._set_wherefieldsrcGs:t|}d|_d||_t|j|j |_d|_dS)zJSet group by. Args: *fields: List of fields. TrN) r/rrBr~rrQZparse_expr_listrrVr!rr"r"r# _set_group_byns  z!FilterableStatement._set_group_bycCs$d|_t||j |_d|_dS)zOSet having. Args: condition (str): The condition. TN)rrrQrr}rVr!rxr"r"r# _set_having|szFilterableStatement._set_havingcGs:t|}d|_d||_t|j|j |_d|_|S)zSet the projection. Args: *fields: List of fields. Returns: :class:`mysqlx.FilterableStatement`: Returns self. Tr) r/rrBrrrQZparse_table_select_projectionrrVrr"r"r#_set_projections   z#FilterableStatement._set_projectionDict[str, Any]rZcCs|jSzlReturns the binding map dictionary. Returns: dict: The binding map dictionary. r{r[r"r"r#get_binding_mapsz#FilterableStatement.get_binding_mapzUnion[Dict[str, Any], List]cCs|jSz\Returns the bindings list. Returns: `list`: The bindings list. r|r[r"r"r# get_bindingssz FilterableStatement.get_bindingsz9List[Union[ProtobufMessageType, ProtobufMessageCextType]]cCs|jS)zrReturns the grouping expression list. Returns: `list`: The grouping expression list. )rr[r"r"r# get_groupingsz FilterableStatement.get_groupingrcCs|jS)zdReturns the having expression. Returns: object: The having expression. )r}r[r"r"r# get_havingszFilterableStatement.get_havingr]cCs|jS)z]Returns the limit row count. Returns: int: The limit row count. )rr[r"r"r#get_limit_row_countsz'FilterableStatement.get_limit_row_countcCs|jS)zWReturns the limit offset. Returns: int: The limit offset. )rr[r"r"r#get_limit_offsetsz$FilterableStatement.get_limit_offsetcCs|jS)zbReturns the where expression. Returns: object: The where expression. )rr[r"r"r#get_where_exprsz"FilterableStatement.get_where_exprcCs|jS)zlReturns the projection expression. Returns: object: The projection expression. )rr[r"r"r#get_projection_exprsz'FilterableStatement.get_projection_exprcCs|jS)z`Returns the sort expression. Returns: object: The sort expression. )rr[r"r"r# get_sort_exprsz!FilterableStatement.get_sort_expr8.0.12cCs ||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. .. deprecated:: 8.0.12 rrr"r"r#wheres zFilterableStatement.wherecGs |j|S)zSets the sorting criteria. Args: *clauses: The expression strings defining the sort criteria. Returns: mysqlx.FilterableStatement: FilterableStatement object. .. deprecated:: 8.0.12 rrr"r"r#sorts zFilterableStatement.sortz Optional[int]) row_countoffsetrcCsnt|tr|dkrtd|jsBt|jdk|_t|jdk |_||_d|_|rj| |t j dt d|S)aSets the maximum number of items to be returned. Args: row_count (int): The maximum number of items. Returns: mysqlx.FilterableStatement: FilterableStatement object. Raises: ValueError: If ``row_count`` is not a positive integer. .. versionchanged:: 8.0.12 The usage of ``offset`` was deprecated. rz0The 'row_count' value must be a positive integerTzh'limit(row_count, offset)' is deprecated, please use 'offset(offset)' to set the number of items to skip)category) r,r]rrr2rUrVrXrrwarningswarnDeprecationWarning)r!rrr"r"r#limits zFilterableStatement.limit)rrcCs$t|tr|dkrtd||_|S)a:Sets the number of items to skip. Args: offset (int): The number of items to skip. Returns: mysqlx.FilterableStatement: FilterableStatement object. Raises: ValueError: If ``offset`` is not a positive integer. .. versionadded:: 8.0.12 rz-The 'offset' value must be a positive integer)r,r]rr)r!rr"r"r#r$szFilterableStatement.offsetrargsrcGsNd|_t|}|dkr&||dn$|dkrB|d|j|d<ntd|S)aBinds value(s) to a specific placeholder(s). 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#Invalid number of arguments to bind)rr+rr|r)r!rcountr"r"r#r7szFilterableStatement.bindcCstdSrorpr[r"r"r#rrOszFilterableStatement.execute)TN)N)r%r&r'r(r$rrrrrrrrrrrrrrrrrrrrrrr __classcell__r"r"rr#rus8$   #rucsveZdZdZddddfdd Zeddd d Zd dd d ZddddZdddddZ ddddZ Z S) SqlStatementzA statement for SQL execution. Args: connection (mysqlx.connection.Connection): Connection object. sql (string): The sql statement to be executed. rr1r) connectionsqlrcs8tjddd||_||_d|_g|_d|_d|_dSNFry)rzr$rS_sqlr{r|rr)r!rrrr"r#r$`szSqlStatement.__init__rZcCs|jS)zstring: The SQL text statement.)rr[r"r"r#riszSqlStatement.sqlrcCs|jSrrr[r"r"r#rnszSqlStatement.get_binding_mapzUnion[Tuple, List]cCs|jSrrr[r"r"r#rvszSqlStatement.get_bindingsrrcGsHt|dkrtdd|_t|}t|ttfr8||_n |j||S)zBinds value(s) to a specific placeholder(s). Args: *args: The value(s) to bind. Returns: mysqlx.SqlStatement: SqlStatement object. rrT) r+rrr/r,r-r.r|append)r!rZbindingsr"r"r#r~s  zSqlStatement.bindrcCs |j|S)zaExecute the statement. Returns: mysqlx.SqlResult: SqlResult object. )rSZsend_sqlr[r"r"r#rrszSqlStatement.execute) r%r&r'r(r$rsrrrrrrrr"r"rr#rXs rcsFeZdZdZddddfdd Zdd d d Zd d d dZZS)WriteStatementz*Provide common write operation attributes.rr2rrMcst||g|_dSr )rzr$_valuesrYrr"r#r$s zWriteStatement.__init__zpList[Union[int, str, DbDoc, Dict[str, Any], List[Optional[Union[str, int, float, ExprParser, Dict[str, Any]]]]]]rZcCs|jS)z^Returns the list of values. Returns: `list`: The list of values. )rr[r"r"r# get_valuesszWriteStatement.get_valuesrcCstdSrorpr[r"r"r#rrszWriteStatement.execute)r%r&r'r(r$rrrrr"r"rr#rs rcsfeZdZdZdddfdd Zddd d Zdddd d dZdddddZddddZZ S) AddStatementzA statement for document addition on a collection. Args: collection (mysqlx.Collection): The Collection object. rr) collectionrcst|dd|_g|_dS)NTF)rzr$_upsertids)r!rrr"r#r$szAddStatement.__init__r2rZcCs|jS)zhReturns `True` if it's an upsert. Returns: bool: `True` if it's an upsert. rr[r"r"r# is_upsertszAddStatement.is_upsertTr`cCs ||_|S)zSets the upset flag to the boolean of the value provided. Setting of this flag allows updating of the matched rows/documents with the provided value. Args: value (optional[bool]): Set or unset the upsert flag. rrbr"r"r#upsertszAddStatement.upsertr r)cGs:t|D],}t|tr$|j|q|jt|q|S)zAdds 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 rr)r!r*valr"r"r#adds  zAddStatement.addrcCs t|jdkrtS|j|S)[Execute the statement. Returns: mysqlx.Result: Result object. r)r+rrrS send_insertr[r"r"r#rrszAddStatement.execute)T) r%r&r'r(r$rrrrrrr"r"rr#rs  rc@s8eZdZdZd ddddddd Zdddd d d ZdS) UpdateSpeczUpdate specification class implementation. Args: update_type (int): The update type. source (str): The source. value (Optional[str]): The value. Raises: ProgrammingError: If `source` is invalid. Nr]r1rr) update_typesourcerarc Csp|tdkr|||nR||_zt|dj|_Wn.tk rd}zt||W5d}~XYnX||_ dS)N*Mysqlx.Crud.UpdateOperation.UpdateType.SETF) r _table_setrrZdocument_fieldr3rrrra)r!rrrarr"r"r#r$ s zUpdateSpec.__init__)rrarcCs$td|_t|d|_||_dS)zlTable set. Args: source (str): The source. value (str): The value. rTN)rrrZparse_table_update_fieldrra)r!rrar"r"r#rs zUpdateSpec._table_set)N)r%r&r'r(r$rr"r"r"r#rs  rcseZdZdZddddfdd Zdddd d Zd d d dZddddddZedddddddZ dddddZ ddddddZ ddddddZ dddd d!Z d"d d#d$ZZS)%ModifyStatementaUA statement for document update operations on a Collection. Args: collection (mysqlx.Collection): The Collection object. condition (str): Sets the search condition to identify the documents to be modified. .. versionchanged:: 8.0.12 The ``condition`` parameter is now mandatory. rr1rrrxrcstj||di|_dSN)rNrxrzr$ _update_opsr!rrxrr"r#r$.szModifyStatement.__init__rcGs |j|S)zSets the sorting criteria. Args: *clauses: The expression strings defining the sort criteria. Returns: mysqlx.ModifyStatement: ModifyStatement object. rrr"r"r#r2s zModifyStatement.sortrrZcCs|jSztReturns the list of update operations. Returns: `list`: The list of update operations. rr[r"r"r#get_update_ops=szModifyStatement.get_update_opsr)doc_pathrarcCs ttd|||j|<d|_|S)a2Sets 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. z/Mysqlx.Crud.UpdateOperation.UpdateType.ITEM_SETTrrrrVr!rrar"r"r#setEs  zModifyStatement.setrcCs ttd|||j|<d|_|S)aAdd 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. .. deprecated:: 8.0.12 z3Mysqlx.Crud.UpdateOperation.UpdateType.ITEM_REPLACETrrr"r"r#changeWs zModifyStatement.change) doc_pathsrcGs,t|D]}ttd||j|<qd|_|S)aRemoves attributes from documents in a collection. Args: doc_paths (list): The list of document paths of the attributes to be removed. Returns: mysqlx.ModifyStatement: ModifyStatement object. z2Mysqlx.Crud.UpdateOperation.UpdateType.ITEM_REMOVET)r/rrrrV)r!ritemr"r"r#unsetms  zModifyStatement.unsetfieldrarcCs ttd|||j|<d|_|S)aInsert 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. z3Mysqlx.Crud.UpdateOperation.UpdateType.ARRAY_INSERTTrr!rrar"r"r# array_inserts  zModifyStatement.array_insertcCs ttd|||j|<d|_|S)aInserts 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. z3Mysqlx.Crud.UpdateOperation.UpdateType.ARRAY_APPENDTrrr"r"r# array_appends  zModifyStatement.array_appendz#Union[Dict, DbDoc, ExprParser, str])docrcCsX|dkr d}t|ttttfs&tdttddt|trB|n||j d<d|_ |S)aTakes a :class:`mysqlx.DbDoc`, string JSON format or a dict with the changes and applies it on all matching documents. Args: doc (object): A generic document (DbDoc), string in JSON format or dict, with the changes to apply to the matching documents. Returns: mysqlx.ModifyStatement: ModifyStatement object. Nr0z>Invalid data for update operation on document collection tablez2Mysqlx.Crud.UpdateOperation.UpdateType.MERGE_PATCH$patchT) r,rrr r1rrrrrrV)r!rr"r"r#rs  zModifyStatement.patchrcCs|jstd|j|S)Execute the statement. Returns: mysqlx.Result: Result object. Raises: ProgrammingError: If condition was not set. z!No condition was found for modifyrrrSZ send_updater[r"r"r#rrs zModifyStatement.execute)r%r&r'r(r$rrrrrrrrrrrrr"r"rr#r"s  rcseZdZdZd$dddddfd d Zed d d dZd ddddZdd ddZdd ddZ e j fd ddddZ e j fd ddddZ dddddZddddd Zd!d d"d#ZZS)% ReadStatementaProvide base functionality for Read operations 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. TNrr2rvrrwcs(t|||d|_d|_tj|_dSNF)rzr$_lock_exclusive _lock_sharedr DEFAULT_lock_contentionrrr"r#r$szReadStatement.__init__r rZcCs|jS)z::class:`mysqlx.LockContention`: The lock contention value.)rr[r"r"r#lock_contentionszReadStatement.lock_contention)rrc CsFzt|j}Wn,tk r:}ztd|W5d}~XYnX||_dS)zSet the lock contention. Args: lock_contention (:class:`mysqlx.LockContention`): Lock contention. Raises: ProgrammingError: If is an invalid lock contention value. z;Invalid lock contention mode. Use 'NOWAIT' or 'SKIP_LOCKED'N)r rarrr)r!r_rr"r"r#_set_lock_contentions z"ReadStatement._set_lock_contentioncCs|jS)zrReturns `True` if is `EXCLUSIVE LOCK`. Returns: bool: `True` if is `EXCLUSIVE LOCK`. )rr[r"r"r#is_lock_exclusiveszReadStatement.is_lock_exclusivecCs|jS)zlReturns `True` if is `SHARED LOCK`. Returns: bool: `True` if is `SHARED LOCK`. )rr[r"r"r#is_lock_sharedszReadStatement.is_lock_sharedcCsd|_d|_|||S)zExecute a read operation with `SHARED LOCK`. Only one lock can be active at a time. Args: lock_contention (:class:`mysqlx.LockContention`): Lock contention. FTrrrr!rr"r"r# lock_shareds  zReadStatement.lock_sharedcCsd|_d|_|||S)zExecute a read operation with `EXCLUSIVE LOCK`. Only one lock can be active at a time. Args: lock_contention (:class:`mysqlx.LockContention`): Lock contention. TFr r r"r"r#lock_exclusives  zReadStatement.lock_exclusiver1rcGs|j||S)zSets a grouping criteria for the resultset. Args: *fields: The string expressions identifying the grouping criteria. Returns: mysqlx.ReadStatement: ReadStatement object. )rrr"r"r#group_by*s zReadStatement.group_byrcCs|||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.ReadStatement: ReadStatement object. )rrr"r"r#having6s zReadStatement.havingzUnion[DocResult, RowResult]cCs |j|Sr)rSZ send_findr[r"r"r#rrDszReadStatement.execute)TN)r%r&r'r(r$rsrrrr r rr r rrrrrr"r"rr#rs    rcsLeZdZdZdddddfdd Zd dd d d Zd dd ddZZS) FindStatementaA 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. Nrrvrrcst|d|dS)NTrzr$rrr"r#r$XszFindStatement.__init__r1rcGs |j|S)zSets a document field filter. Args: *fields: The string expressions identifying the fields to be extracted. Returns: mysqlx.FindStatement: FindStatement object. )rrr"r"r#r]s zFindStatement.fieldsrcGs |j|SzSets the sorting criteria. Args: *clauses: The expression strings defining the sort criteria. Returns: mysqlx.FindStatement: FindStatement object. rrr"r"r#ris zFindStatement.sort)N)r%r&r'r(r$rrrr"r"rr#rMs   rcsXeZdZdZddddfdd Zdddd d Zddd d d ZddddZZS)SelectStatementzA statement for record retrieval operations on a Table. Args: table (mysqlx.Table): The Table object. *fields: The fields to be retrieved. rr1rtablerrcst|d|j|dSr)rzr$rr!rrrr"r#r$}szSelectStatement.__init__rcCs ||S)zSets the search condition to filter. Args: condition (str): Sets the search condition to filter records. Returns: mysqlx.SelectStatement: SelectStatement object. rrr"r"r#rs zSelectStatement.wherercGs |j|S)zSets the order by criteria. Args: *clauses: The expression strings defining the order by criteria. Returns: mysqlx.SelectStatement: SelectStatement object. rrr"r"r#order_bys zSelectStatement.order_byrZc Cs|jrd|jnd}|jr(d|jnd}|jr>d|jnd}|jrTd|jnd}|jrrd|j d|j nd}d|j pd d |j j d |jj ||||| }|S) zYReturns the generated SQL. Returns: str: The generated SQL. z WHERE r0z GROUP BY z HAVING z ORDER BY z LIMIT z OFFSET zSELECT *z FROM r>)rrrr~rr}rrrrrrr\namerN)r!rrrrrZstmtr"r"r#get_sqls6zSelectStatement.get_sql) r%r&r'r(r$rrrrr"r"rr#rus   rcsHeZdZdZddddfdd Zdddd d Zd d d dZZS)InsertStatementzA statement for insert operations on Table. Args: table (mysqlx.Table): The Table object. *fields: The fields to be inserted. rrrrcst|dt||_dSr)rzr$r/_fieldsrrr"r#r$szInsertStatement.__init__r)cGs|jtt||S)zSet the values to be inserted. Args: *values: The values of the columns to be inserted. Returns: mysqlx.InsertStatement: InsertStatement object. )rrr-r/)r!r*r"r"r#r*s zInsertStatement.valuesrrZcCs |j|Sr)rSrr[r"r"r#rrszInsertStatement.execute)r%r&r'r(r$r*rrrr"r"rr#rs rcsveZdZdZdddfdd Zdddd d Zddd d d ZddddZddddddZddddZ Z S)UpdateStatementzA statement for record update operations on a Table. Args: table (mysqlx.Table): The Table object. .. versionchanged:: 8.0.12 The ``fields`` parameters were removed. rrrrcstj|ddi|_dSrrr!rrr"r#r$szUpdateStatement.__init__r1rcCs ||S)zSets the search condition to filter. Args: condition (str): Sets the search condition to filter records. Returns: mysqlx.UpdateStatement: UpdateStatement object. rrr"r"r#rs zUpdateStatement.wherercGs |j|S)zSets the order by criteria. Args: *clauses: The expression strings defining the order by criteria. Returns: mysqlx.UpdateStatement: UpdateStatement object. rrr"r"r#rs zUpdateStatement.order_byrrZcCs|jSrrr[r"r"r#rszUpdateStatement.get_update_opsrrcCs ttd|||j|<d|_|S)aUpdates 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. rTrrr"r"r#rs  zUpdateStatement.setrcCs|jstd|j|S)zExecute the statement. Returns: mysqlx.Result: Result object Raises: ProgrammingError: If condition was not set. z!No condition was found for updaterr[r"r"r#rr s zUpdateStatement.execute) r%r&r'r(r$rrrrrrrr"r"rr#rs   rcsHeZdZdZddddfdd Zdddd d Zd d d dZZS)RemoveStatementaEA statement for document removal from a collection. Args: collection (mysqlx.Collection): The Collection object. condition (str): Sets the search condition to identify the documents to be removed. .. versionchanged:: 8.0.12 The ``condition`` parameter was added. rr1rrcstj||ddSrrrrr"r#r$'szRemoveStatement.__init__rcGs |j|Srrrr"r"r#r*s zRemoveStatement.sortrrZcCs|jstd|j|S)rz!No condition was found for removerrrSZ send_deleter[r"r"r#rr5s zRemoveStatement.execute)r%r&r'r(r$rrrrr"r"rr#r!s  r!csVeZdZdZdddfdd Zdddd d Zddd d d ZddddZZS)DeleteStatementzA statement that drops a table. Args: table (mysqlx.Table): The Table object. .. versionchanged:: 8.0.12 The ``condition`` parameter was removed. rrrcstj|dddSrrr rr"r#r$MszDeleteStatement.__init__r1rcCs ||S)zSets the search condition to filter. Args: condition (str): Sets the search condition to filter records. Returns: mysqlx.DeleteStatement: DeleteStatement object. rrr"r"r#rPs zDeleteStatement.wherercGs |j|S)zSets the order by criteria. Args: *clauses: The expression strings defining the order by criteria. Returns: mysqlx.DeleteStatement: DeleteStatement object. rrr"r"r#r[s zDeleteStatement.order_byrrZcCs|jstd|j|S)rz!No condition was found for deleter"r[r"r"r#rrfs zDeleteStatement.execute) r%r&r'r(r$rrrrrr"r"rr#r#Cs    r#cs:eZdZdZdddddfdd Zd d d d ZZS) CreateCollectionIndexStatementa)A statement that creates an index on a collection. Args: collection (mysqlx.Collection): Collection. index_name (string): Index name. index_desc (dict): A dictionary containing the fields members that constraints the index to be created. It must have the form as shown in the following:: {"fields": [{"field": member_path, "type": member_type, "required": member_required, "collation": collation, "options": options, "srid": srid}, # {... more members, # repeated as many times # as needed} ], "type": type} rr1rr)r index_name index_descrcs4tj|dt||_||_|jdg|_dS)N)rNr)rzr$copydeepcopy _index_desc _index_namepop _fields_desc)r!rr%r&rr"r#r$s z'CreateCollectionIndexStatement.__init__rrZc Csh|jdkrtt|jz`t|j}t|trZ|dt dkrxtt|jn|j t dkrxtt|jWn8t t fk r}ztt|j|W5d}~XYnX|j std|jt|j tstdi}|j|d<|jj|d<|jjj|d<d|jkr$|jd|d<nd |d<|jd d |d <|d rPtd g|d <|jrptd|jz|j D]x}i}|d|d<|d|d<|dd |d<|dd |d<t|dtstdt|dtstd|ddkr|dstd|dd kr<|ddkrsN $      j\B*;#/}(9!L(1