hdZddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZddlmZddlmZdd lmZmZdd lmZmZdHdZdIdZdJdZdJdZdKdZdLdZ dLdZ!dMdZ"dNdZ#dOd Z$dPd"Z%Gd#d$Z&Gd%d&Z'Gd'd(Z(Gd)d*Z)Gd+d,e)Z*Gd-d.e*Z+Gd/d0e*Z,Gd1d2e*Z-Gd3d4e*Z.Gd5d6e*Z/Gd7d8e*Z0Gd9d Z1Gd:d;Z2Gd<d=Z3Gd>d?e3Z4Gd@dAe3Z5GdBdCe5Z6GdDdEe6Z7GdFdGe5Z8dS)Qz%Implementation of the Result classes.) annotationsN)datetime timedelta)AnyCallableDictListOptionalSequenceTupleUnion)MYSQL_CHARACTER_SETS)DbDoc)decode_from_bytes deprecated)ConnectionType FieldTypescolumnColumnpayloadbytesreturnrct|dkrdS|tjkrt |ddS t j||S#t$r^}tj |tj | dYd}~dSd}~wwxYw)Nrhex) lenget_type ColumnTypeSTRINGrColumnProtoType converter_mapget_proto_typeKeyErrorsysstderrwriteencode)rrerrs U/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/mysqlx/result.py from_protobufr+/s 7||qt J--- "...,V-B-B-D-DEgNNN  C""" GNN5113444ttttts+A<< C$ACC$c|ddS)Nrrs r*bytes_from_protobufr/>s 3B3<floatcbt|dksJtjd|dS)NzN)r{r|r}NOT_NULL PRIMARY_KEY UNIQUE_KEY MULTIPLE_KEYAUTO_INCREMENTr-r0r*rrs'HKJLNNNr0rceZdZdZdS)DatetimeColumnFlagsrN)r{r|r}rr-r0r*r r IIIr0r ceZdZdZdS)UIntColumnFlagsrN)r{r|r}ZEROFILLr-r0r*r r HHHr0r ceZdZdZdS)DoubleColumnFlagsrNr{r|r}UNSIGNEDr-r0r*rrrr0rceZdZdZdS)FloatColumnFlagsrNrr-r0r*rrrr0rceZdZdZdS)BytesColumnFlagsrN)r{r|r} RIGHT_PADr-r0r*rrr r0rceZdZdZdZdZdS)BytesContentTyperrrN)r{r|r}rrrr-r0r*rrsH D CCCr0rceZdZdZ d6d7dZd8dZd9dZd9dZd9dZd9dZ d9dZ e d8dZ e d8dZ e d8dZe d8dZe d8d Ze d:d!Ze d:d"Ze d:d#Ze d8d$Ze d8d%Zd8d&Zd8d'Zd8d(Zd8d)Zd8d*Zd:d+Zd:d,Zd:d-Zd:d.Zd8d/Zd8d0Z d;d2Z!dR4R O    O N     ,  r0rclt|j|j|jt|jdS)N)rrrr%)rr0r*r-r5rs r*rzColumn.__str__s: N,T[))      r0cb|jtjkrtj|_nx|jtjkrtj|_nQ|jtjkrtj|_n*|jrtj|_ntj |_|j dz|_ dS)z Map bytes.rN) r6rrrr0rrr9rr r5r8rs r* _map_byteszColumn._map_bytess  !1!: : :'0DNN  #3#8 8 8'_DNN  #3#7 7 7'^DNN _ /'-DNN'.DN+/r0c|jdkrtj|_dS|jt jzdkrtj|_dS|jdkrtj|_dStd)z Map datetime.r_rrz#Datetime mapping scenario unhandledN) r2rrr0r5r rrrArs r* _map_datetimezColumn._map_datetimesj <2  '_DNNN [.8 81 < <'1DNNN \R  '0DNNNBCC Cr0c|jdkrtj|_nh|jdkrtj|_nK|jdkrtj|_n.|jdkrtj|_ntj|_d|_dS)z Map int type.r3rrrTN) r2rrr0rrrrr7rs r* _map_int_typezColumn._map_int_type's} <1  '/DNN \Q  '0DNN \Q  '1DNN \R  '^DNN'.DN"r0c.|jdkrtj|_nh|jdkrtj|_nK|jdkrtj|_n.|jdkrtj|_ntj|_|jdz|_ dS)zMap uint type.rrr:r_rN) r2rrr0rrrrr5r=rs r*_map_uint_typezColumn._map_uint_type5s <1  '/DNN \Q  '0DNN \Q  '1DNN \R  '^DNN'.DN+/r0cf|jtjkr|dS|jtjkr|dS|jtjkr0tj|_|j tj zdk|_ dS|jtj kr0tj |_|j tj zdk|_ dS|jtjkr0tj|_|j tj zdk|_ dS|jtjkr|dS|jtjkrtj|_dS|jtjkr|dS|jtjkrtj|_dS|jtjkrtj|_dS|jtjkrtj|_dSt/d|j)z Map type.rzUnknown column type N)r/r!rrFrrHrrr0r5rr_is_number_signedrrrrBrrrDrrrrArs r*r>zColumn._map_typeCs  3 3 3     !5 5 5    ! ! ! ! !  !6 6 6'-DN&*k4D4M&MRS%SD " " "  !8 8 8'/DN&*k4D4M&MRS%SD " " "  !7 7 7'.DN&*k4D4M&MRS%SD " " "  !6 6 6 OO       !5 5 5'_DNNN  !9 9 9     !4 4 4'^DNNN  !5 5 5'_DNNN  !4 4 4'^DNNNFD4DFFGG Gr0c|jS)z@str: The schema name. .. versionadded:: 8.0.12 r*rs r* schema_namezColumn.schema_namea |r0c|jp|jS)z?str: The table name. .. versionadded:: 8.0.12 r.r-rs r* table_namezColumn.table_namei #2t{2r0c|jp|jS)z@str: The table label. .. versionadded:: 8.0.12 r-r.rs r* table_labelzColumn.table_labelq {2d22r0c|jp|jS)z@str: The column name. .. versionadded:: 8.0.12 r,r+rs r* column_namezColumn.column_namey "0dj0r0c|jp|jS)zAstr: The column label. .. versionadded:: 8.0.12 r+r,rs r* column_labelzColumn.column_label z0T00r0c|jS)z@int: The column type. .. versionadded:: 8.0.12 r0rs r*typez Column.type ~r0c|jS)zBint. The column length. .. versionadded:: 8.0.12 r2rs r*r!z Column.lengthrNr0c|jS)zMint: The column fractional digits. .. versionadded:: 8.0.12 r4rs r*r$zColumn.fractional_digits &&r0c|jS)zCstr: The collation name. .. versionadded:: 8.0.12 r;rs r*collation_namezColumn.collation_name ##r0c|jS)zGstr: The character set name. .. versionadded:: 8.0.12 r<rs r*character_set_namezColumn.character_set_name ''r0c|jS)zUReturns the schema name. Returns: str: The schema name. rLrs r*get_schema_namezColumn.get_schema_namerNr0c|jp|jS)zSReturns the table name. Returns: str: The table name. rPrs r*get_table_namezColumn.get_table_namerRr0c|jp|jS)zUReturns the table label. Returns: str: The table label. rTrs r*get_table_labelzColumn.get_table_labelrVr0c|jp|jS)zUReturns the column name. Returns: str: The column name. rXrs r*get_column_namezColumn.get_column_namerZr0c|jp|jS)zWReturns the column label. Returns: str: The column label. r\rs r*get_column_labelzColumn.get_column_labelr^r0c|jS)zaReturns the column proto type. Returns: int: The column proto type. )r/rs r*r#zColumn.get_proto_types r0c|jS)zUReturns the column type. Returns: int: The column type. r`rs r*rzColumn.get_typerbr0c|jS)zYReturns the column length. Returns: int: The column length. rdrs r* get_lengthzColumn.get_lengthrNr0c|jS)zoReturns the column fractional digits. Returns: int: The column fractional digits. rfrs r*get_fractional_digitszColumn.get_fractional_digitsrgr0c|jS)z[Returns the collation name. Returns: str: The collation name. rirs r*get_collation_namezColumn.get_collation_namerkr0c|jS)zcReturns the character set name. Returns: str: The character set name. rmrs r*get_character_set_namezColumn.get_character_set_nameror0rc|jS)zxReturns `True` if is a number signed. Returns: bool: Returns `True` if is a number signed. )r7rs r*is_number_signedzColumn.is_number_signed  ""r0Union[bool, int]c|jS)zfReturns `True` if is padded. Returns: bool: Returns `True` if is padded. )r8rs r* is_paddedzColumn.is_paddeds r0c|jS)zdReturns `True` if is bytes. Returns: bool: Returns `True` if is bytes. )r:rs r*is_byteszColumn.is_bytesrbr0) NNNNNNNNNNN)rrCrrrrrrrrrrr rr!r"r#r"r$r"r%r"r&r"rrrrrrrrrr)$r{r|r}__doc__rrrBrDrFrHr>rrMrQrUrYr]rar!r$rjrnrqrsrurwryr#rr}rrrrrrr-r0r*rrs3,"& $#(,"'+ $#'+/#&*4 4 4 4 4 l     * * * * D D D D # # # # * * * *HHHH<X333X3333X3111X1111X1XX'''X'$$$X$(((X(3333333311111111    ''''$$$$((((####r0cTeZdZdZddZdd ZddZedddZdS)RowzRepresents a row element returned from a SELECT query. Args: resultset (mysqlx.SqlResult or mysqlx.RowResult): The result set. fields (`list`): The list of fields. resultset!Union[BufferingResult, RowResult]fieldsSequence[FieldTypes]rrc"||_||_dSr)_fields _resultset)rrrs r*rz Row.__init__*s.4 =Fr0rc*t|jSr)reprrrs r*__repr__z Row.__repr__0sDL!!!r0indexUnion[int, str]rc4t|tr|j|n|}|dkr(t|trt d|d|t |jks|dkrtd|j|S)z\Returns the value of a column by name or index. .. versionchanged:: 8.0.12 r Column name ' ' not foundrzIndex out of range)rBrrindex_ofrArr IndexError)rr int_indexs r* __getitem__zRow.__getitem__3s 8B%7M7MXDO,,U333SX ??z%55??U???@@ @ DL)) ) )Y]]122 2|I&&r08.0.12 str_indexc|j|}|t|jkrt d|dkrt d|dt |j|S)zReturns the value using the column name. Args: str_index (str): The column name. .. deprecated:: 8.0.12 zArgument out of rangerrr)rrrrrrAr)rrrs r* get_stringzRow.get_string?suO,,Y77 DL)) ) )455 5 ??CYCCCDD D4< *+++r0N)rrrrrrr)rrrr)rrrr) r{r|r}rrrrrrr-r0r*rr"sGGGG """" ' ' ' 'Z , , , , , ,r0rcZeZdZdZddZddZdd Zdd ZddZd dZ d!dZ d"dZ d#dZ dS)$ BaseResultzProvides base functionality for result objects. Args: connection (mysqlx.connection.Connection): The Connection object. connectionrrrc||_d|_d|_d|_g|_g|_| d|_dS|j|_|dSNFrr) _connection_closed_rows_affected _generated_id_generated_ids _warnings _protocolprotocolfetch_active_result)rrs r*rzBaseResult.__init__Wsc+5" #$"$)+;=  !DNNN'0DN  * * , , , , ,r0rCc|jS)zReturns the number of affected items for the last operation. Returns: int: The number of affected items. rrs r*get_affected_items_countz#BaseResult.get_affected_items_counterr0 List[Dict[str, Union[int, str]]]c|jS)zZReturns the warnings. Returns: `list`: The list of warnings. )rrs r* get_warningszBaseResult.get_warningsmrbr0c*t|jS)zcReturns the number of warnings. Returns: int: The number of warnings. )rrrs r*get_warnings_countzBaseResult.get_warnings_countus 4>"""r0rrc||_dS)z Sets if resultset fetch is done.N)rrrs r* set_closedzBaseResult.set_closed}s  r0levelcodemsgrcB|j|||ddS)zAppend a warning. Args: level (int): The warning level. code (int): The warning code. msg (str): The warning message. )rrrN)rrW)rrrrs r*append_warningzBaseResult.append_warnings* tCHHIIIIIr0 generated_ids List[int]c||_dS)zSets the generated ids.Nr)rrs r*set_generated_idszBaseResult.set_generated_idss+r0 generated_idc||_dS)zSets the generated insert id.Nr)rrs r*set_generated_insert_idz"BaseResult.set_generated_insert_ids)r0totalc||_dS)z!Sets the number of rows affected.Nr)rrs r*set_rows_affectedzBaseResult.set_rows_affecteds#r0Nrrrrr)rrrrrr)rrCrrCrrrr)rrrr)rrCrr)rrCrr) r{r|r}rrrrrrrrrrr-r0r*rrPs - - - -########JJJJ,,,,****$$$$$$r0rceZdZdZ ddfd Zdd Zed ddZed ddZddZ xZ S)ResultaAllows retrieving information about non query operations performed on the database. Args: connection (mysqlx.connection.Connection): The Connection object. ids (`list`): A list of IDs. NrOptional[ConnectionType]idsOptional[List[int]]rrct|||_||j|dSdSr)superr_idsr close_result)rrrrs r*rzResult.__init__sO $$$),  !   ) )$ / / / / / " !r0rCc|jS)zjReturns the last insert id auto generated. Returns: int: The last insert id. rrs r*get_autoincrement_valuezResult.get_autoincrement_value !!r0rr"c^|jt|jdkrdS|jdS)zdReturns ID of the last document inserted into a collection. .. deprecated:: 8.0.12 Nr)rrrs r*get_document_idzResult.get_document_ids/ 9 DI! 3 34y|r0c|jS)zIReturns the generated insert id. .. deprecated:: 8.0.12 rrs r*get_generated_insert_idzResult.get_generated_insert_idrr0rc|jS)zReturns the generated ids.rrs r*get_generated_idszResult.get_generated_idss ""r0)NN)rrrrrrr)rr")rr) r{r|r}rrrrrrr __classcell__rs@r*rrs04#' 0 0 0 0 0 0 0""""ZZ""""########r0rceZdZdZdfd Zdd Zedd Zdd Zd dZ d!dZ d"dZ d#dZ d$dZ d%dZd%dZxZS)&BufferingResultzProvides base functionality for buffering result objects. Args: connection (mysqlx.connection.Connection): The Connection object. ids (`list`): A list of IDs. rrrrct|g|_d|_d|_g|_d|_d|_|dSr) rr_columns _has_data_has_more_results_items _page_size _position _init_result)rrrs r*rzBufferingResult.__init__s^ $$$&( $',/1    r0rrCUnion[Row, DbDoc]c|j|Sr)r)rrs r*rzBufferingResult.__getitem__s{5!!r0c*t|jS)zint: The total of items.)rrrs r*countzBufferingResult.counts4;r0c|j||_t|jdk|_g|_d|_d|_|j|jr|nddS)zInitialize the result.rrrN) rget_column_metadatarr_has_more_datarrrset_active_resultrs r*rzBufferingResult._init_resultso(<|jrdS|dS)zuFetch one item. Returns: :class:`mysqlx.Row` or :class:`mysqlx.DbDoc`: one result item. NF)rrrs r* fetch_onezBufferingResult.fetch_one s% < 4u%%%r0List[Union[Row, DbDoc]]c> |sn|jS)zFetch all items. Returns: `list`: The list of items of :class:`mysqlx.DbDoc` or :class:`mysqlx.Row`. )rrrs r* fetch_allzBufferingResult.fetch_all+s- &&((  {r0rc||_dS)zdSets if result has data. Args: flag (bool): `True` if result has data. Nrrs r* set_has_datazBufferingResult.set_has_data7s r0c||_dS)zfSets if has more results. Args: flag (bool): `True` if has more results. N)rrs r*set_has_more_resultsz$BufferingResult.set_has_more_results?s "&r0r)rrCrrrr)rrrrr)rrrrC)rr)rr r)r{r|r}rrrrrrrrrrr rrrrs@r*rrs""""   X RRRR&$     & & & &    &&&&&&&&r0rc2eZdZdZeddZddZdS) RowResultzAllows traversing the Row objects returned by a Table.select operation. Args: connection (mysqlx.connection.Connection): The Connection object. r List[Column]c|jS)z`list`: The list of columns.rrs r*columnszRowResult.columnsOs }r0c|jS)zReturns the list of columns. Returns: `list`: The list of columns. .. versionadded:: 8.0.12 rrs r* get_columnszRowResult.get_columnsTs }r0N)rr)r{r|r}rrrrr-r0r*rrHsR Xr0rc*eZdZdZd dZd dZd dZdS) SqlResultzRepresents a result from a SQL statement. Args: connection (mysqlx.connection.Connection): The Connection object. rrCc|jS)zReturns the identifier for the last record inserted. Returns: str: The identifier of the last record inserted. rrs r*rz!SqlResult.get_autoincrement_valuefrr0rcN|jrdSd|_|dS)zjProcess the next result. Returns: bool: Returns `True` if the fetch is done. FT)rrrrs r* next_resultzSqlResult.next_resultns2 < 5!& tr0c|jS)zReturns True if result has data. Returns: bool: Returns `True` if result has data. .. versionadded:: 8.0.12 r rs r*has_datazSqlResult.has_datazs ~r0Nrr)r{r|r}rrrrr-r0r*rr_sZ """"    r0rc$eZdZdZdfd ZxZS) DocResultzAllows traversing the DbDoc objects returned by a Collection.find operation. Args: connection (mysqlx.connection.Connection): The Connection object. rrrrct|}|dStt|dS)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.DbDoc`: A `DbDoc` object. Nr)rrrr)rrrrs r*rzDocResult._read_itemsAgg  )) ;4&s1v..///r0)rrrr)r{r|r}rrrrs@r*r!r!sG 0 0 0 0 0 0 0 0 0 0r0r!)rrrrrr)rrrr)rrrr1)rrrr<)rrrrC)rrrrU)rrrr[)rrrr)rrrr)9r __future__rrbr5r%rrtypingrrrr r r r r charsetsrdbdocrhelpersrrtypesrrr+r/r7r;rMrRrTrZrirrrwryrr!rrr r rrrrrrrrrrrr!r-r0r*r)s:,+"""""" ((((((((NNNNNNNNNNNNNNNNNNNN******22222222--------     ++++ ++++ ((((,         *!3!3!3!3HHHHH*Ba a a a a a a a H8$$$$$$$$N%+k {{{eeeeeeeeP +,+,+,+,+,+,+,+,\E$E$E$E$E$E$E$E$P0#0#0#0#0#Z0#0#0#fz&z&z&z&z&jz&z&z&z.##### ###L0000000000r0