o by@sdZddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z dd Z d d Zd d ZddZddZddZddZddZddZddZddZGdddeZGd d!d!eZGd"d#d#eZGd$d%d%eZGd&d'd'eZGd(d)d)eZGd*d+d+eZGd,d-d-eZ Gd.d/d/eZ!Gd0d1d1eZ"Gd2d3d3eZ#Gd4d5d5eZ$Gd6d7d7eZ%Gd8d9d9eZ&Gd:d;d;e&Z'Gdd?d?e(Z)Gd@dAdAe)Z*GdBdCdCe(Z+dS)Dz%Implementation of the Result classes.N)datetime timedelta)DbDoc)MYSQL_CHARACTER_SETS)decode_from_bytes deprecatedcCst|dkrdSz(tjttjttjttjt tj t tj t tjttjttjttjttjti ||WStyX}ztjd|tjd|dWYd}~dSd}~ww)Nr{0}hex)lenColumnProtoTypeSINTvarsint_from_protobufUINTvarint_from_protobufBYTESbytes_from_protobufDATETIMEdatetime_from_protobufTIMEtime_from_protobufFLOATfloat_from_protobufDOUBLEdouble_from_protobufBITSETset_from_protobufENUMDECIMALdecimal_from_protobufKeyErrorsysstderrwriteformatencode)col_typepayloaderrr*L/var/www/secure340b-portal/env/lib/python3.10/site-packages/mysqlx/result.py from_protobuf+s2    r,cCst|ddS)N)rr(r*r*r+rCrcCt|dksJtd|S)Nzsr c@s6eZdZddZddZeddZejddZdS) FlagscCsVi|_i|_|jjD]\}}|drq t|tr%||j|<||j|<q ||_dS)N__) Z_allowed_flags _flag_names __class____dict__r startswithr;r<_value)selfrrvalr*r*r+__init__Ms     zFlags.__init__c Csd}g}|j}tddD]5}|dK}||@}|rAz ||j|Wq ty@tjd|jtjd|j j Yq wq d |S)Nrr?r ,) rrangerHrr!r"r#r$r%rrjoin)rmaskZ flag_namesr_flagr*r*r+__str__Xs  z Flags.__str__cC|jSrrrr*r*r+rjsz Flags.valuecCs ||_dSrr)rrr*r*r+rns N)r`rarbrrpropertyrsetterr*r*r*r+rLs  rc@s eZdZdZdZdZdZdZdS) ColumnFlagsrlr|@r7N)r`rarbZNOT_NULLZ PRIMARY_KEYZ UNIQUE_KEYZ MULTIPLE_KEYZAUTO_INCREMENTr*r*r*r+rss rc@r])DatetimeColumnFlagsrN)r`rarbrr*r*r*r+r{rcrc@r])UIntColumnFlagsrN)r`rarbZZEROFILLr*r*r*r+rrcrc@r])DoubleColumnFlagsrNr`rarbUNSIGNEDr*r*r*r+rrcrc@r])FloatColumnFlagsrNrr*r*r*r+rrcrc@r])BytesColumnFlagsrN)r`rarbZ RIGHT_PADr*r*r*r+rrcrc@seZdZdZdZdZdS)BytesContentTyperrerfN)r`rarbrrrr*r*r*r+rsrc@s:eZdZdZ    dAddZddZddZd d Zd d Zd dZ ddZ e ddZ e ddZ e ddZe ddZe ddZe ddZe ddZe dd Ze d!d"Ze d#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Z d;d<Z!d=d>Z"d?d@Z#dS)BColumna_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 Nc 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)NFrz"No mapping found for collation {0}rbinaryZ_bin) r_schema_name_original_name_table_original_table _proto_type _col_type_catalog_lengthZ _collation_fractional_digits_flags _content_type_number_signed _is_padded _is_binary _is_bytes_collation_name_character_set_name _zero_fillr rr:r% _map_typerdrrrrr)rr'catalogschematableZoriginal_tablename original_namelengthZ collationfractional_digitsflags content_typeinfor*r*r+rsH            zColumn.__init__cCst|j|j|jt|jdS)N)r'rrr)strrrrrrr*r*r+rs zColumn.__str__cCsj|jtjkr tj|_n"|jtjkrtj|_n|jtjkr!tj|_n |jr)tj|_ntj |_|j d@|_ dS)z Map bytes.rN) rrrrdrrrrrrrrrr*r*r+ _map_bytess       zColumn._map_bytescCsP|jdkr tj|_dS|jtj@dkrtj|_dS|jdkr$tj|_dStd)z Map datetime.rKrroz#Datetime mapping scenario unhandledN) rrdrrrrrrr:rr*r*r+ _map_datetimes     zColumn._map_datetimecCsb|jdkr tj|_n"|jdkrtj|_n|jdkrtj|_n|jdkr(tj|_ntj|_d|_dS)z Map int type.r1rhrirjTN) rrdrrrrrrrrr*r*r+ _map_int_types         zColumn._map_int_typecCsh|jdkr tj|_n"|jdkrtj|_n|jdkrtj|_n|jdkr(tj|_ntj|_|jd@|_ dS)zMap uint type.rfrgr6rKrN) rrdrrrrrrrrrr*r*r+_map_uint_types        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 {0}N)rr r rrrrrdrrrrZ_is_number_signedrrrrrrrrrrr:r%rr*r*r+r s:                   zColumn._map_typecCr)z@str: The schema name. .. versionadded:: 8.0.12 rrr*r*r+ schema_name.zColumn.schema_namecC |jp|jS)z?str: The table name. .. versionadded:: 8.0.12 rrrr*r*r+ table_name6 zColumn.table_namecCr)z@str: The table label. .. versionadded:: 8.0.12 rrrr*r*r+ table_label>rzColumn.table_labelcCr)z@str: The column name. .. versionadded:: 8.0.12 rrrr*r*r+ column_nameFrzColumn.column_namecCr)zAstr: The column label. .. versionadded:: 8.0.12 rrrr*r*r+ column_labelNrzColumn.column_labelcCr)z@int: The column type. .. versionadded:: 8.0.12 rrr*r*r+typeVrz Column.typecCr)zBint. The column length. .. versionadded:: 8.0.12 rrr*r*r+r^rz Column.lengthcCr)zMint: The column fractional digits. .. versionadded:: 8.0.12 rrr*r*r+rfrzColumn.fractional_digitscCr)zCstr: The collation name. .. versionadded:: 8.0.12 rrr*r*r+collation_namenrzColumn.collation_namecCr)zGstr: The character set name. .. versionadded:: 8.0.12 rrr*r*r+character_set_namevrzColumn.character_set_namecCr)zUReturns the schema name. Returns: str: The schema name. rrr*r*r+get_schema_name~rzColumn.get_schema_namecCr)zSReturns the table name. Returns: str: The table name. rrr*r*r+get_table_namerzColumn.get_table_namecCr)zUReturns the table label. Returns: str: The table label. rrr*r*r+get_table_labelrzColumn.get_table_labelcCr)zUReturns the column name. Returns: str: The column name. rrr*r*r+get_column_namerzColumn.get_column_namecCr)zWReturns the column label. Returns: str: The column label. rrr*r*r+get_column_labelrzColumn.get_column_labelcCr)zaReturns the column proto type. Returns: int: The column proto type. )rrr*r*r+get_proto_typerzColumn.get_proto_typecCr)zUReturns the column type. Returns: int: The column type. rrr*r*r+get_typerzColumn.get_typecCr)zYReturns the column length. Returns: int: The column length. rrr*r*r+ get_lengthrzColumn.get_lengthcCr)zoReturns the column fractional digits. Returns: int: The column fractional digits. rrr*r*r+get_fractional_digitsrzColumn.get_fractional_digitscCr)z[Returns the collation name. Returns: str: The collation name. rrr*r*r+get_collation_namerzColumn.get_collation_namecCr)zcReturns the character set name. Returns: str: The character set name. rrr*r*r+get_character_set_namerzColumn.get_character_set_namecCr)zxReturns `True` if is a number signed. Returns: bool: Returns `True` if is a number signed. )rrr*r*r+is_number_signedrzColumn.is_number_signedcCr)zfReturns `True` if is padded. Returns: bool: Returns `True` if is padded. )rrr*r*r+ is_paddedrzColumn.is_paddedcCr)zdReturns `True` if is bytes. Returns: bool: Returns `True` if is bytes. )rrr*r*r+is_bytesrzColumn.is_bytes) NNNNNNNNNNN)$r`rarb__doc__rrrrrrrrrrrrrrrrrrrrrrrr r r r r rrrrr*r*r*r+rs^ ' !           rc@s0eZdZdZddZddZedddZd S) RowzRepresents a row element returned from a SELECT query. Args: rs (mysqlx.SqlResult or mysqlx.RowResult): The result set. fields (`list`): The list of fields. cCs||_||_dSr)_fields _resultset)rrsfieldsr*r*r+rs z Row.__init__cCsbt|tr |j|n|}|dkrt|trtd||t|jks(|dkr,td|j|S)z\Returns the value of a column by name or index. .. versionchanged:: 8.0.12 r-Column name '{0}' not foundrzIndex out of range) r;rrindex_ofr:r%r r IndexError)rindex int_indexr*r*r+ __getitem__s zRow.__getitem__8.0.12cCsF|j|}|t|jkrtd|dkrtd|t|j|S)zReturns the value using the column name. Args: str_index (str): The column name. .. deprecated:: 8.0.12 zArgument out of ranger-r)rrr rrr:r%r)rZ str_indexrr*r*r+ get_strings zRow.get_stringN)r`rarbrrrrrr*r*r*r+rs  rc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS) BaseResultzProvides base functionality for result objects. Args: connection (mysqlx.connection.Connection): The Connection object. cCsJ||_d|_d|_d|_g|_g|_|durd|_dS|j|_|dSNFrr-) _connection_closed_rows_affected _generated_id_generated_ids _warnings _protocolprotocolZfetch_active_resultr connectionr*r*r+rs  zBaseResult.__init__cCr)zReturns the number of affected items for the last operation. Returns: int: The number of affected items. r$rr*r*r+get_affected_items_count,rz#BaseResult.get_affected_items_countcCr)zZReturns the warnings. Returns: `list`: The list of warnings. )r'rr*r*r+ get_warnings4rzBaseResult.get_warningscC t|jS)zcReturns the number of warnings. Returns: int: The number of warnings. )r r'rr*r*r+get_warnings_count< zBaseResult.get_warnings_countcC ||_dS)z)Sets if resultset fetch is done. N)r#rrr*r*r+ set_closedD zBaseResult.set_closedcCs|j|||ddS)zAppend a warning. Args: level (int): The warning level. code (int): The warning code. msg (str): The warning message. )levelcodemsgN)r'rH)rr6r7r8r*r*r+append_warningIszBaseResult.append_warningcCr2)z Sets the generated ids. Nr&)rZ generated_idsr*r*r+set_generated_idsSr5zBaseResult.set_generated_idscCr2)z&Sets the generated insert id. Nr%)rZ generated_idr*r*r+set_generated_insert_idXr5z"BaseResult.set_generated_insert_idcCr2)z*Sets the number of rows affected. Nr,)rtotalr*r*r+set_rows_affected]r5zBaseResult.set_rows_affectedN) r`rarbrrr-r.r0r4r9r;r=r?r*r*r*r+r s  r csReZdZdZdfdd ZddZeddd Zedd d Zd d Z 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. Ncs2tt||||_|dur|j|dSdSr)superr@r_idsr"Z close_result)rr+idsrr*r+rks zResult.__init__cCr)zjReturns the last insert id auto generated. Returns: int: The last insert id. r<rr*r*r+get_autoincrement_valuerrzResult.get_autoincrement_valuercCs&|jdus t|jdkrdS|jdS)zdReturns ID of the last document inserted into a collection. .. deprecated:: 8.0.12 Nr)rBr rr*r*r+get_document_idzs zResult.get_document_idcCr)zIReturns the generated insert id. .. deprecated:: 8.0.12 r<rr*r*r+get_generated_insert_idrzResult.get_generated_insert_idcCr)z#Returns the generated ids. r:rr*r*r+get_generated_idszResult.get_generated_ids)NN) r`rarbrrrErrFrGrH __classcell__r*r*rDr+r@cs  r@csteZdZdZfddZddZeddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZZS)BufferingResultzProvides base functionality for buffering result objects. Args: connection (mysqlx.connection.Connection): The Connection object. ids (`list`): A list of IDs. cs@tt||g|_d|_d|_g|_d|_d|_| dSr!) rArKr_columns _has_data_has_more_results_items _page_size _position _init_resultr*rDr*r+rs zBufferingResult.__init__cCs |j|Sr)rO)rrr*r*r+rs zBufferingResult.__getitem__cCr/)zint: The total of items.)r rOrr*r*r+countr5zBufferingResult.countcCsX|j||_t|jdkrdnd|_g|_d|_d|_|j|jr'|dSddS)zInitialize the result.rTFrpr-N) r"Zget_column_metadatarLr Z_has_more_datarOrPrQZset_active_resultrr*r*r+rRszBufferingResult._init_resultcCsr|j|}|dur dSdgt|d}|s4tt|dD]}|j|}t||d|||<qt||S)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.Row`: A `Row` object. Nfield)r"Zread_rowr rrLr,r r)rdumpingrowrBrcolr*r*r+ _read_items    zBufferingResult._read_itemcCsN|jrdSd}t|jD]}|d}|dur|S|j||d7}q |S)zSReads the page items. Returns: int: Total items read. FrNr)r#rrPrXrOrH)rrSrrBr*r*r+_page_in_itemss  zBufferingResult._page_in_itemscCs0d}|jD]}||kr|S|d7}qdS)zeReturns the index of the column. Returns: int: The index of the column. rrr-)rLr)rZcol_namerrWr*r*r+rs    zBufferingResult.index_ofcCs|jrdS|dS)zuFetch one item. Returns: :class:`mysqlx.Row` or :class:`mysqlx.DbDoc`: one result item. NF)r#rXrr*r*r+ fetch_ones zBufferingResult.fetch_onecCs |s |jSq)zFetch all items. Returns: `list`: The list of items of :class:`mysqlx.DbDoc` or :class:`mysqlx.Row`. )rYrOrr*r*r+ fetch_alls zBufferingResult.fetch_allcCr2)zdSets if result has data. Args: flag (bool): `True` if result has data. NrMr3r*r*r+ set_has_datar1zBufferingResult.set_has_datacCr2)zfSets if has more results. Args: flag (bool): `True` if has more results. N)rNr3r*r*r+set_has_more_resultsr1z$BufferingResult.set_has_more_results)r`rarbrrrrrSrRrXrYrrZr[r]r^rJr*r*rDr+rKs       rKcs4eZdZdZfddZeddZddZZS) RowResultzAllows traversing the Row objects returned by a Table.select operation. Args: connection (mysqlx.connection.Connection): The Connection object. ctt||dSr)rAr_rr*rDr*r+rzRowResult.__init__cCr)z`list`: The list of columns.rLrr*r*r+columnsrIzRowResult.columnscCr)zReturns the list of columns. Returns: `list`: The list of columns. .. versionadded:: 8.0.12 rbrr*r*r+ get_columnszRowResult.get_columns) r`rarbrrrrcrdrJr*r*rDr+r_s   r_cs8eZdZdZfddZddZddZdd ZZS) SqlResultzRepresents a result from a SQL statement. Args: connection (mysqlx.connection.Connection): The Connection object. cr`r)rArfrr*rDr*r+r/razSqlResult.__init__cCr)zReturns the identifier for the last record inserted. Returns: str: The identifier of the last record inserted. r<rr*r*r+rE2rz!SqlResult.get_autoincrement_valuecCs|jrdSd|_|dS)zjProcess the next result. Returns: bool: Returns `True` if the fetch is done. FT)r#rNrRrr*r*r+ next_result:s zSqlResult.next_resultcCr)zReturns True if result has data. Returns: bool: Returns `True` if result has data. .. versionadded:: 8.0.12 r\rr*r*r+has_dataFrezSqlResult.has_data) r`rarbrrrErgrhrJr*r*rDr+rf)s   rfcs,eZdZdZfddZfddZZS) DocResultzAllows traversing the DbDoc objects returned by a Collection.find operation. Args: connection (mysqlx.connection.Connection): The Connection object. cr`r)rArirr*rDr*r+rWrazDocResult.__init__cs,tt||}|durdStt|dS)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.DbDoc`: A `DbDoc` object. Nr)rArirXrr)rrUrVrDr*r+rXZs zDocResult._read_item)r`rarbrrrXrJr*r*rDr+riPs ri),rrNr3r"rrZdbdocrZcharsetsrhelpersrrr,rrrrDrrrr rrobjectr^rdr rrrrrrrrrrr r@rKr_rfrir*r*r*r+sP  $ L'[)K/~'