@igddZddlmZddlZddlZddlmZmZmZm 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)erdd l*m+Z+d Z,dZ-dZ.dZ/dZ0GddZ1Gdde1Z2Gdde1Z3Gdde1Z4Gdde4Z5y)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}ceZdZdZddZeddZeddZeddZddZ ddZ ddZ dd Z dd Z ed d dd Zed dddZy)DatabaseObjectzProvides base functionality for database objects. Args: schema (mysqlx.Schema): The Schema object. name (str): The database object name. c||_t|tr|jn||_|jj |_|j j|_yN) _schema isinstancebytesdecode_name get_session_sessionget_connection _connection)selfschemanames E/var/www/html/aiagents/fe/lib/python3.12/site-packages/mysqlx/crud.py__init__zDatabaseObject.__init__VsJ#) +5dE+B$++- %)\\%=%=%? +/==+G+G+Ic|jS)z,:class:`mysqlx.Session`: The Session object.r.r1s r4sessionzDatabaseObject.session\s}}r6c|jS)z*:class:`mysqlx.Schema`: The Schema object.r(r9s r4r2zDatabaseObject.schemaas||r6c|jS)z&str: The name of this database object.r,r9s r4r3zDatabaseObject.namefszzr6c|jS)z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )r0r9s r4r/zDatabaseObject.get_connectionks r6c|jS)zwReturns the session of this database object. Returns: mysqlx.Session: The Session object. r8r9s r4r-zDatabaseObject.get_sessionss }}r6c|jS)z{Returns the Schema object of this database object. Returns: mysqlx.Schema: The Schema object. r<r9s r4 get_schemazDatabaseObject.get_schema{s ||r6c|jS)zwReturns the name of this database object. Returns: str: The name of this database object. r>r9s r4get_namezDatabaseObject.get_names zzr6ct)zVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. Raises: NotImplementedError: This method must be implemented. )NotImplementedErrorr9s r4exists_in_databasez!DatabaseObject.exists_in_databases "!r6z8.0.12z)Use 'exists_in_database()' method insteadc"|jS)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. )rGr9s r4 am_i_realzDatabaseObject.am_i_reals&&((r6zUse 'get_name()' method insteadc"|jS)zReturns the name of this database object. Returns: str: The name of this database object. .. deprecated:: 8.0.12 Use ``get_name()`` method instead. )rDr9s r4who_am_izDatabaseObject.who_am_is}}r6N)r2r r3r"returnNone)rLr!)rLr )rLstr)rLr)rLr)__name__ __module__ __qualname____doc__r5propertyr:r2r3r/r-rBrDrGrrIrKr6r4r%r%NsJ   "EF )G );< = r6r%ceZdZdZd fd ZddZddZ d ddZddZdddZ dddZ ddd Z dd Z d dd Z d dd ZxZS)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. c4||_t| ||yr')r.superr5)r1r:r3 __class__s r4r5zSchema.__init__s%,  t$r6ctjt|j}|jj |dk(SzVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. r )_COUNT_SCHEMAS_QUERYformatrr,r0execute_sql_scalarr1sqls r4rGzSchema.exists_in_databases9 #))&*<=22371<> r6cXt||}|r|js td|S)ztReturns the view of the given name for this schema. Returns: mysqlx.View: View object. zView does not exist)ViewrGr)r1r3rrviews r4get_viewzSchema.get_views0 D$ **,&'<== r6cXt||}|r|js td|S)zReturns the collection of the given name for this schema. Returns: mysqlx.Collection: Collection object. zCollection does not exist)rhrGr)r1r3rrrns r4get_collectionzSchema.get_collections1  d+ 002&'BCCr6c |jjdtjt |j t |dy)zmDrops a collection. Args: name (str): The name of the collection to be dropped. r`FN)r0execute_nonquery_DROP_TABLE_QUERYr]rr,)r1r3s r4drop_collectionzSchema.drop_collection sB ))   $ $ ,.>t.D    r6c |s tdd|vrtjdt|d}t ||}|j |d}|t |tr|s tdd}|D]}||vstd|g} d |vr3|d } t | ts td | jd | fd |vr^|d } t | ttfs td | jd t | trtj| n| fd | f|d< |jjddd||S#t$rp} | jt k(r t#d| | jt$k(r|s1td|d| t| j&| j| Yd} ~ |Sd} ~ wwxYw)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)r2r3NInvalid value for 'validation'levelr2 Invalid option in 'validation': rInvalid value for 'level'r2Invalid value for 'schema' validationoptionsmysqlxcreate_collectionTlYour MySQL server does not support the requested operation. Please update to MySQL 8.0.19 or a later version Collection 'z' already exists)rwarningswarnDeprecationWarningrhr,r)dictrNrjjsondumpsr0rrerrnorrr msg) r1r3reuse_existingrkwargsrnfields valid_optionsoptionrrr2errs r4rzSchema.create_collection.s<"#?@ @ f  MM6"  $G_Nd+ ,0JJ!E  !j$/z&'GHH/M$ X.*-MfX+VWW XG*$"7+!%-*+FGG/0:%#H-!&3+6*+GHH .8.F 6*F".w 7F9  D    - --tV $ Dyy22'  yy11%*&tf,<='sww :C & Ds)E GA%F==Gc|s tdt|tr|s tdd}|D]}||vstd|g}d|vr3|d}t|ts td|j d|fd|vr^|d}t|ttfs td|j dt|trt j |n|f|j|d |fd } |jjd d d |y#t$rE} | jtk(r td| t| j| j| d} ~ wwxYw)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 rrrrrrr2rr)r2r3rrmodify_collection_optionsTrN)rr)rrNrjrrr,r0rrrrrr) r1r3rrrrrr2rrs r4modify_collectionzSchema.modify_collections("#?@ @*d+:"#CD D+   TF]*&)I&'RSS T j w'EeS)&'BCC NNGU+ , z !)FfsDk2&'CDD NN*4VT*BDJJv& jj$g.  @    - -5tV   @yy66'  #377CII6C ? @s4D E!AEE!)r:r!r3rNrLrMrLbool)rLzList[Collection])F)r3rNrrrrLrw)rLz List[Table])r3rNrrrrLr~)r3rNrrrrLrh)r3rNrLrM)FN) r3rNrrr%Optional[Dict[str, Union[str, Dict]]]rrrLrhr')r3rNrrrLrM)rOrPrQrRr5rGrorsr{rqrrrrr __classcell__)rYs@r4rVrVs%=(27 5 5*. 5  5,     " %<@ ]]]: ]  ]  ]@NRB@B@%JB@ B@r6rVc~eZdZdZddZdddZddZddZddZddZ dd Z dd Z dd Z dd Z dd ZddZy)rhzRepresents a collection of documents on a schema. Args: schema (mysqlx.Schema): The Schema object. name (str): The collection name. ctjt|jjt|j }|j j|dk(Sr[_COUNT_TABLES_QUERYr]rr(r3r,r0r^r_s r4rGzCollection.exists_in_databaseJ "(( 0A0A)BF4::DVW22371<>@@r6c|jdjd|j}|j}|jj |S)zReturns a Document matching the Document ID. Args: doc_id (str): Document ID Returns: mysqlx.DbDoc: The Document matching the Document ID. rr)rrr fetch_oner0fetch_active_result)r1rresultrs r4get_onezCollection.get_onezsM;',,T6:BBD  ,,. r6c`|jdjd|jS)zRemoves a Document matching the Document ID. Args: doc_id (str): Document ID Returns: mysqlx.Result: Result object. rr)rrr)r1rs r4 remove_onezCollection.remove_ones){{;',,T6:BBDDr6rr')rz Optional[str]rLr)rr rLr)rrNrLr)rrNrLrrLint)rrNrzDict[str, Any]rLr)rrNrLrM)rrNrzUnion[Dict, DbDoc]rL'Result')rrNrLr )rrNrLr)rOrPrQrRrGrrrrrrrrrrrrTr6r4rhrhsb=  /""(MM,:M 'M4 " SA"  Er6rhcHeZdZdZd dZd dZd dZd dZddZddZ d dZ y )rwzRepresents 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. ctjt|jjt|j }|j j|dk(Sr[rr_s r4rGzTable.exists_in_databaserr6c\t|g|}|jj|_|S)zCreates a new :class:`mysqlx.SelectStatement` object. Args: *fields: The fields to be retrieved. Returns: mysqlx.SelectStatement: SelectStatement object )rr0rrr1rrs r4selectz Table.select.t-f-''==?  r6c\t|g|}|jj|_|S)zCreates a new :class:`mysqlx.InsertStatement` object. Args: *fields: The fields to be inserted. Returns: mysqlx.InsertStatement: InsertStatement object )rr0rrrs r4insertz Table.insertrr6cZt|}|jj|_|S)zCreates a new :class:`mysqlx.UpdateStatement` object. Returns: mysqlx.UpdateStatement: UpdateStatement object )rr0rrr1rs r4updatez Table.updates) t$''==?  r6cZt|}|jj|_|S)zCreates a new :class:`mysqlx.DeleteStatement` object. Returns: mysqlx.DeleteStatement: DeleteStatement object .. versionchanged:: 8.0.12 The ``condition`` parameter was removed. )rr0rrrs r4deletez Table.deletes)t$''==?  r6cvtjt|jjt|j } |j j|}|S#t$rJ}|jtk(r1td|j d|jjd|d}~wwxYw)ziCounts the rows in the table. Returns: int: The total of rows in the table. zTable 'rrNrrs r4rz Table.counts !! T\\.. /1A$**1M  ""55c:C   yy,,&djj\* ))*!-   rctjt|jjt|j }|j j|dk(S)zDetermine if the underlying object is a view or not. Returns: bool: `True` if the underlying object is a view. r _COUNT_VIEWS_QUERYr]rr(r3r,r0r^r_s r4is_viewz Table.is_viewJ !''t||/@/@(A6$**CUV22371<rs:3" BB JI99   GF8 8 Q. 2aaH]@^]@@ EED`=N`=F=5=r6