o hg@sFdZddlmZddlZddlZddlmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZddlmZmZmZdd lmZmZmZdd lmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(m)Z)erpdd l*m+Z+d Z,dZ-dZ.dZ/dZ0GdddZ1Gddde1Z2Gddde1Z3Gddde1Z4Gddde4Z5dS)z,Implementation of the CRUD database objects.) annotationsN) TYPE_CHECKINGAnyDictListOptionalUnion)DbDoc)ER_NO_SUCH_TABLEER_TABLE_EXISTS_ERRORER_X_CMD_NUM_ARGUMENTSER_X_INVALID_ADMIN_COMMAND)NotSupportedErrorOperationalErrorProgrammingError) deprecatedescapequote_identifier) AddStatementCreateCollectionIndexStatementDeleteStatement FindStatementInsertStatementModifyStatementRemoveStatementSelectStatementUpdateStatement)ConnectionType SchemaType SessionType StrOrBytes)Resultz_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@seZdZdZd&dd Zed'd d Zed(d dZed)ddZd*ddZ d'ddZ d(ddZ d)ddZ d+ddZ eddd+d d!Zedd"d)d#d$Zd%S),DatabaseObjectzProvides base functionality for database objects. Args: schema (mysqlx.Schema): The Schema object. name (str): The database object name. schemarnamer!returnNonecCs:||_t|tr |n||_|j|_|j|_dSN) _schema isinstancebytesdecode_name get_session_sessionget_connection _connection)selfr$r%r3F/var/www/html/scripts/venv/lib/python3.10/site-packages/mysqlx/crud.py__init__Vs zDatabaseObject.__init__r cC|jS)z,:class:`mysqlx.Session`: The Session object.r/r2r3r3r4session\zDatabaseObject.sessioncCr6)z*:class:`mysqlx.Schema`: The Schema object.r)r8r3r3r4r$ar:zDatabaseObject.schemastrcCr6)z&str: The name of this database object.r-r8r3r3r4r%fr:zDatabaseObject.namercCr6)z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )r1r8r3r3r4r0kzDatabaseObject.get_connectioncCr6)zwReturns the session of this database object. Returns: mysqlx.Session: The Session object. r7r8r3r3r4r.sr>zDatabaseObject.get_sessioncCr6)z{Returns the Schema object of this database object. Returns: mysqlx.Schema: The Schema object. r;r8r3r3r4 get_schema{r>zDatabaseObject.get_schemacCr6)zwReturns the name of this database object. Returns: str: The name of this database object. r=r8r3r3r4get_namer>zDatabaseObject.get_namercCst)zVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. Raises: NotImplementedError: This method must be implemented. )NotImplementedErrorr8r3r3r4exists_in_databases 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. )rBr8r3r3r4 am_i_reals zDatabaseObject.am_i_realzUse 'get_name()' method insteadcCrC)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@r8r3r3r4who_am_is zDatabaseObject.who_am_iN)r$rr%r!r&r')r&r )r&r)r&r<)r&r)r&r)__name__ __module__ __qualname____doc__r5propertyr9r$r%r0r.r?r@rBrrDrEr3r3r3r4r#Ns$          r#cseZdZdZd,fdd Zd-d d Zd.ddZ d/d0ddZd1ddZd/d0ddZ d/d2ddZ d/d3ddZ d4d d!Z  "d5d6d(d)Z "d7d8d*d+ZZS)9SchemazA client-side representation of a database schema. Provides access to the schema contents. Args: session (mysqlx.XSession): Session object. name (str): The Schema name. r9r r%r<r&r'cs||_t||dSr()r/superr5)r2r9r% __class__r3r4r5szSchema.__init__boolcCs tt|j}|j|dkSzVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. r )_COUNT_SCHEMAS_QUERYformatrr-r1execute_sql_scalarr2sqlr3r3r4rBszSchema.exists_in_databaseList[Collection]c Csz|jdd|ji}|g}|D](}|ddkrqz t||d}Wnty4t||d}Ynw||q|S)zyReturns a list of collections for this schema. Returns: `list`: List of Collection objects. list_objectsr$type COLLECTION TABLE_NAMEr%)r1get_row_resultr- fetch_all Collection ValueErrorappend)r2rows collectionsrow collectionr3r3r4get_collectionss   zSchema.get_collectionsFcheck_existenceTablecCs |||S)zuReturns a a table object for the given collection Returns: mysqlx.Table: Table object. ) get_table)r2r%rer3r3r4get_collection_as_tables zSchema.get_collection_as_table List[Table]c Cs||jdd|ji}|g}d}|D]'}|d|vr;z t||d}Wnty5t||d}Ynw||q|S)zoReturns a list of tables for this schema. Returns: `list`: List of Table objects. rWr$)TABLEVIEWrXrZr%)r1r[r-r\rfr^r_)r2r`tables object_typesrbtabler3r3r4 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)rfrBr)r2r%rernr3r3r4rg zSchema.get_tableViewcCrp)ztReturns the view of the given name for this schema. Returns: mysqlx.View: View object. zView does not exist)rrrBr)r2r%reviewr3r3r4get_viewrqzSchema.get_viewr]cCrp)zReturns the collection of the given name for this schema. Returns: mysqlx.Collection: Collection object. zCollection does not exist)r]rBr)r2r%rercr3r3r4get_collectionrqzSchema.get_collectioncCs&|jdtt|jt|ddS)zmDrops a collection. Args: name (str): The name of the collection to be dropped. rUFN)r1execute_nonquery_DROP_TABLE_QUERYrRrr-)r2r%r3r3r4drop_collection szSchema.drop_collectionNreuse_existing validation%Optional[Dict[str, Union[str, Dict]]]kwargsrc Ks|stdd|vrtdt|d}t||}|j|d}|durt|tr*|s.tdd}|D] }||vr?td|q2g} d |vrZ|d } t| tsStd | d | fd |vr~|d } t| ttfsmtd | d t| trzt | n| fd | f|d<z |j ddd|W|Sty} z*| jtkrtd| | jtkr|std|d| nt| j| j| WYd} ~ |Sd} ~ ww)aJCreates 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_existing (bool): `True` to reuse an existing collection. validation (Optional[dict]): A dict, containing the keys `level` with the validation level and `schema` with a dict or a string representation of a JSON schema specification. Returns: mysqlx.Collection: Collection object. Raises: :class:`mysqlx.ProgrammingError`: If ``reuse_existing`` is False and collection exists or the collection name is invalid. :class:`mysqlx.NotSupportedError`: If schema validation is not supported by the server. .. versionchanged:: 8.0.21 Collection name is invalidreusezG'reuse' is deprecated since 8.0.21. Please use 'reuse_existing' instead)r$r%NInvalid value for 'validation'levelr$ Invalid option in 'validation': rInvalid value for 'level'r$Invalid value for 'schema'rzoptionsmysqlxcreate_collectionTlYour MySQL server does not support the requested operation. Please update to MySQL 8.0.19 or a later version Collection 'z' already exists)rwarningswarnDeprecationWarningr]r-r*dictr<r_jsondumpsr1rvrerrnor rr msg) r2r%ryrzr|rcfields valid_optionsoptionrrr$errr3r3r4r.sv        zSchema.create_collectionc Cs*|stdt|tr |stdd}|D] }||vr"td|qg}d|vr=|d}t|ts6td|d|fd|vra|d}t|ttfsPtd|dt|tr]t|n|f|j|d |fd }z |j d d d |WdSt y} z| j t krt d| t| j| j | d} ~ ww)a Modifies a collection using a JSON schema validation. Args: name (str): The name of the collection. validation (Optional[dict]): A dict, containing the keys `level` with the validation level and `schema` with a dict or a string representation of a JSON schema specification. Raises: :class:`mysqlx.ProgrammingError`: If the collection name or validation is invalid. :class:`mysqlx.NotSupportedError`: If schema validation is not supported by the server. .. versionadded:: 8.0.21 r}rrrrrr$rrz)r$r%rrmodify_collection_optionsTrN)rr*rr<r_rrr-r1rvrrrrr) r2r%rzrrrrr$rrr3r3r4modify_collectionsT   zSchema.modify_collection)r9r r%r<r&r'r&rO)r&rV)F)r%r<rerOr&rf)r&ri)r%r<rerOr&rr)r%r<rerOr&r])r%r<r&r')FN) r%r<ryrOrzr{r|rr&r]r()r%r<rzr{r&r')rFrGrHrIr5rBrdrhrorgrtrurxrr __classcell__r3r3rMr4rKs"     `rKc@seZdZdZd0ddZd1d2d d Zd3ddZd4ddZd5ddZd6ddZ d7dd Z d8d"d#Z d9d(d)Z d9d*d+Z d:d,d-Zd;d.d/ZdS)s:   ,  d"Cc