o h5I@sdZddlmZddlmZddlmZmZddlm Z m Z m Z ddl m Z mZmZmZmZmZmZmZmZdZdZd Zd Zd ZGd d d eZGdddeZGdddeZGdddeZGdddeZ dS)z,Implementation of the CRUD database objects.)DbDoc)ER_NO_SUCH_TABLE)OperationalErrorProgrammingError) deprecatedescapequote_identifier) FindStatement AddStatementRemoveStatementModifyStatementSelectStatementInsertStatementDeleteStatementUpdateStatementCreateCollectionIndexStatementz_SELECT COUNT(*) FROM information_schema.views WHERE table_schema = '{0}' AND table_name = '{1}'z`SELECT COUNT(*) FROM information_schema.tables WHERE table_schema = '{0}' AND table_name = '{1}'zJSELECT COUNT(*) FROM information_schema.schemata WHERE schema_name = '{0}'zSELECT COUNT(*) FROM {0}.{1}zDROP TABLE IF EXISTS {0}.{1}c@seZdZdZddZeddZeddZedd Zd d Z d d Z ddZ ddZ ddZ eddddZeddddZdS)DatabaseObjectzProvides base functionality for database objects. Args: schema (mysqlx.Schema): The Schema object. name (str): The database object name. cCs(||_||_|j|_|j|_dSN)_schema_name get_session_sessionget_connection _connection)selfschemanamer=/var/www/html/env/lib/python3.10/site-packages/mysqlx/crud.py__init__:s zDatabaseObject.__init__cC|jS)z5:class:`mysqlx.Session`: The Session object. rrrrrsession@zDatabaseObject.sessioncCr )z3:class:`mysqlx.Schema`: The Schema object. rr"rrrrFr$zDatabaseObject.schemacCr )z/str: The name of this database object. rr"rrrrLr$zDatabaseObject.namecCr )z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )rr"rrrrRzDatabaseObject.get_connectioncCr )zwReturns the session of this database object. Returns: mysqlx.Session: The Session object. r!r"rrrrZr'zDatabaseObject.get_sessioncCr )z{Returns the Schema object of this database object. Returns: mysqlx.Schema: The Schema object. r%r"rrr get_schemabr'zDatabaseObject.get_schemacCr )zwReturns the name of this database object. Returns: str: The name of this database object. r&r"rrrget_namejr'zDatabaseObject.get_namecCst)zVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. Raises: NotImplementedError: This method must be implemented. )NotImplementedErrorr"rrrexists_in_databasers z!DatabaseObject.exists_in_databasez8.0.12z)Use 'exists_in_database()' method insteadcC|S)a+Verifies if this object exists in the database. Returns: bool: `True` if object exists in database. Raises: NotImplementedError: This method must be implemented. .. deprecated:: 8.0.12 Use ``exists_in_database()`` method instead. )r+r"rrr am_i_real}s zDatabaseObject.am_i_realzUse 'get_name()' method insteadcCr,)zReturns the name of this database object. Returns: str: The name of this database object. .. deprecated:: 8.0.12 Use ``get_name()`` method instead. )r)r"rrrwho_am_is zDatabaseObject.who_am_iN)__name__ __module__ __qualname____doc__rpropertyr#rrrrr(r)r+rr-r.rrrrr3s$    rcsreZdZdZfddZddZddZdd d Zd d Zdd dZ dddZ dddZ ddZ dddZ ZS)SchemazA client-side representation of a database schema. Provides access to the schema contents. Args: session (mysqlx.XSession): Session object. name (str): The Schema name. cs||_tt|||dSr)rsuperr4r)rr#r __class__rrrszSchema.__init__cCs tt|j}|j|dkSzVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. r)_COUNT_SCHEMAS_QUERYformatrrrexecute_sql_scalarrsqlrrrr+szSchema.exists_in_databasec Csv|jd|j}|g}|D](}|ddkrqz t||d}Wnty2t||d}Ynw||q|S)zyReturns a list of collections for this schema. Returns: `list`: List of Collection objects. list_objectstype COLLECTION TABLE_NAMEr)rget_row_resultr fetch_all Collection ValueErrorappend)rrows collectionsrow collectionrrrget_collectionss   zSchema.get_collectionsFcCs |||S)zuReturns a a table object for the given collection Returns: mysqlx.Table: Table object. ) get_table)rrcheck_existencerrrget_collection_as_tables zSchema.get_collection_as_tablec Csx|jd|j}|g}d}|D]'}|d|vr9z t||d}Wnty3t||d}Ynw||q|S)zoReturns a list of tables for this schema. Returns: `list`: List of Table objects. r>)TABLEVIEWr?rAr)rrBrrCTablerErF)rrGtables object_typesrItablerrr get_tabless   zSchema.get_tablescC"t||}|r|std|S)zwReturns the table of the given name for this schema. Returns: mysqlx.Table: Table object. zTable does not exist)rQr+r)rrrMrTrrrrL zSchema.get_tablecCrV)ztReturns the view of the given name for this schema. Returns: mysqlx.View: View object. zView does not exist)Viewr+r)rrrMviewrrrget_viewrWzSchema.get_viewcCrV)zReturns the collection of the given name for this schema. Returns: mysqlx.Collection: Collection object. zCollection does not exist)rDr+r)rrrMrJrrrget_collectionrWzSchema.get_collectioncCs&|jdtt|jt|ddS)zmDrops a collection. Args: name (str): The name of the collection to be dropped. r=FN)rexecute_nonquery_DROP_TABLE_QUERYr:rr)rrrrrdrop_collections zSchema.drop_collectioncCsH|stdt||}|s|jddd|j||S|s"td|S)aCreates in the current schema a new collection with the specified name and retrieves an object representing the new collection created. Args: name (str): The name of the collection. reuse (bool): `True` to reuse an existing collection. Returns: mysqlx.Collection: Collection object. Raises: :class:`mysqlx.ProgrammingError`: If ``reuse`` is False and collection exists. zCollection name is invalidxplugincreate_collectionTzCollection already exists)rrDr+rr\r)rrreuserJrrrr` s  zSchema.create_collection)F)r/r0r1r2rr+rKrNrUrLrZr[r^r` __classcell__rrr6rr4s      r4c@sreZdZdZddZdddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZdS)rDzRepresents a collection of documents on a schema. Args: schema (mysqlx.Schema): The Schema object. name (str): The collection name. cC*tt|jjt|j}|j|dkSr8_COUNT_TABLES_QUERYr:rrrrrr;r<rrrr+-zCollection.exists_in_databaseNcC t||S)zRetrieves documents from a collection. Args: condition (Optional[str]): The string with the filter expression of the documents to be retrieved. )r r conditionrrrfind7s zCollection.findcGst|j|S)zAdds a list of documents to a collection. Args: *values: The document list to be added into the collection. Returns: mysqlx.AddStatement: AddStatement object. )r add)rvaluesrrrrk@s zCollection.addcCrg)apRemoves documents based on the ``condition``. Args: condition (str): The string with the filter expression of the documents to be removed. Returns: mysqlx.RemoveStatement: RemoveStatement object. .. versionchanged:: 8.0.12 The ``condition`` parameter is now mandatory. )r rhrrrremoveL zCollection.removecCrg)arModifies documents based on the ``condition``. Args: condition (str): The string with the filter expression of the documents to be modified. Returns: mysqlx.ModifyStatement: ModifyStatement object. .. versionchanged:: 8.0.12 The ``condition`` parameter is now mandatory. )r rhrrrmodify[rnzCollection.modifyc Chtt|jjt|j}z |j|}W|Sty3}z|j t kr.td|j|jjd}~ww)z}Counts the documents in the collection. Returns: int: The total of documents in the collection. z-Collection '{}' does not exist in schema '{}'N _COUNT_QUERYr:rrrrrr;rerrnorrr=reserrrrrcountj zCollection.countcCs t|||S)a"Creates a collection index. Args: index_name (str): Index name. fields_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} )r)r index_name fields_descrrr create_index|s zCollection.create_indexcCs |jddd|jj|j|dS)z[Drops a collection index. Args: index_name (str): Index name. r_drop_collection_indexFN)rr\rrr)rryrrr drop_indexs  zCollection.drop_indexcCs|dd|d|S)zReplaces the Document matching the document ID with a new document provided. Args: doc_id (str): Document ID doc (DbDoc/dict): New Document _id = :id$id)rosetbindexecuterdoc_iddocrrr replace_ones zCollection.replace_onecCs,t|ts t|}|||dS)zUpserts the Document matching the document ID with a new document provided. Args: doc_id (str): Document ID doc (DbDoc/dict): New Document T) isinstancerrkcopyupsertrrrrradd_or_replace_ones zCollection.add_or_replace_onecCs|dd|S)zjReturns a Document matching the Document ID. Args: doc_id (str): Document ID r~r)rjrr fetch_onerrrrrget_oneszCollection.get_onecCs|dd|S)zRemoves a Document matching the Document ID. Args: doc_id (str): Document ID Returns: mysqlx.Result: Result object. r~r)rmrrrrrr remove_ones zCollection.remove_oner)r/r0r1r2r+rjrkrmrorwr{r}rrrrrrrrrD%s      rDc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)rQzRepresents a database table on a schema. Provides access to the table through standard INSERT/SELECT/UPDATE/DELETE statements. Args: schema (mysqlx.Schema): The Schema object. name (str): The table name. cCrcr8rdr<rrrr+rfzTable.exists_in_databasecGt|g|RS)zCreates a new :class:`mysqlx.SelectStatement` object. Args: *fields: The fields to be retrieved. Returns: mysqlx.SelectStatement: SelectStatement object )r rfieldsrrrselect z Table.selectcGr)zCreates a new :class:`mysqlx.InsertStatement` object. Args: *fields: The fields to be inserted. Returns: mysqlx.InsertStatement: InsertStatement object )rrrrrinsertrz Table.insertcCt|S)zCreates a new :class:`mysqlx.UpdateStatement` object. Returns: mysqlx.UpdateStatement: UpdateStatement object )rr"rrrupdatesz Table.updatecCr)zCreates a new :class:`mysqlx.DeleteStatement` object. Returns: mysqlx.DeleteStatement: DeleteStatement object .. versionchanged:: 8.0.12 The ``condition`` parameter was removed. )rr"rrrdeletes z Table.deletec Crp)ziCounts the rows in the table. Returns: int: The total of rows in the table. z(Table '{}' does not exist in schema '{}'Nrqrtrrrrwrxz Table.countcCrc)zDetermine if the underlying object is a view or not. Returns: bool: `True` if the underlying object is a view. r_COUNT_VIEWS_QUERYr:rrrrrr;r<rrris_viewrfz Table.is_viewN) r/r0r1r2r+rrrrrwrrrrrrQs     rQc@seZdZdZddZdS)rXzRepresents a database view on a schema. Provides a mechanism for creating, alter and drop views. Args: schema (mysqlx.Schema): The Schema object. name (str): The table name. cCrcr8rr<rrrr+-rfzView.exists_in_databaseN)r/r0r1r2r+rrrrrX#s rXN)!r2dbdocr errorcodererrorsrrhelpersrrr statementr r r r r rrrrrrer9rrr]objectrr4rDrQrXrrrrs$  ,f $[