hgdZddlmZddlZddlZddlmZmZmZm 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/dZ0GddZ1Gdde1Z2Gdde1Z3Gdde1Z4Gdde4Z5dS)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}ceZdZdZddZedd Zedd Zedd ZddZ ddZ ddZ ddZ d dZ eddd dZeddddZdS)!DatabaseObjectzProvides base functionality for database objects. Args: schema (mysqlx.Schema): The Schema object. name (str): The database object name. schemar namer"returnNonec||_t|tr|n||_|j|_|j|_dSN) _schema isinstancebytesdecode_name get_session_sessionget_connection _connection)selfr&r's S/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/mysqlx/crud.py__init__zDatabaseObject.__init__Vs\#) +5dE+B+BL$++--- %)\%=%=%?%? +/=+G+G+I+Ir!c|jS)z,:class:`mysqlx.Session`: The Session object.r2r5s r6sessionzDatabaseObject.session\s }r8c|jS)z*:class:`mysqlx.Schema`: The Schema object.r,r;s r6r&zDatabaseObject.schemaas |r8strc|jS)z&str: The name of this database object.r0r;s r6r'zDatabaseObject.namefs zr8rc|jS)z~Returns the underlying connection. Returns: mysqlx.connection.Connection: The connection object. )r4r;s r6r3zDatabaseObject.get_connectionks r8c|jS)zwReturns the session of this database object. Returns: mysqlx.Session: The Session object. r:r;s r6r1zDatabaseObject.get_sessionss }r8c|jS)z{Returns the Schema object of this database object. Returns: mysqlx.Schema: The Schema object. r>r;s r6 get_schemazDatabaseObject.get_schema{s |r8c|jS)zwReturns the name of this database object. Returns: str: The name of this database object. rAr;s r6get_namezDatabaseObject.get_names zr8rct)zVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. Raises: NotImplementedError: This method must be implemented. )NotImplementedErrorr;s r6exists_in_databasez!DatabaseObject.exists_in_databases "!r8z8.0.12z)Use 'exists_in_database()' method insteadc*|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. )rJr;s r6 am_i_realzDatabaseObject.am_i_reals&&(((r8zUse 'get_name()' method insteadc*|S)zReturns the name of this database object. Returns: str: The name of this database object. .. deprecated:: 8.0.12 Use ``get_name()`` method instead. )rGr;s r6who_am_izDatabaseObject.who_am_is}}r8N)r&r r'r"r(r))r(r!)r(r )r(r?)r(r)r(r)__name__ __module__ __qualname____doc__r7propertyr<r&r'r3r1rErGrJrrLrNr8r6r%r%NsWJJJJ XXX     " " " "ZEFF ) ) )GF )Z;<<   =<   r8r%ceZdZdZd!fd Zd"d Zd#d Z d$d%dZd&dZd$d%dZ d$d'dZ d$d(dZ d)dZ d*d+dZ d,d-d 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. r<r!r'r?r(r)cZ||_t||dSr+)r2superr7)r5r<r' __class__s r6r7zSchema.__init__s*%,  t$$$$$r8boolctt|j}|j|dkSzVerifies if this object exists in the database. Returns: bool: `True` if object exists in database. r )_COUNT_SCHEMAS_QUERYformatrr0r4execute_sql_scalarr5sqls r6rJzSchema.exists_in_databases= #))&*<*<==223771<|jdd|ji}|g}d}|D]_}|d|vrS t ||d}n&#t $rt ||d}YnwxYw||`|S)zoReturns a list of tables for this schema. Returns: `list`: List of Table objects. rdr&)TABLEVIEWrergr')r4rhr0rirsrkrl)r5rmtables object_typesrotables r6 get_tableszSchema.get_tabless ..~$*?UVV   % %C6{l**5!$L(9::EE!555!$F 44EEE5 e$$$ s A!! BBcpt||}|r#|std|S)zwReturns the table of the given name for this schema. Returns: mysqlx.Table: Table object. zTable does not exist)rsrJr)r5r'rrr}s r6ruzSchema.get_tablesC dD!!  ?++-- ?&'=>>> r8Viewcpt||}|r#|std|S)ztReturns the view of the given name for this schema. Returns: mysqlx.View: View object. zView does not exist)rrJr)r5r'rrviews r6get_viewzSchema.get_viewsC D$  >**,, >&'<=== r8rjcpt||}|r#|std|S)zReturns the collection of the given name for this schema. Returns: mysqlx.Collection: Collection object. zCollection does not exist)rjrJr)r5r'rrrps r6get_collectionzSchema.get_collectionsF  d++  D0022 D&'BCCCr8c |jdtt |jt |ddS)zmDrops a collection. Args: name (str): The name of the collection to be dropped. raFN)r4execute_nonquery_DROP_TABLE_QUERYr^rr0)r5r's r6drop_collectionzSchema.drop_collection s\ ))   $ $ ,,.>t.D.D         r8Nreuse_existing validation%Optional[Dict[str, Union[str, Dict]]]kwargsrc |stdd|vr"tjdt|d}t ||}|j|d}| t |tr|stdd}|D]}||vrtd|g} d |vrC|d } t | tstd | d | fd |vrs|d } t | ttfstd | d t | trtj | n| fd | f|d< |j ddd|ny#t$rl} | jt krt#d| | jt$kr|std|d| nt| j| j| Yd} ~ nd} ~ wwxYw|S)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'roptionsmysqlxcreate_collectionTlYour MySQL server does not support the requested operation. Please update to MySQL 8.0.19 or a later version Collection 'z' already exists)rwarningswarnDeprecationWarningrjr0r-dictr?rljsondumpsr4rrerrnorrr msg) r5r'rrrrpfields valid_optionsoptionrrr&errs r6rzSchema.create_collection.s< A"#?@@ @ f   M6"    $G_Nd++ ,0J!E!E  !j$// Iz I&'GHHH/M$ X X..*+Vf+V+VWWW/G*$$"7+!%--H*+FGGG/000:%%#H-!&3+66I*+GHHH .8.F.FR 6***F".w 7F9  D   - --tV     D D Dy222'  y111%*=t=== 'sw ::C  Ds E>> G4A"G//G4c|stdt|tr|stdd}|D]}||vrtd|g}d|vrC|d}t|tstd|d|fd|vrs|d}t|ttfstd|dt|trt j|n|f|j|d |fd } |j d d d |dS#t$r@} | j tkrtd| 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 rrrrrrr&rr)r&r'rrmodify_collection_optionsTrN)rr-rr?rlrrr0r4rrrrrr) r5r'rrrrrr&rrs r6modify_collectionzSchema.modify_collections ( A"#?@@ @*d++ E: E"#CDD D+   T TF]**&'R&'R'RSSS+ j w'EeS)) D&'BCCC NNGU+ , , , z ! !)FfsDk22 E&'CDDD NN*4VT*B*BNDJv&&&   j$g.   @   - -5tV       @ @ @y666'  #37CI66C ? @s#E F  ;FF )r<r!r'r?r(r)r(rZ)r(rb)F)r'r?rrrZr(rs)r(rw)r'r?rrrZr(r)r'r?rrrZr(rj)r'r?r(r))FN) r'r?rrZrrrrr(rjr+)r'r?rrr(r))rOrPrQrRr7rJrqrvr~rurrrrr __classcell__)rYs@r6rVrVsY%%%%%%====(27 5 5 5 5 5,                    " %<@ ]]]]]@NRB@B@B@B@B@B@B@B@B@r8rVcteZdZdZd$dZd%d&d Zd'd Zd(dZd)dZd*dZ d+dZ d,dZ d-d Z d-d!Z d.d"Zd/d#ZdS)0rjzRepresents a collection of documents on a schema. Args: schema (mysqlx.Schema): The Schema object. name (str): The collection name. r(rZctt|jjt|j}|j|dkSr\_COUNT_TABLES_QUERYr^rr,r'r0r4r_r`s r6rJzCollection.exists_in_databaseL "(( 0A)B)BF4:DVDVWW223771<>@@@r8c|dd|}|}|j|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_oner4fetch_active_result)r5rresultrs r6get_onezCollection.get_onezs\;'',,T6::BBDD   ,,... r8cx|dd|S)zRemoves a Document matching the Document ID. Args: doc_id (str): Document ID Returns: mysqlx.Result: Result object. rr)rrr)r5rs r6 remove_onezCollection.remove_ones2{{;'',,T6::BBDDDr8rr+)rrr(r)rr r(r)rr?r(r)rr?r(rr(r)rr?rrr(r)rr?r(r))rr?rrr(r)rr?r(r )rr?r(r)rOrPrQrRrJrrrrrrrrrrrrTr8r6rjrjs%====      / / / /""(MMMM4    " S S S SAAAA"     E E E E E Er8rjcJeZdZdZddZddZdd Zdd ZddZddZ ddZ dS)rszRepresents 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. r(rZctt|jjt|j}|j|dkSr\rr`s r6rJzTable.exists_in_databaserr8rr?rc\t|g|R}|j|_|S)zCreates a new :class:`mysqlx.SelectStatement` object. Args: *fields: The fields to be retrieved. Returns: mysqlx.SelectStatement: SelectStatement object )rr4rrr5rrs r6selectz Table.select4t-f---'==??  r8rrc\t|g|R}|j|_|S)zCreates a new :class:`mysqlx.InsertStatement` object. Args: *fields: The fields to be inserted. Returns: mysqlx.InsertStatement: InsertStatement object )rr4rrrs r6insertz Table.insertrr8rc`t|}|j|_|S)zCreates a new :class:`mysqlx.UpdateStatement` object. Returns: mysqlx.UpdateStatement: UpdateStatement object )rr4rrr5rs r6updatez Table.updates, t$$'==??  r8rc`t|}|j|_|S)zCreates a new :class:`mysqlx.DeleteStatement` object. Returns: mysqlx.DeleteStatement: DeleteStatement object .. versionchanged:: 8.0.12 The ``condition`` parameter was removed. )rr4rrrs r6deletez Table.deletes,t$$'==??  r8rcXtt|jjt|j} |j|}nI#t$r<}|j tkr&td|jd|jjd|d}~wwxYw|S)ziCounts the rows in the table. Returns: int: The total of rows in the table. zTable 'rrNrrs r6rz Table.counts !! T\. / /1A$*1M1M   "55c::CC   y,,,&-dj-- )---    rctt|jjt|j}|j|dkS)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,r'r0r4r_r`s r6is_viewz Table.is_viewL !''t|/@(A(A6$*CUCUVV223771<r s:32"""""" BBBBBBBBBBBBBBBB JIIIIIIIII9999999999                      GFFFFFFFFFFF8 8 Q. 2aaaaaaaaH]@]@]@]@]@^]@]@]@@ EEEEEEEED`=`=`=`=`=N`=`=`=F=====5=====r8