o hc@sdZddlmZddlZddlZddlZddlmZmZm 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(dGddZ)dHdIddZ*dHdJddZ+dHdKd"d#Z, dHdLd'd(Z-Gd)d*d*Z.Gd+d,d,e.Z/Gd-d.d.e.Z0Gd/d0d0e.Z1Gd1d2d2e1Z2Gd3d4d4Z3Gd5d6d6e/Z4Gd7d8d8e/Z5Gd9d:d:e5Z6Gd;d<dd>e1Z8Gd?d@d@e/Z9GdAdBdBe/Z:GdCdDdDe/Z;GdEdFdFe.Zzsz.quote_multipart_identifier..)join)r>r4r#rBr$quote_multipart_identifieros rEdefault_schema table_nameTuple[str, str]cCs^d|vrdnd}||vrd|nd}||d}t|dkr!|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. r6r9r7r@r rr8)splitr,strip)rFrGr4quote delimitertempr#r#r$parse_table_name~s    rNc@seZdZdZd,d-d d Zed.d d Zed/ddZed0ddZej d1ddZed0ddZ ed2ddZ e j d3ddZ ed2ddZ e j d3ddZ ed2ddZ ed2dd Zej d3d!d Zd2d"d#Zd4d$d%Zd4d&d'Zd5d)d*Zd+S)6 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. Ttargetr doc_basedr5rrcCs@||_||_|r |nd|_d|_d|_d|_d|_d|_dS)NrTF) _target _doc_basedget_connection _connection_stmt_id _exec_counter_changed _prepared_deallocate_prepare_executer"rPrQr#r#r$r%s zStatement.__init__cC|jS)z#object: The database object target.)rRr"r#r#r$rPzStatement.targetrcCs|jjS)z*:class:`mysqlx.Schema`: The Schema object.)rRschemar]r#r#r$r_szStatement.schemaintcCr\)zXReturns this statement ID. Returns: int: The statement ID. rVr]r#r#r$stmt_idszStatement.stmt_idvaluecCr r!rar"rcr#r#r$rb cCr\)z5int: The number of times this statement was executed.rWr]r#r#r$ exec_counterr^zStatement.exec_countercCr\)z+bool: `True` if this statement has changes.rXr]r#r#r$changedr^zStatement.changedcCr r!rhrdr#r#r$rirecCr\)z1bool: `True` if this statement has been prepared.rYr]r#r#r$preparedr^zStatement.preparedcCr r!rjrdr#r#r$rkrecCs |jdkS)z;bool: `True` if this statement was executed more than once.r rfr]r#r#r$repeateds zStatement.repeatedcCr\)z9bool: `True` to deallocate + prepare + execute statement.rZr]r#r#r$deallocate_prepare_executer^z$Statement.deallocate_prepare_executecCr r!rmrdr#r#r$rnrecCr\)zkCheck if it is document based. Returns: bool: `True` if it is document based. )rSr]r#r#r$ is_doc_basedzStatement.is_doc_basedcCs|jd7_dS)z@Increments the number of times this statement has been executed.r Nrfr]r#r#r$increment_exec_countersz Statement.increment_exec_countercCs d|_dS)z                rOcseZdZdZ  dHdIfd d ZdJddZdKddZdLddZdMddZdNddZ dOddZ dPd d!Z dQd#d$Z dRd&d'Z dSd)d*ZdTd,d-ZdTd.d/ZdSd0d1ZdRd2d3ZdRd4d5Zed6dLd7d8Zed6dKd9d:Z dUdVd>d?ZdWd@dAZdXdDdEZdYdFdGZZS)ZFilterableStatementaA 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. TNrPrrQr5 condition Optional[str]rrcstj||di|_i|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_d|_d|_d|_d|_|rM||dSdS)NrPrQr1rF)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"rPrQr __class__r#r$r% s6zFilterableStatement.__init__obj!Union[DbDoc, Dict[str, Any], str]c Cst|tr|t|dSt|tr||dSt|trTzt|}t|ts0tWntyB}zt d|d}~ww| D] }||||qGdSt d)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 as_strr3jsonloads ValueErrorrkeys)r"rreserrkeyr#r#r$ _bind_single.s$       z FilterableStatement._bind_singleclausesr3cGs6d|_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,) rrDr0rrrSparse_order_specrrXr"rr#r#r$_sortHs  zFilterableStatement._sortc Cs^d|_||_zt||j }||_Wnty%}ztd|d}~ww|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) rrrrSrrrrplaceholder_name_to_positionrrX)r"rrrr#r#r$rYs  zFilterableStatement._set_wherefieldscGs:t|}d|_d||_t|j|j |_d|_dS)zJSet group by. Args: *fields: List of fields. TrN) r0rrDrrrSparse_expr_listrrXr"rr#r#r$ _set_group_byns   z!FilterableStatement._set_group_bycCs$d|_t||j |_d|_dS)zOSet having. Args: condition (str): The condition. TN)rrrSrrrXr"rr#r#r$ _set_having|s zFilterableStatement._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) r0rrDrrrSparse_table_select_projectionrrXrr#r#r$_set_projections   z#FilterableStatement._set_projectionDict[str, Any]cCr\zlReturns the binding map dictionary. Returns: dict: The binding map dictionary. rr]r#r#r$get_binding_maprpz#FilterableStatement.get_binding_mapUnion[Dict[str, Any], List]cCr\z\Returns the bindings list. Returns: `list`: The bindings list. rr]r#r#r$ get_bindingsrpz FilterableStatement.get_bindings9List[Union[ProtobufMessageType, ProtobufMessageCextType]]cCr\)zrReturns the grouping expression list. Returns: `list`: The grouping expression list. )rr]r#r#r$ get_groupingrpz FilterableStatement.get_groupingrcCr\)zdReturns the having expression. Returns: object: The having expression. )rr]r#r#r$ get_havingrpzFilterableStatement.get_havingr`cCr\)z]Returns the limit row count. Returns: int: The limit row count. )rr]r#r#r$get_limit_row_countrpz'FilterableStatement.get_limit_row_countcCr\)zWReturns the limit offset. Returns: int: The limit offset. )rr]r#r#r$get_limit_offsetrpz$FilterableStatement.get_limit_offsetcCr\)zbReturns the where expression. Returns: object: The where expression. )rr]r#r#r$get_where_exprrpz"FilterableStatement.get_where_exprcCr\)zlReturns the projection expression. Returns: object: The projection expression. )rr]r#r#r$get_projection_exprz'FilterableStatement.get_projection_exprcCr\)z`Returns the sort expression. Returns: object: The sort expression. )rr]r#r#r$ get_sort_exprrz!FilterableStatement.get_sort_expr8.0.12cC ||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.wherecG |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.sort row_countoffset Optional[int]cCsnt|tr |dkr td|js!t|jdk|_t|jdk |_||_d|_|r5| |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`rrr5rWrXrZrrwarningswarnDeprecationWarning)r"rrr#r#r$limits zFilterableStatement.limitcCs$t|tr |dkr td||_|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.offsetargsrcGsNd|_t|}|dkr||d|S|dkr#|d|j|d<|Std)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,rrr)r"rcountr#r#r$r7szFilterableStatement.bindcCrsrtrur]r#r#r$rwOrxzFilterableStatement.executeTNrPrrQr5rrrr)rrrr)rr3rr)rr3rr)rr3rr)rr3rr)rr3rrrr)rr)rr)rrr{r!)rr`rrrr)rr`rr)rrrrr})r&r'r(r)r%rrrrrrrrrrrrrrrrrrrrrrw __classcell__r#r#rr$rs: $                # rcsXeZdZdZdfdd Zedd d Zdd dZdddZdddZ dddZ Z S) SqlStatementzA statement for SQL execution. Args: connection (mysqlx.connection.Connection): Connection object. sql (string): The sql statement to be executed. connectionrsqlr3rrcs8tjddd||_||_d|_g|_d|_d|_dSNFr)rr%rU_sqlrrrr)r"rrrr#r$r%`s zSqlStatement.__init__cCr\)zstring: The SQL text statement.)rr]r#r#r$rir^zSqlStatement.sqlrcCr\rrr]r#r#r$rnrpzSqlStatement.get_binding_mapUnion[Tuple, List]cCr\rrr]r#r#r$rvrpzSqlStatement.get_bindingsrrcGsJt|dkr tdd|_t|}t|ttfr||_|S|j||S)zBinds value(s) to a specific placeholder(s). Args: *args: The value(s) to bind. Returns: mysqlx.SqlStatement: SqlStatement object. rrT) r,rrr0r-r.r/rappend)r"rbindingsr#r#r$r~s  zSqlStatement.bindrcC |j|S)zaExecute the statement. Returns: mysqlx.SqlResult: SqlResult object. )rUsend_sqlr]r#r#r$rw zSqlStatement.execute)rrrr3rrrr3r)rr)rrrr)rr) r&r'r(r)r%r~rrrrrwrr#r#rr$rXs    rc6eZdZdZdfdd Zdd d ZdddZZS)WriteStatementz*Provide common write operation attributes.rPrrQr5rrcst||g|_dSr!)rr%_valuesr[rr#r$r%s zWriteStatement.__init__pList[Union[int, str, DbDoc, Dict[str, Any], List[Optional[Union[str, int, float, ExprParser, Dict[str, Any]]]]]]cCr\)z^Returns the list of values. Returns: `list`: The list of values. )rr]r#r#r$ get_valuesszWriteStatement.get_valuesrcCrsrtrur]r#r#r$rwrxzWriteStatement.executerz)rrr})r&r'r(r)r%rrwrr#r#rr$rs  rcsLeZdZdZdfdd Zdd d Zddd dZdddZdddZZ S) AddStatementzA statement for document addition on a collection. Args: collection (mysqlx.Collection): The Collection object. collectionrrrcst|dd|_g|_dS)NTF)rr%_upsertids)r"rrr#r$r%s zAddStatement.__init__r5cCr\)zhReturns `True` if it's an upsert. Returns: bool: `True` if it's an upsert. rr]r#r#r$ is_upsertrpzAddStatement.is_upsertTrccCs ||_|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. rrdr#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. )r0r-r rr)r"r*valr#r#r$adds zAddStatement.addrcCs t|jdkr tS|j|S)[Execute the statement. Returns: mysqlx.Result: Result object. r)r,rrrU send_insertr]r#r#r$rws zAddStatement.execute)rrrrr|ry)rcr5rr)r*r rrrr) r&r'r(r)r%rrrrwrr#r#rr$rs   rc@s&eZdZdZddd d Zdd dZdS) 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. N update_typer`sourcer3rcrrrc Csh|tdkr|||dS||_z t|dj|_Wnty.}zt||d}~ww||_ dS)N*Mysqlx.Crud.UpdateOperation.UpdateType.SETF) r _table_setrrdocument_fieldr2rrrrc)r"rrrcrr#r#r$r% s   zUpdateSpec.__init__cCs$td|_t|d|_||_dS)zlTable set. Args: source (str): The source. value (str): The value. rTN)rrrparse_table_update_fieldrrc)r"rrcr#r#r$rs  zUpdateSpec._table_setr!)rr`rr3rcrrr)rr3rcrrr)r&r'r(r)r%rr#r#r#r$rs  rcseZdZdZd'fdd Zd(d d Zd)ddZd*ddZedd*ddZ d+ddZ d,ddZ d*ddZ d-d"d#Z 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. rrrr3rrcstj||di|_dSN)rPrrr% _update_opsr"rrrr#r$r%. zModifyStatement.__init__rcGr)zSets the sorting criteria. Args: *clauses: The expression strings defining the sort criteria. Returns: mysqlx.ModifyStatement: ModifyStatement object. rrr#r#r$r2 zModifyStatement.sortrcCr\ztReturns the list of update operations. Returns: `list`: The list of update operations. rr]r#r#r$get_update_ops=rpzModifyStatement.get_update_opsdoc_pathrcrcC 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_SETTrrrrXr"rrcr#r#r$setE  zModifyStatement.setrcCr)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_pathscGs,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)r0rrrrX)r"ritemr#r#r$unsetms  zModifyStatement.unsetfieldcCr)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"rrcr#r#r$ array_inserts  zModifyStatement.array_insertcCr)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_appenddoc#Union[Dict, DbDoc, ExprParser, str]cCsX|durd}t|ttttfstdttddt|tr!|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. Nr1z>Invalid data for update operation on document collection tablez2Mysqlx.Crud.UpdateOperation.UpdateType.MERGE_PATCH$patchT) r-rrr r3rrrrrrX)r"rr#r#r$rs  zModifyStatement.patchrcC|jstd|j|S)Execute the statement. Returns: mysqlx.Result: Result object. Raises: ProgrammingError: If condition was not set. z!No condition was found for modifyrrrU send_updater]r#r#r$rw  zModifyStatement.executerrrr3rr)rr3rrr)rr3rcrrr)rr3rr)rr3rcrrr)rrrrr)r&r'r(r)r%rr rrrrrrrrwrr#r#rr$r"s       rcseZdZdZ  d%d&fd d Zed'ddZd(ddZd)ddZd)ddZ e j fd*ddZ e j fd*ddZ d+ddZd,d d!Zd-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. TNrPrrQr5rrrrcs(t|||d|_d|_tj|_dSNF)rr%_lock_exclusive _lock_sharedr DEFAULT_lock_contentionrrr#r$r%s zReadStatement.__init__r cCr\)z::class:`mysqlx.LockContention`: The lock contention value.)r+r]r#r#r$lock_contentionr^zReadStatement.lock_contentionr,c Cs<zt|j}Wnty}ztd|d}~ww||_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 rcrrr+)r"r,_rr#r#r$_set_lock_contentions  z"ReadStatement._set_lock_contentioncCr\)zrReturns `True` if is `EXCLUSIVE LOCK`. Returns: bool: `True` if is `EXCLUSIVE LOCK`. )r(r]r#r#r$is_lock_exclusiverpzReadStatement.is_lock_exclusivecCr\)zlReturns `True` if is `SHARED LOCK`. Returns: bool: `True` if is `SHARED LOCK`. )r)r]r#r#r$is_lock_sharedrpzReadStatement.is_lock_sharedcCd|_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. FTr(r)r.r"r,r#r#r$ lock_shared  zReadStatement.lock_sharedcCr1)zExecute a read operation with `EXCLUSIVE LOCK`. Only one lock can be active at a time. Args: lock_contention (:class:`mysqlx.LockContention`): Lock contention. TFr2r3r#r#r$lock_exclusiver5zReadStatement.lock_exclusiverr3cGs|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_bycCs|||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.havingUnion[DocResult, RowResult]cCrr)rU send_findr]r#r#r$rwDrzReadStatement.executerr)rr )r,r rrr|)r,r rr&)rr3rr&)rr3rr&)rr9)r&r'r(r)r%r~r,r.r/r0r r*r4r6r7r8rwrr#r#rr$r&s"       r&cs:eZdZdZ ddfd d Zdd dZdddZZS) 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. Nrrrrrrcst|d|dS)NTrr%rrr#r$r%XszFindStatement.__init__rr3cGr)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.fieldsrcGrzSets the sorting criteria. Args: *clauses: The expression strings defining the sort criteria. Returns: mysqlx.FindStatement: FindStatement object. rrr#r#r$rir zFindStatement.sortr!)rrrrrr)rr3rr<)rr3rr<)r&r'r(r)r%rrrr#r#rr$r<Ms    r<cs@eZdZdZdfdd Zdd d ZdddZdddZZS)SelectStatementzA statement for record retrieval operations on a Table. Args: table (mysqlx.Table): The Table object. *fields: The fields to be retrieved. tablerrr3rrcst|d|j|dSr')rr%rr"r@rrr#r$r%}zSelectStatement.__init__rcCr)zSets the search condition to filter. Args: condition (str): Sets the search condition to filter records. Returns: mysqlx.SelectStatement: SelectStatement object. rrr#r#r$rr zSelectStatement.wherercGr)zSets the order by criteria. Args: *clauses: The expression strings defining the order by criteria. Returns: mysqlx.SelectStatement: SelectStatement object. rrr#r#r$order_byr zSelectStatement.order_byc Cs|jr d|jnd}|jrd|jnd}|jrd|jnd}|jr*d|jnd}|jr9d|j d|j nd}d|j p@d d |j j d |jj ||||| }|S) zYReturns the generated SQL. Returns: str: The generated SQL. z WHERE r1z GROUP BY z HAVING z ORDER BY z LIMIT z OFFSET zSELECT *z FROM r@)rrrrrrrrrrrrr_namerP)r"rr7r8rCrstmtr#r#r$get_sqls2zSelectStatement.get_sql)r@rrr3rr)rr3rr?)rr3rr?r) r&r'r(r)r%rrCrGrr#r#rr$r?us    r?cr)InsertStatementzA statement for insert operations on Table. Args: table (mysqlx.Table): The Table object. *fields: The fields to be inserted. r@rrrrrcst|dt||_dSr')rr%r0_fieldsrArr#r$r%rBzInsertStatement.__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.r0)r"r*r#r#r$r*s zInsertStatement.valuesrcCrr:)rUrr]r#r#r$rwrzInsertStatement.execute)r@rrrrr)r*rrrHr)r&r'r(r)r%r*rwrr#r#rr$rHs   rHcsTeZdZdZdfdd Zdd d Zdd dZdddZdddZdddZ 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. r@rrrcstj|ddi|_dSrrr"r@rr#r$r%r zUpdateStatement.__init__rr3cCr)zSets the search condition to filter. Args: condition (str): Sets the search condition to filter records. Returns: mysqlx.UpdateStatement: UpdateStatement object. rrr#r#r$rr zUpdateStatement.wherercGr)zSets the order by criteria. Args: *clauses: The expression strings defining the order by criteria. Returns: mysqlx.UpdateStatement: UpdateStatement object. rrr#r#r$rCr zUpdateStatement.order_byrcCr\r r r]r#r#r$r rpzUpdateStatement.get_update_opsrrcrcCr)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$rrzUpdateStatement.setrcCr )zExecute the statement. Returns: mysqlx.Result: Result object Raises: ProgrammingError: If condition was not set. z!No condition was found for updater"r]r#r#r$rw r$zUpdateStatement.executer@rrr)rr3rrJ)rr3rrJr)rr3rcrrrJr) r&r'r(r)r%rrCr rrwrr#r#rr$rJs  rJcr)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. rrrr3rrcstj||ddSrr=rrr#r$r%'zRemoveStatement.__init__rcGrr>rrr#r#r$r*r zRemoveStatement.sortrcCr )r!z!No condition was found for removerrrU send_deleter]r#r#r$rw5r$zRemoveStatement.executer%)rr3rrMr)r&r'r(r)r%rrwrr#r#rr$rMs   rMcs@eZdZdZdfdd Zdd d Zdd dZdddZZS)DeleteStatementzA statement that drops a table. Args: table (mysqlx.Table): The Table object. .. versionchanged:: 8.0.12 The ``condition`` parameter was removed. r@rrrcstj|dddSrr=rKrr#r$r%MrNzDeleteStatement.__init__rr3cCr)zSets the search condition to filter. Args: condition (str): Sets the search condition to filter records. Returns: mysqlx.DeleteStatement: DeleteStatement object. rrr#r#r$rPr zDeleteStatement.wherercGr)zSets the order by criteria. Args: *clauses: The expression strings defining the order by criteria. Returns: mysqlx.DeleteStatement: DeleteStatement object. rrr#r#r$rC[r zDeleteStatement.order_byrcCr )r!z!No condition was found for deleterOr]r#r#r$rwfr$zDeleteStatement.executerL)rr3rrQ)rr3rrQr) r&r'r(r)r%rrCrwrr#r#rr$rQCs    rQcs,eZdZdZdfd d Zdd dZZS)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} rr index_namer3 index_descrrrcs4tj|dt||_||_|jdg|_dS)N)rPr)rr%copydeepcopy _index_desc _index_namepop _fields_desc)r"rrSrTrr#r$r%s z'CreateCollectionIndexStatement.__init__rc Cs@|jdur tt|jz0t|j}t|tr-|dt dkr,tt|jn|j t dkrr?r4r3rr3)rFr3rGr3r4r3rrH)=r) __future__rrUrrtypingrrrrrrr constantsr dbdocr errorsr rrrhelpersrprotobufrresultrrrrtypesrrrrrrrkrr0r:r=rErNrOrrrrrrr&r<r?rHrJrMrQrRr#r#r#r$sP $            j\B*;#/}(9!L(1