o h@s,dZddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZddlmZddlmZdd lmZmZdd lmZmZdSddZdTddZdUddZdUddZdVddZdWddZ dWd d!Z!dXd#d$Z"dYd&d'Z#dZd)d*Z$d[d,d-Z%Gd.d/d/Z&Gd0d1d1Z'Gd2d3d3Z(Gd4d5d5Z)Gd6d7d7e)Z*Gd8d9d9e*Z+Gd:d;d;e*Z,Gdd?d?e*Z.Gd@dAdAe*Z/GdBdCdCe*Z0GdDd d Z1GdEdFdFZ2GdGdHdHZ3GdIdJdJe3Z4GdKdLdLe3Z5GdMdNdNe5Z6GdOdPdPe6Z7GdQdRdRe5Z8dS)\z%Implementation of the Result classes.) annotationsN)datetime timedelta)AnyCallableDictListOptionalSequenceTupleUnion)MYSQL_CHARACTER_SETS)DbDoc)decode_from_bytes deprecated)ConnectionType FieldTypescolumnColumnpayloadbytesreturnrc Cst|dkrdS|tjkrt|ddSz tj||WStyE}zt j |t j | dWYd}~dSd}~ww)Nrhex) lenget_type ColumnTypeSTRINGrColumnProtoType converter_mapget_proto_typeKeyErrorsysstderrwriteencode)rrerrr(H/var/www/html/scripts/venv/lib/python3.10/site-packages/mysqlx/result.py from_protobuf/s r*cCs |ddS)Nrr(rr(r(r)bytes_from_protobuf>s r,floatcC t|dksJtd|dS)NzrNr;decimalDecimal)rdigitssignscalerCrDhigh_bcdlow_bcdr(r(r)decimal_from_protobufs:"      r_rcCsd}d}d}d}t|\}}t|\}}t|\}}zt|\}}t|\}}t|\}}t|\}}Wn ty=Ynwt|||||||S)Nr)rGr;r)rhourminutessecondsusecondsyearmonthdayr(r(r)datetime_from_protobufs        rgrcCsd}d}d}d}|ddk}|dd}zt|\}}t|\}}t|\}}t|\}}Wn ty7Ynw|rU|rA|d9}n|rH|d9}n |rO|d9}n|rU|d9}t||||dS)Nrr r)hoursrarb microseconds)rGr;r)rr`rarbrcnegater(r(r)time_from_protobufs4         rkc@eZdZdZdS) Collations!N)__name__ __module__ __qualname__UTF8_GENERAL_CIr(r(r(r)rmrmc@seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#d"Z$d#Z%e&d>d(d)Z'e&d?d,d-Z(e&d@d1d2Z)e&d@d3d4Z*e&d@d5d6Z+e&d@d7d8Z,e&d@d9d:Z-e&d@d;d<Z.d=S)Arr r/r:r5 rU rVrWrT rn"#needlerr Optional[str]cCs*t|D] \}}||kr|SqdSN)varsitems)clsrkeyvaluer(r(r) to_strings zColumnType.to_stringrstrcCst||dSr)getattrupper)rrr(r(r) from_stringszColumnType.from_stringcol_typer=boolcC||j|jfvSr)CHARVARCHARrrr(r(r)is_charzColumnType.is_charcCrr)BINARY VARBINARYrr(r(r) is_binary%rzColumnType.is_binarycCs||j|j|j|jfvSr)TEXTTINYTEXT MEDIUMTEXTLONGTEXTrr(r(r)is_text,s zColumnType.is_textcCs||j|j|j|j|jfvSr)REALDOUBLEFLOATDECIMALNUMERICrr(r(r) is_decimals5szColumnType.is_decimalscCs ||j|j|j|j|j|jfvSr)BITTINYINTSMALLINT MEDIUMINTINTBIGINTrr(r(r) is_numeric?szColumnType.is_numericcCrr)SETENUMrr(r(r) is_finite_setJrzColumnType.is_finite_setN)rrrr)rrrr)rr=rr)/rorprqrrrrrrrrrrrJSONrBYTESTIMEDATEDATETIME TIMESTAMPrrGEOMETRYXMLYEARrrrrTINYBLOBBLOB MEDIUMBLOBLONGBLOBrrrr classmethodrrrrrrrrr(r(r(r)rsh        rc@sreZdZUdZdZdZdZdZdZdZ dZ d Z d Z d Z eeeeeee eeeeeeee ee ee ee ei Zd ed <dS)rr rtrvrwr:rUrVrTr{r|r}z!Dict[int, Callable[[bytes], Any]]r N)rorprqSINTUINTrrrrrrrrrrLrKr,rgrkr3r6rQr_r __annotations__r(r(r(r)rRs0 rc@s>eZdZdddZddd Zedd d Zejdd d ZdS)Flagsrr=rNonecCsVi|_i|_|jjD]\}}|drq t|tr%||j|<||j|<q ||_dS)N__) _allowed_flags _flag_names __class____dict__r startswithr<r=_value)selfrrvalr(r(r)__init__os     zFlags.__init__rc Csd}g}|j}tddD]1}|dK}||@}|r=z ||j|Wq ty<tj|jtj|jj Yq wq d |S)Nr r?,) rrangerNrr"r#r$r%rrjoin)rmask flag_namesr_flagr(r(r)__str__zs  z Flags.__str__cC|jSrrrr(r(r)rz Flags.valuercCs ||_dSrr)rrr(r(r)r N)rr=rrrrrr=)rr=rr)rorprqrrpropertyrsetterr(r(r(r)rns   rc@s eZdZdZdZdZdZdZdS) ColumnFlagsr{r@r8N)rorprqNOT_NULL PRIMARY_KEY UNIQUE_KEY MULTIPLE_KEYAUTO_INCREMENTr(r(r(r)rs rc@rl)DatetimeColumnFlagsr N)rorprqrr(r(r(r)rrsrc@rl)UIntColumnFlagsr N)rorprqZEROFILLr(r(r(r)rrsrc@rl)DoubleColumnFlagsr NrorprqUNSIGNEDr(r(r(r)rrsrc@rl)FloatColumnFlagsr Nrr(r(r(r)rrsrc@rl)BytesColumnFlagsr N)rorprq RIGHT_PADr(r(r(r)rrsrc@seZdZdZdZdZdS)BytesContentTyper rtruN)rorprqrrrr(r(r(r)rsrc@seZdZdZ           dUdVddZdWddZdXddZdXddZdXddZdXdd Z dXd!d"Z e dWd#d$Z e dWd%d&Z e dWd'd(Ze dWd)d*Ze dWd+d,Ze dYd-d.Ze dYd/d0Ze dYd1d2Ze dWd3d4Ze dWd5d6ZdWd7d8ZdWd9d:ZdWd;d<ZdWd=d>ZdWd?d@ZdYdAdBZdYdCdDZdYdEdFZdYdGdHZdWdIdJZdWdKdLZ dZdNdOZ!d[dQdRZ"dZdSdTZ#dS)\ra_Represents meta data for a table column. Args: col_type (int): The column type. catalog (str): The catalog. schema (str): The schema name. table (str): The table name. original_table (str): The original table name. name (str): The column name. original_name (str): The original table name. length (int): The column length, collation (str): The collation name. fractional_digits (int): The fractional digits. flags (int): The flags. content_type (int): The content type. .. versionchanged:: 8.0.12 Nrr=catalogrschematableoriginal_tablename original_namelength Optional[int] collationfractional_digitsflags content_typerrc Cst||_t||_t||_t||_t||_||_d|_||_||_ | |_ | |_ | |_ | |_ d|_d|_d|_d|_d|_d|_d|_|j dkrt|j ttkrZtd|j t|j } | d|_| d|_d|jvprd|jv|_||jtjtjtjtjtjfv|_dS)NFrzNo mapping found for collation r binary_bin)r_schema_name_original_name_table_original_table _proto_type _col_type_catalog_length _collation_fractional_digits_flags _content_type_number_signed _is_padded _is_binary _is_bytes_collation_name_character_set_name _zero_fillrrr; _map_typerrrrrr)rrrrrrrrrrrrrinfor(r(r)rsH          zColumn.__init__rcCst|j|j|jt|jdS)N)rrrr)rrr r rrr(r(r)rszColumn.__str__cCsj|jtjkr tj|_n"|jtjkrtj|_n|jtjkr!tj|_n |jr)tj|_ntj |_|j d@|_ dS)z Map bytes.r N) rrrrrrrrrrrrrr(r(r) _map_bytess       zColumn._map_bytescCsP|jdkr tj|_dS|jtj@dkrtj|_dS|jdkr$tj|_dStd)z Map datetime.rUrr~z#Datetime mapping scenario unhandledN) rrrrrrrrr;rr(r(r) _map_datetimes     zColumn._map_datetimecCsb|jdkr tj|_n"|jdkrtj|_n|jdkrtj|_n|jdkr(tj|_ntj|_d|_dS)z Map int type.r/rwrxryTN) rrrrrrrrrrr(r(r) _map_int_type's         zColumn._map_int_typecCsh|jdkr tj|_n"|jdkrtj|_n|jdkrtj|_n|jdkr(tj|_ntj|_|jd@|_ dS)zMap uint type.rurvr5rUr N) rrrrrrrrrrrr(r(r)_map_uint_type5s        zColumn._map_uint_typecCsN|jtjkr |dS|jtjkr|dS|jtjkr-tj|_|j t j @dk|_ dS|jtj krBtj |_|j t j @dk|_ dS|jtjkrWtj|_|j t j @dk|_ dS|jtjkrc|dS|jtjkrotj|_dS|jtjkr{|dS|jtjkrtj|_dS|jtjkrtj|_dS|jtjkrtj|_dStd|j)z Map type.rzUnknown column type N)rrrr"rr#rrrrrr_is_number_signedrrrr rrr!rrrr;rr(r(r)rCs4                   zColumn._map_typecCr)z@str: The schema name. .. versionadded:: 8.0.12 r rr(r(r) schema_nameazColumn.schema_namecC |jp|jS)z?str: The table name. .. versionadded:: 8.0.12 rr rr(r(r) table_namei zColumn.table_namecCr()z@str: The table label. .. versionadded:: 8.0.12 r rrr(r(r) table_labelqr+zColumn.table_labelcCr()z@str: The column name. .. versionadded:: 8.0.12 r r rr(r(r) column_nameyr+zColumn.column_namecCr()zAstr: The column label. .. versionadded:: 8.0.12 r r rr(r(r) column_labelr+zColumn.column_labelcCr)z@int: The column type. .. versionadded:: 8.0.12 rrr(r(r)typer'z Column.typecCr)zBint. The column length. .. versionadded:: 8.0.12 rrr(r(r)rr'z Column.lengthcCr)zMint: The column fractional digits. .. versionadded:: 8.0.12 rrr(r(r)rr'zColumn.fractional_digitscCr)zCstr: The collation name. .. versionadded:: 8.0.12 rrr(r(r)collation_namer'zColumn.collation_namecCr)zGstr: The character set name. .. versionadded:: 8.0.12 rrr(r(r)character_set_namer'zColumn.character_set_namecCr)zUReturns the schema name. Returns: str: The schema name. r%rr(r(r)get_schema_namer'zColumn.get_schema_namecCr()zSReturns the table name. Returns: str: The table name. r)rr(r(r)get_table_namer+zColumn.get_table_namecCr()zUReturns the table label. Returns: str: The table label. r,rr(r(r)get_table_labelr+zColumn.get_table_labelcCr()zUReturns the column name. Returns: str: The column name. r.rr(r(r)get_column_namer+zColumn.get_column_namecCr()zWReturns the column label. Returns: str: The column label. r0rr(r(r)get_column_labelr+zColumn.get_column_labelcCr)zaReturns the column proto type. Returns: int: The column proto type. )rrr(r(r)r!r'zColumn.get_proto_typecCr)zUReturns the column type. Returns: int: The column type. r2rr(r(r)rr'zColumn.get_typecCr)zYReturns the column length. Returns: int: The column length. r4rr(r(r) get_lengthr'zColumn.get_lengthcCr)zoReturns the column fractional digits. Returns: int: The column fractional digits. r5rr(r(r)get_fractional_digitsr'zColumn.get_fractional_digitscCr)z[Returns the collation name. Returns: str: The collation name. r6rr(r(r)get_collation_namer'zColumn.get_collation_namecCr)zcReturns the character set name. Returns: str: The character set name. r8rr(r(r)get_character_set_namer'zColumn.get_character_set_namercCr)zxReturns `True` if is a number signed. Returns: bool: Returns `True` if is a number signed. )rrr(r(r)is_number_signed r'zColumn.is_number_signedUnion[bool, int]cCr)zfReturns `True` if is padded. Returns: bool: Returns `True` if is padded. )rrr(r(r) is_paddedr'zColumn.is_paddedcCr)zdReturns `True` if is bytes. Returns: bool: Returns `True` if is bytes. )rrr(r(r)is_bytesr'zColumn.is_bytes) NNNNNNNNNNN)rr=rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrD)$rorprq__doc__rrr r!r"r#rrr&r*r-r/r1r3rrr7r9r:r;r<r=r>r!rr?r@rArBrCrErFr(r(r(r)rsl 6                          c@s@eZdZdZddd Zdd d ZdddZeddddZdS)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]rrcCs||_||_dSr)_fields _resultset)rrLrNr(r(r)r*s z Row.__init__rcC t|jSr)reprrPrr(r(r)__repr__0 z Row.__repr__indexUnion[int, str]rcCsdt|tr |j|n|}|dkrt|trtd|d|t|jks)|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<rrQindex_ofr;rrP IndexError)rrV int_indexr(r(r) __getitem__3s  zRow.__getitem__8.0.12 str_indexcCsH|j|}|t|jkrtd|dkrtd|dt|j|S)zReturns the value using the column name. Args: str_index (str): The column name. .. deprecated:: 8.0.12 zArgument out of rangerrXrY)rQrZrrPr[r;r)rr_r\r(r(r) get_string?s zRow.get_stringN)rLrMrNrOrrr)rVrWrr)r_rrr) rorprqrJrrTr]rr`r(r(r(r)rK"s    rKc@sjeZdZdZd%ddZd&d d Zd'd d Zd&ddZd(ddZd)ddZ d*ddZ d+dd Z d,d"d#Z d$S)- BaseResultzProvides base functionality for result objects. Args: connection (mysqlx.connection.Connection): The Connection object. connectionrrrcCsJ||_d|_d|_d|_g|_g|_|durd|_dS|j|_|dSNFrr) _connection_closed_rows_affected _generated_id_generated_ids _warnings _protocolprotocolfetch_active_resultrrbr(r(r)rWs  zBaseResult.__init__r=cCr)zReturns the number of affected items for the last operation. Returns: int: The number of affected items. rfrr(r(r)get_affected_items_counter'z#BaseResult.get_affected_items_count List[Dict[str, Union[int, str]]]cCr)zZReturns the warnings. Returns: `list`: The list of warnings. )rirr(r(r) get_warningsmr'zBaseResult.get_warningscCrR)zcReturns the number of warnings. Returns: int: The number of warnings. )rrirr(r(r)get_warnings_countu zBaseResult.get_warnings_countrrcC ||_dS)z Sets if resultset fetch is done.N)rerrr(r(r) set_closed}rzBaseResult.set_closedlevelcodemsgrcCs|j|||ddS)zAppend a warning. Args: level (int): The warning level. code (int): The warning code. msg (str): The warning message. )rwrxryN)rirN)rrwrxryr(r(r)append_warningszBaseResult.append_warning generated_ids List[int]cCrt)zSets the generated ids.Nrh)rr{r(r(r)set_generated_idsrzBaseResult.set_generated_ids generated_idcCrt)zSets the generated insert id.Nrg)rrr(r(r)set_generated_insert_idrz"BaseResult.set_generated_insert_idtotalcCrt)z!Sets the number of rows affected.Nrn)rrr(r(r)set_rows_affectedrzBaseResult.set_rows_affectedNrbrrrr)rrprrrr)rwr=rxr=ryrrr)r{r|rr)rr=rr)rr=rr) rorprqrJrrorqrrrvrzr~rrr(r(r(r)raPs       racs`eZdZdZ  ddfd d Zdd d ZeddddZeddddZdddZ Z 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. NrbOptional[ConnectionType]idsOptional[List[int]]rrcs.t|||_|dur|j|dSdSr)superr_idsrd close_result)rrbrrr(r)rs zResult.__init__r=cCr)zjReturns the last insert id auto generated. Returns: int: The last insert id. rrr(r(r)get_autoincrement_valuer'zResult.get_autoincrement_valuer^rcCs&|jdus t|jdkrdS|jdS)zdReturns ID of the last document inserted into a collection. .. deprecated:: 8.0.12 Nr)rrrr(r(r)get_document_ids zResult.get_document_idcCr)zIReturns the generated insert id. .. deprecated:: 8.0.12 rrr(r(r)get_generated_insert_idr'zResult.get_generated_insert_idr|cCr)zReturns the generated ids.r}rr(r(r)get_generated_idsrzResult.get_generated_ids)NN)rbrrrrrr)rr)rr|) rorprqrJrrrrrr __classcell__r(r(rr)rs    rcseZdZdZd'fdd Zd(d d Zed)d dZd*ddZd+ddZ d,ddZ d-ddZ d.ddZ d/d d!Z d0d#d$Zd0d%d&ZZS)1BufferingResultzProvides base functionality for buffering result objects. Args: connection (mysqlx.connection.Connection): The Connection object. ids (`list`): A list of IDs. rbrrrcs<t|g|_d|_d|_g|_d|_d|_|dSrc) rr_columns _has_data_has_more_results_items _page_size _position _init_resultrmrr(r)rs  zBufferingResult.__init__rVr=Union[Row, DbDoc]cCs |j|Sr)r)rrVr(r(r)r]rUzBufferingResult.__getitem__cCrR)zint: The total of items.)rrrr(r(r)counts zBufferingResult.countcCsP|j||_t|jdk|_g|_d|_d|_|j|jr#|dSddS)zInitialize the result.rrrN) rdget_column_metadatarr_has_more_datarrrset_active_resultrr(r(r)rs  zBufferingResult._init_resultdumpingrOptional[Union[Row, DbDoc]]cCsn|j|}|dur dSdgt|d}|s2tt|dD]}|j|}t||d|||<qt||S)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.Row`: A `Row` object. Nfield)rdread_rowrrrr*rK)rrrowrCrrr(r(r) _read_items   zBufferingResult._read_itemrDcCsN|jrdSd}t|jD]}|d}|dur|S|j||d7}q |S)zSReads the page items. Returns: int: Total items read. FrNr )rerrrrrN)rrrrCr(r(r)_page_in_itemss  zBufferingResult._page_in_itemscol_namercCs0d}|jD]}||kr|S|d7}qdS)zeReturns the index of the column. Returns: int: The index of the column. rr r)rr>)rrrVcolr(r(r)rZs    zBufferingResult.index_ofcCs|jrdS|dS)zuFetch one item. Returns: :class:`mysqlx.Row` or :class:`mysqlx.DbDoc`: one result item. NF)rerrr(r(r) fetch_one s zBufferingResult.fetch_oneList[Union[Row, DbDoc]]cCs |s |jSq)zFetch all items. Returns: `list`: The list of items of :class:`mysqlx.DbDoc` or :class:`mysqlx.Row`. )rrrr(r(r) fetch_all+s zBufferingResult.fetch_allrcCrt)zdSets if result has data. Args: flag (bool): `True` if result has data. Nrrur(r(r) set_has_data7rszBufferingResult.set_has_datacCrt)zfSets if has more results. Args: flag (bool): `True` if has more results. N)rrur(r(r)set_has_more_results?rsz$BufferingResult.set_has_more_resultsr)rVr=rrrrG)rrrrrI)rrrr=)rr)rrr)rorprqrJrr]rrrrrrZrrrrrr(r(rr)rs      rc@s(eZdZdZed ddZd ddZdS) RowResultzAllows traversing the Row objects returned by a Table.select operation. Args: connection (mysqlx.connection.Connection): The Connection object. r List[Column]cCr)z`list`: The list of columns.rrr(r(r)columnsOszRowResult.columnscCr)zReturns the list of columns. Returns: `list`: The list of columns. .. versionadded:: 8.0.12 rrr(r(r) get_columnsTzRowResult.get_columnsN)rr)rorprqrJrrrr(r(r(r)rHs  rc@s.eZdZdZd ddZd ddZd d d Zd S) SqlResultzRepresents a result from a SQL statement. Args: connection (mysqlx.connection.Connection): The Connection object. rr=cCr)zReturns the identifier for the last record inserted. Returns: str: The identifier of the last record inserted. rrr(r(r)rfr'z!SqlResult.get_autoincrement_valuercCs|jrdSd|_|dS)zjProcess the next result. Returns: bool: Returns `True` if the fetch is done. FT)rerrrr(r(r) next_resultns zSqlResult.next_resultcCr)zReturns True if result has data. Returns: bool: Returns `True` if result has data. .. versionadded:: 8.0.12 rrr(r(r)has_datazrzSqlResult.has_dataNrrH)rorprqrJrrrr(r(r(r)r_s    rcs"eZdZdZdfdd ZZS) DocResultzAllows traversing the DbDoc objects returned by a Collection.find operation. Args: connection (mysqlx.connection.Connection): The Connection object. rrrrcs(t|}|dur dStt|dS)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.DbDoc`: A `DbDoc` object. Nr)rrrr)rrrrr(r)rs zDocResult._read_item)rrrr)rorprqrJrrr(r(rr)rsr)rrrrrr)rrrr)rrrr-)rrrr7)rrrr=)rrrrM)rrrrR)rrrr)rrrr)9rJ __future__rrXr1r#rrtypingrrrrr r r r charsetsrdbdocrhelpersrrtypesrrr*r,r3r6rGrKrLrQr_rgrkrmrrrrrrrrrrrrKrarrrrrr(r(r(r)sV (            $ !d' j.H3}&