@idZddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZddlmZddlmZdd lmZmZdd lmZmZddZd>dZd?dZd@dZ d@dZ!dAdZ"dBdZ#dCdZ$dDdZ%GddZ&GddZ'GddZ(GddZ)Gdd e)Z*Gd!d"e*Z+Gd#d$e*Z,Gd%d&e*Z-Gd'd(e*Z.Gd)d*e*Z/Gd+d,e*Z0Gd-d Z1Gd.d/Z2Gd0d1Z3Gd2d3e3Z4Gd4d5e3Z5Gd6d7e5Z6Gd8d9e6Z7Gd:d;e5Z8y)Ez%Implementation of the Result classes.) annotationsN)datetime timedelta)AnyCallableDictListOptionalSequenceTupleUnion)MYSQL_CHARACTER_SETS)DbDoc)decode_from_bytes deprecated)ConnectionType FieldTypesColumnct|dk(ry|jtjk(rt |ddS t j |j|S#t$rY}tjj|tjj|jdYd}~yd}~wwxYw)Nrhex) lenget_type ColumnTypeSTRINGrColumnProtoType converter_mapget_proto_typeKeyErrorsysstderrwriteencode)columnpayloaderrs G/var/www/html/aiagents/fe/lib/python3.12/site-packages/mysqlx/result.py from_protobufr)/s 7|q J--- "..,,V-B-B-DEgNN  C5" GNN5124s%A&& C/ACCc |ddS)Nrr&s r(bytes_from_protobufr->s 3B<cTt|dk(sJtjd|dS)NzA;2A;; BBcg}d}t|dtr|dn t|d}|dd}|D]}t|tr|n t|}|dzdz }|dz}|dkrE|j||dkr|j|Z|dk(rd}nC|dk(rd}n:t d |dk(r d}|dk(sJn|dk(r d}|dk(sJnt d |t j ||| fS) Nrrr0 z Invalid BCDz Invalid BCD: )r>r?r@rRr=decimalDecimal)r&digitssignscalerErFhigh_bcdlow_bcds r(decimal_from_protobufrcs$ F D$WQZ5GAJ3wqz?EabkG9!$,t#d)4KA%+ d? MM( #~ g&DD //  Dd? "?   Dd? "? }XJ78 8396 ??D&5&1 22r.c d}d}d}d}t|\}}t|\}}t|\}} t|\}}t|\}}t|\}}t|\}}t|||||||S#t$rYwxYw)Nr)rIr=r)r&hourminutessecondsusecondsyearmonthdays r(datetime_from_protobufrls DGGH/8MD'09NE7.w7LC 3G< g6w?6w?7@' D%dGWh GG    s8A== B B cd}d}d}d}|ddk(}|dd} t|\}}t|\}}t|\}}t|\}}|r|r|dz}n|r|dz}n|r|dz}n|r|dz}t||||S#t$rY;wxYw)Nrrr)hoursrfrg microseconds)rIr=r)r&rerfrgrhnegates r(time_from_protobufrqs DGGH QZ1_FabkG 3G< g6w?6w?7@'  BJD  rMG  rMG  NH GW8     s8A?? B  B ceZdZdZy) Collations!N)__name__ __module__ __qualname__UTF8_GENERAL_CIr+r.r(rsrssOr.rsc(eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%e&d-d$Z'e&d.d%Z(e&d/d&Z)e&d/d'Z*e&d/d(Z+e&d/d)Z,e&d/d*Z-e&d/d+Z.y,)0rrr0r<r7 rY rZr[rX rt"#cXt|jD]\}}||k(s |cSyN)varsitems)clsneedlekeyvalues r( to_stringzColumnType.to_strings1s)//+ JC  r.c8t||jdSr)getattrupper)rrs r( from_stringzColumnType.from_stringssCIIK..r.c6||j|jfvSr)CHARVARCHARrcol_types r(is_charzColumnType.is_chars" HH KK   r.c6||j|jfvSr)BINARY VARBINARYrs r( is_binaryzColumnType.is_binary%s" JJ MM   r.cb||j|j|j|jfvSr)TEXTTINYTEXT MEDIUMTEXTLONGTEXTrs r(is_textzColumnType.is_text,s0 HH LL NN LL    r.cx||j|j|j|j|jfvSr)REALDOUBLEFLOATDECIMALNUMERICrs r( is_decimalszColumnType.is_decimals5s7 HH JJ II KK KK    r.c||j|j|j|j|j|j fvSr)BITTINYINTSMALLINT MEDIUMINTINTBIGINTrs r( is_numericzColumnType.is_numeric?s> GG KK LL MM GG JJ    r.c6||j|jfvSr)SETENUMrs r( is_finite_setzColumnType.is_finite_setJs" GG HH   r.N)rrreturn Optional[str])rstrrr)rr?rbool)/rurvrwrrrrrrrrrrrJSONrBYTESTIMEDATEDATETIME TIMESTAMPrrGEOMETRYXMLYEARrrrrTINYBLOBBLOB MEDIUMBLOBLONGBLOBrrrr classmethodrrrrrrrrr+r.r(rrsO CGHI C F D EGG F D F E D DHI C DH C D DG FIH DJHH DJH //            r.rcteZdZUdZdZdZdZdZdZdZ dZ d Z d Z d Z eeeeeee eeeeeeee ee ee ee ei Zd ed <y)rrrzr|r}r<rYrZrXrrrz!Dict[int, Callable[[bytes], Any]]rN)rurvrwSINTUINTrrrrrrrrrrPrNr-rlrqr4r8rUrcr__annotations__r+r.r(rrRs D D F E E DH C D CG # " "(   "$ !  !& 8M4 r.rcTeZdZddZddZeddZejd dZy) Flagsci|_i|_|jjj D]F\}}|j drt |ts)||j|<||j|<H||_y)N__) _allowed_flags _flag_names __class____dict__r startswithr>r?_value)selfrrvals r(__init__zFlags.__init__osz.0+-//557 ,HC~~d##s#+.##C((+  %  , ! r.cd}g}|j}tddD].}|dz}||z}|s |j|j|0dj|S#t$rat j j|jt j j|jjYwxYw)Nrr?,) rrangerRrr r!r"r#rrjoin)rmask flag_namesr_flags r(__str__z Flags.__str__zs  q" CA QJD4R>R4R O     OO NN       ,  r.ct|j|j|jt|jdS)N)rr/r0r7)rrrrr$rs r(rzColumn.__str__s5 NN,,T[[)    r.c|jtjk(rtj|_n|jtj k(rtj |_nj|jtj k(rtj |_n7|jrtj|_ntj|_|jdz|_ y)z Map bytes.rN) r%rrrrrrr(rrr$r'rs r( _map_byteszColumn._map_bytess   !1!:!: :'00DN   #3#8#8 8'__DN   #3#7#7 7'^^DN __'--DN'..DN++/r.c|jdk(rtj|_y|jt j zdkDrtj |_y|jdk\rtj|_ytd)z Map datetime.rYrrz#Datetime mapping scenario unhandledN) r!rrrr$rrrr=rs r( _map_datetimezColumn._map_datetimesc <<2 '__DN [[.88 81 <'11DN \\R '00DNBC Cr.c|jdkrtj|_d|_y|jdkrtj|_d|_y|jdkrtj |_d|_y|jdkrtj |_d|_ytj|_d|_y)z Map int type.r0r}r~rTN) r!rrrrrrrr&rs r( _map_int_typezColumn._map_int_type's <<1 '//DN#\\Q '00DN# \\Q '11DN # \\R '^^DN#(..DN"r.c~|jdkrtj|_n|jdkrtj|_n_|jdkrtj |_n:|jdkrtj |_ntj|_|jdz|_ y)zMap uint type.r{r|r7rYrN) r!rrrrrrrr$r,rs r(_map_uint_typezColumn._map_uint_type5s <<1 '//DN \\Q '00DN \\Q '11DN \\R '^^DN'..DN++/r.cJ|jtjk(r|jy|jtjk(r|j y|jtj k(r;tj |_|jtjzdk(|_ y|jtjk(r;tj|_|jtjzdk(|_ y|jtjk(r;tj|_|jtjzdk(|_ y|jtjk(r|j!y|jtj"k(rtj"|_y|jtj$k(r|j'y|jtj(k(rtj(|_y|jtj*k(rtj*|_y|jtj,k(rtj,|_yt/d|j)z Map type.rzUnknown column type N)rrrr@rrBrrrr$rr _is_number_signedrrrr<rrr>rrrr=rs r(r-zColumn._map_typeCs   33 3      !5!5 5    !   !6!6 6'--DN&*kk4D4M4M&MRS%SD "   !8!8 8'//DN&*kk4D4M4M&MRS%SD "   !7!7 7'..DN&*kk4D4M4M&MRS%SD "   !6!6 6 OO    !5!5 5'__DN   !9!9 9      !4!4 4'^^DN   !5!5 5'__DN   !4!4 4'^^DN3D4D4D3EFG Gr.c|jS)z@str: The schema name. .. versionadded:: 8.0.12 rrs r( schema_namezColumn.schema_namea ||r.c6|jxs |jS)z?str: The table name. .. versionadded:: 8.0.12 rrrs r( table_namezColumn.table_namei ##2t{{2r.c6|jxs |jS)z@str: The table label. .. versionadded:: 8.0.12 rrrs r( table_labelzColumn.table_labelq {{2d222r.c6|jxs |jS)z@str: The column name. .. versionadded:: 8.0.12 rrrs r( column_namezColumn.column_namey ""0djj0r.c6|jxs |jS)zAstr: The column label. .. versionadded:: 8.0.12 rrrs r( column_labelzColumn.column_label zz0T000r.c|jS)z@int: The column type. .. versionadded:: 8.0.12 rrs r(typez Column.type ~~r.c|jS)zBint. The column length. .. versionadded:: 8.0.12 r!rs r(r4z Column.lengthrHr.c|jS)zMint: The column fractional digits. .. versionadded:: 8.0.12 r#rs r(r6zColumn.fractional_digits &&&r.c|jS)zCstr: The collation name. .. versionadded:: 8.0.12 r*rs r(collation_namezColumn.collation_name ###r.c|jS)zGstr: The character set name. .. versionadded:: 8.0.12 r+rs r(character_set_namezColumn.character_set_name '''r.c|jS)zUReturns the schema name. Returns: str: The schema name. rFrs r(get_schema_namezColumn.get_schema_namerHr.c6|jxs |jS)zSReturns the table name. Returns: str: The table name. rJrs r(get_table_namezColumn.get_table_namerLr.c6|jxs |jS)zUReturns the table label. Returns: str: The table label. rNrs r(get_table_labelzColumn.get_table_labelrPr.c6|jxs |jS)zUReturns the column name. Returns: str: The column name. rRrs r(get_column_namezColumn.get_column_namerTr.c6|jxs |jS)zWReturns the column label. Returns: str: The column label. rVrs r(get_column_labelzColumn.get_column_labelrXr.c|jS)zaReturns the column proto type. Returns: int: The column proto type. )rrs r(rzColumn.get_proto_types r.c|jS)zUReturns the column type. Returns: int: The column type. rZrs r(rzColumn.get_typer\r.c|jS)zYReturns the column length. Returns: int: The column length. r^rs r( get_lengthzColumn.get_lengthrHr.c|jS)zoReturns the column fractional digits. Returns: int: The column fractional digits. r`rs r(get_fractional_digitszColumn.get_fractional_digitsrar.c|jS)z[Returns the collation name. Returns: str: The collation name. rcrs r(get_collation_namezColumn.get_collation_namerer.c|jS)zcReturns the character set name. Returns: str: The character set name. rgrs r(get_character_set_namezColumn.get_character_set_namerir.c|jS)zxReturns `True` if is a number signed. Returns: bool: Returns `True` if is a number signed. )r&rs r(is_number_signedzColumn.is_number_signed  """r.c|jS)zfReturns `True` if is padded. Returns: bool: Returns `True` if is padded. )r'rs r( is_paddedzColumn.is_paddeds r.c|jS)zdReturns `True` if is bytes. Returns: bool: Returns `True` if is bytes. )r)rs r(is_byteszColumn.is_bytesr\r.) NNNNNNNNNNN)rr?r.rr/rr0rr1rr2rr3rr4 Optional[int]r5rr6rr7rr8rrrrrrrrrrzUnion[bool, int])$rurvrw__doc__rrr<r>r@rBr-rrGrKrOrSrWr[r4r6rdrhrkrmrorqrsrrrwryr{r}rrrr+r.r(rrs,"& $#(,"'+ $#'+/#&*4 4 4  4  4 & 4 4 %4 4 !4 )4 4 $4  4 l  * D # *H<33331111''$$((3311 '$(#r.cReZdZdZ ddZd dZd dZedd dZy) 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. c ||_||_yr)_fields _resultset)r resultsetfieldss r(rz Row.__init__*s.4 =Fr.c,t|jSr)reprrrs r(__repr__z Row.__repr__0sDLL!!r.ct|tr|jj|n|}|dk(rt|trt d|d|t |j k\s|dkr td|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)r>rrindex_ofr=rr IndexError)rindex int_indexs r( __getitem__zRow.__getitem__3s{ 8B%7MDOO,,U3SX ?z%5}UG;?@ @ DLL) )Y]12 2||I&&r.8.0.12c|jj|}|t|jk\r t d|dk(rt 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)rrrrrr=r)r str_indexrs r( get_stringzRow.get_string?sdOO,,Y7 DLL) )45 5 ?}YK{CD D4<< *++r.N)rz!Union[BufferingResult, RowResult]rzSequence[FieldTypes]rrr)rzUnion[int, str]rr)rrrr) rurvrwrrrrrrr+r.r(rr"sKG:GDXG G " ' , ,r.rcXeZdZdZd dZd dZddZd dZddZddZ ddZ dd Z dd Z y ) BaseResultzProvides base functionality for result objects. Args: connection (mysqlx.connection.Connection): The Connection object. c||_d|_d|_d|_g|_g|_|d|_y|j|_|jyNFrr) _connection_closed_rows_affected _generated_id_generated_ids _warnings _protocolprotocolfetch_active_result)r connections r(rzBaseResult.__init__WsV+5" #$"$)+;=  !DN'00DN  * * ,r.c|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_counterr.c|jS)zZReturns the warnings. Returns: `list`: The list of warnings. )rrs r( get_warningszBaseResult.get_warningsmr\r.c,t|jS)zcReturns the number of warnings. Returns: int: The number of warnings. )rrrs r(get_warnings_countzBaseResult.get_warnings_countus 4>>""r.c||_y)z Sets if resultset fetch is done.N)rrrs r( set_closedzBaseResult.set_closed}s  r.cB|jj|||dy)zAppend a warning. Args: level (int): The warning level. code (int): The warning code. msg (str): The warning message. )levelcodemsgN)rrR)rrrrs r(append_warningzBaseResult.append_warnings tCHIr.c||_y)zSets the generated ids.Nr)r generated_idss r(set_generated_idszBaseResult.set_generated_idss +r.c||_y)zSets the generated insert id.Nr)r generated_ids r(set_generated_insert_idz"BaseResult.set_generated_insert_ids )r.c||_y)z!Sets the number of rows affected.Nr)rtotals r(set_rows_affectedzBaseResult.set_rows_affecteds #r.Nrrrrr)rz List[Dict[str, Union[int, str]]]rrrr)rr?rr?rrrr)r List[int]rr)rr?rr)rr?rr) rurvrwrrrrrrrrrrr+r.r(rrPs5 -##J,*$r.rceZdZdZ d d fd Zd dZedd dZedd dZd dZ 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. cnt||||_||jj |yyr)superr_idsr close_result)rridsrs r(rzResult.__init__s8 $),  !    ) )$ / "r.c|jS)zjReturns the last insert id auto generated. Returns: int: The last insert id. rrs r(get_autoincrement_valuezResult.get_autoincrement_value !!!r.rcj|jt|jdk(ry|jdS)zdReturns ID of the last document inserted into a collection. .. deprecated:: 8.0.12 Nr)rrrs r(get_document_idzResult.get_document_ids. 99 DII! 3yy|r.c|jS)zIReturns the generated insert id. .. deprecated:: 8.0.12 rrs r(get_generated_insert_idzResult.get_generated_insert_idrr.c|jS)zReturns the generated ids.rrs r(get_generated_idszResult.get_generated_idss"""r.)NN)rzOptional[ConnectionType]rzOptional[List[int]]rrr)rr)rr) rurvrwrrrrrrr __classcell__rs@r(rrsm04#' 0, 0! 0  0"""#r.rc~eZdZdZd fd ZddZeddZddZddZ ddZ ddZ dd Z dd Z dd Zdd ZxZS)BufferingResultzProvides base functionality for buffering result objects. Args: connection (mysqlx.connection.Connection): The Connection object. ids (`list`): A list of IDs. ct||g|_d|_d|_g|_d|_d|_|jyr) rr_columns _has_data_has_more_results_items _page_size _position _init_result)rrrs r(rzBufferingResult.__init__sG $&( $',/1    r.c |j|Sr)r)rrs r(rzBufferingResult.__getitem__s{{5!!r.c,t|jS)zint: The total of items.)rrrs r(countzBufferingResult.counts4;;r.c|jj||_t|jdkD|_g|_d|_d|_|jj|jr|ydy)zInitialize the result.rrrN) rget_column_metadatarr_has_more_datarrrset_active_resultrs r(rzBufferingResult._init_resultsf((<|jry|jdS)zuFetch one item. Returns: :class:`mysqlx.Row` or :class:`mysqlx.DbDoc`: one result item. NF)rrrs r( fetch_onezBufferingResult.fetch_one s <<u%%r.c@ |js |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%&&({{r.c||_y)zdSets if result has data. Args: flag (bool): `True` if result has data. Nrrs r( set_has_datazBufferingResult.set_has_data7s r.c||_y)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 "&r.r)rr?rzUnion[Row, DbDoc]rr)rrrOptional[Union[Row, DbDoc]]r)rrrr?)rr)rzList[Union[Row, DbDoc]]r)rurvrwrrrrrrrrrrrrrrrs@r(rrsO"  R&$  & &r.rc*eZdZdZeddZddZy) RowResultzAllows traversing the Row objects returned by a Table.select operation. Args: connection (mysqlx.connection.Connection): The Connection object. c|jS)z`list`: The list of columns.rrs r(columnszRowResult.columnsOs}}r.c|jS)zReturns the list of columns. Returns: `list`: The list of columns. .. versionadded:: 8.0.12 rrs r( get_columnszRowResult.get_columnsTs}}r.N)rz List[Column])rurvrwrrrr r+r.r(rrHs  r.rc(eZdZdZddZddZddZy) SqlResultzRepresents a result from a SQL statement. Args: connection (mysqlx.connection.Connection): The Connection object. c|jS)zReturns the identifier for the last record inserted. Returns: str: The identifier of the last record inserted. rrs r(rz!SqlResult.get_autoincrement_valuefrr.cL|jryd|_|jy)zjProcess the next result. Returns: bool: Returns `True` if the fetch is done. FT)rrrrs r( next_resultzSqlResult.next_resultns% <<!& r.c|jS)zReturns True if result has data. Returns: bool: Returns `True` if result has data. .. versionadded:: 8.0.12 rrs r(has_datazSqlResult.has_datazs~~r.Nrr)rurvrwrrrrr+r.r(r r _s " r.r c$eZdZdZdfd ZxZS) DocResultzAllows traversing the DbDoc objects returned by a Collection.find operation. Args: connection (mysqlx.connection.Connection): The Connection object. cVt||}|ytt|dS)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.DbDoc`: A `DbDoc` object. Nr)rrrr)rrrrs r(rzDocResult._read_items1g ) ;&s1v.//r.)rrrr)rurvrwrrrrs@r(rrs 0 0r.r)r%rr&bytesrr)r&rrr)r&rrfloat)r&rrzTuple[int, bytes])r&rrr?)r&rrz List[bytes])r&rrzdecimal.Decimal)r&rrr)r&rrr)9r __future__rr\r2r!rrtypingrrrr r r r r charsetsrdbdocrhelpersrrtypesrrr)r-r4r8rIrNrPrUrcrlrqrsrrrrrrr rrrrrrrrrr rr+r.r(rsN:," (NNN*2-  + + (,   *!3HH*Ba a H8$$N%+k {{{eeP +,+,\E$E$P0#Z0#fz&jz&z.# #L00r.