U ecx@sdZddlZddlZddlZddlmZmZddlmZddlm Z ddl m Z m Z dd Z d d Zd d ZddZddZddZddZddZddZddZddZGdddZGd d!d!ZGd"d#d#ZGd$d%d%ZGd&d'd'eZGd(d)d)eZGd*d+d+eZGd,d-d-eZGd.d/d/eZ Gd0d1d1eZ!Gd2d3d3eZ"Gd4d5d5Z#Gd6d7d7Z$Gd8d9d9Z%Gd:d;d;e%Z&Gdd?d?e'Z(Gd@dAdAe(Z)GdBdCdCe'Z*dS)Dz%Implementation of the Result classes.N)datetime timedelta)MYSQL_CHARACTER_SETS)DbDoc)decode_from_bytes deprecatedc Cst|dkrdS|tjkr.t|ddSztj||WStk r}z,t j |t j | dWYdSd}~XYnXdS)Nrhex) lenget_type ColumnTypeSTRINGrColumnProtoType converter_mapget_proto_typeKeyErrorsysstderrwriteencode)columnpayloaderrr1/tmp/pip-unpacked-wheel-ix8dg7li/mysqlx/result.py from_protobuf+s rcCs |ddS)Nr rrrrrbytes_from_protobuf:srcCs t|dksttd|dS)Nz r?r@r= rS"#cCs*t|D]\}}||kr |Sq dSN)varsitems)clsneedlekeyvaluerrr to_strings zColumnType.to_stringcCst||dSrq)getattrupper)rtrvrrr from_stringszColumnType.from_stringcCs||j|jfkSrq)CHARVARCHARrtcol_typerrris_charszColumnType.is_charcCs||j|jfkSrq)BINARY VARBINARYr~rrr is_binary!szColumnType.is_binarycCs||j|j|j|jfkSrq)TEXTTINYTEXT MEDIUMTEXTLONGTEXTr~rrris_text(s zColumnType.is_textcCs||j|j|j|j|jfkSrq)REALDOUBLEFLOATDECIMALNUMERICr~rrr is_decimals1szColumnType.is_decimalscCs ||j|j|j|j|j|jfkSrq)BITTINYINTSMALLINT MEDIUMINTINTBIGINTr~rrr is_numeric;szColumnType.is_numericcCs||j|jfkSrq)SETENUMr~rrr is_finite_setFszColumnType.is_finite_setN)/rTrUrVrrrrrrrrrrrJSONrBYTESTIMEDATEDATETIME TIMESTAMPrrGEOMETRYXMLZYEARr|r}rrZTINYBLOBZBLOBZ MEDIUMBLOBZLONGBLOBrrrr classmethodrxr{rrrrrrrrrrr sf       r c@sheZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z eeeeeee eeeeeeee ee ee ee ei Zd S) rrrWrYrZr)r>r?r=r^r_r`N)rTrUrVSINTUINTrrrrrrrrrr9r8rrMrQr$r&r;rErrrrrrNsDrc@s6eZdZddZddZeddZejddZdS) FlagscCsVi|_i|_|jjD]2\}}|dr,qt|tr||j|<||j|<q||_dS)N__) Z_allowed_flags _flag_names __class____dict__rs startswithr+r,_value)selfrwrvvalrrr__init__ks    zFlags.__init__c Csd}g}|j}tddD]d}|dK}||@}|rz||j|Wqtk rztj|jtj|jj YqXqd |S)Nrr?,) rranger:rrrrrrrjoin)rmaskZ flag_namesrw_flagrrr__str__vsz Flags.__str__cCs|jSrqrrrrrrwsz Flags.valuecCs ||_dSrqr)rrrrrrwsN)rTrUrVrrpropertyrwsetterrrrrrjs   rc@s eZdZdZdZdZdZdZdS) ColumnFlagsr^rn@r'N)rTrUrVZNOT_NULLZ PRIMARY_KEYZ UNIQUE_KEYZ MULTIPLE_KEYZAUTO_INCREMENTrrrrrs rc@seZdZdZdS)DatetimeColumnFlagsrN)rTrUrVrrrrrrsrc@seZdZdZdS)UIntColumnFlagsrN)rTrUrVZZEROFILLrrrrrsrc@seZdZdZdS)DoubleColumnFlagsrNrTrUrVUNSIGNEDrrrrrsrc@seZdZdZdS)FloatColumnFlagsrNrrrrrrsrc@seZdZdZdS)BytesColumnFlagsrN)rTrUrVZ RIGHT_PADrrrrrsrc@seZdZdZdZdZdS)BytesContentTyperrWrXN)rTrUrVrrrrrrrrsrc @s2eZdZdZdAddZddZddZd 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 dkr|j ttkrtd|j t|j } | d|_| d|_d|jkpd|jk|_||jtjtjtjtjtjfk|_dS)NFrzNo mapping found for collation 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* _map_typer rrrrr)rrcatalogschematableZoriginal_tablename original_namelengthZ collationfractional_digitsflags content_typeinforrrrsH         zColumn.__init__cCst|j|j|jt|jdS)N)rrrr)strrrrrrrrrrszColumn.__str__cCsj|jtjkrtj|_nD|jtjkr,tj|_n.|jtjkrBtj|_n|jrRtj|_ntj |_|j d@|_ dS)z Map bytes.rN) rrrr rrrrrrrrrrrr _map_bytes s       zColumn._map_bytescCsN|jdkrtj|_n6|jtj@dkr.tj|_n|jdkrBtj|_ntddS)z Map datetime.r>rraz#Datetime mapping scenario unhandledN) rr rrrrrrr*rrrr _map_datetimes     zColumn._map_datetimecCsb|jdkrtj|_nD|jdkr(tj|_n0|jdkrrN) rr rrrrrrrrrrrr_map_uint_type1s        zColumn._map_uint_typecCsF|jtjkr|n*|jtjkr0|n|jtjkrXtj|_|j t j @dk|_ n|jtj krtj |_|j t j @dk|_ n|jtjkrtj|_|j t j @dk|_ n|jtjkr|n|jtjkrtj|_nn|jtjkr|nX|jtjkrtj|_n@|jtjkrtj|_n(|jtjkr2tj|_ntd|jdS)z Map type.rzUnknown column type N)rrrrrrrr rrrrZ_is_number_signedrrrrrrrrrrr*rrrrr?s4                zColumn._map_typecCs|jS)z@str: The schema name. .. versionadded:: 8.0.12 rrrrr schema_name]szColumn.schema_namecCs |jp |jS)z?str: The table name. .. versionadded:: 8.0.12 rrrrrr table_nameeszColumn.table_namecCs |jp |jS)z@str: The table label. .. versionadded:: 8.0.12 rrrrrr table_labelmszColumn.table_labelcCs |jp |jS)z@str: The column name. .. versionadded:: 8.0.12 rrrrrr column_nameuszColumn.column_namecCs |jp |jS)zAstr: The column label. .. versionadded:: 8.0.12 rrrrrr column_label}szColumn.column_labelcCs|jS)z@int: The column type. .. versionadded:: 8.0.12 rrrrrtypesz Column.typecCs|jS)zBint. The column length. .. versionadded:: 8.0.12 rrrrrrsz Column.lengthcCs|jS)zMint: The column fractional digits. .. versionadded:: 8.0.12 rrrrrrszColumn.fractional_digitscCs|jS)zCstr: The collation name. .. versionadded:: 8.0.12 rrrrrcollation_nameszColumn.collation_namecCs|jS)zGstr: The character set name. .. versionadded:: 8.0.12 rrrrrcharacter_set_nameszColumn.character_set_namecCs|jS)zUReturns the schema name. Returns: str: The schema name. rrrrrget_schema_nameszColumn.get_schema_namecCs |jp |jS)zSReturns the table name. Returns: str: The table name. rrrrrget_table_nameszColumn.get_table_namecCs |jp |jS)zUReturns the table label. Returns: str: The table label. rrrrrget_table_labelszColumn.get_table_labelcCs |jp |jS)zUReturns the column name. Returns: str: The column name. rrrrrget_column_nameszColumn.get_column_namecCs |jp |jS)zWReturns the column label. Returns: str: The column label. rrrrrget_column_labelszColumn.get_column_labelcCs|jS)zaReturns the column proto type. Returns: int: The column proto type. )rrrrrrszColumn.get_proto_typecCs|jS)zUReturns the column type. Returns: int: The column type. rrrrrr szColumn.get_typecCs|jS)zYReturns the column length. Returns: int: The column length. rrrrr get_lengthszColumn.get_lengthcCs|jS)zoReturns the column fractional digits. Returns: int: The column fractional digits. rrrrrget_fractional_digitsszColumn.get_fractional_digitscCs|jS)z[Returns the collation name. Returns: str: The collation name. rrrrrget_collation_nameszColumn.get_collation_namecCs|jS)zcReturns the character set name. Returns: str: The character set name. rrrrrget_character_set_nameszColumn.get_character_set_namecCs|jS)zxReturns `True` if is a number signed. Returns: bool: Returns `True` if is a number signed. )rrrrris_number_signedszColumn.is_number_signedcCs|jS)zfReturns `True` if is padded. Returns: bool: Returns `True` if is padded. )rrrrr is_padded szColumn.is_paddedcCs|jS)zdReturns `True` if is bytes. Returns: bool: Returns `True` if is bytes. )rrrrris_bytesszColumn.is_bytes) NNNNNNNNNNN)$rTrUrV__doc__rrrrrrrrrrrrrrrrrrrrrrrrr rrrrrrrrrrrrsj 6            rc@s8eZdZdZddZddZddZedd d Zd S) 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. cCs||_||_dSrq)_fields _resultset)rZ resultsetfieldsrrrr&sz Row.__init__cCs t|jSrq)reprr rrrr__repr__*sz Row.__repr__cCsdt|tr|j|n|}|dkrt||S)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.Row`: A `Row` object. Nfield)rZread_rowr rr?rr )rdumpingrowr2rvrrrr _read_items  zBufferingResult._read_itemcCsL|jr dSd}t|jD].}|d}|dkr2qH|j||d7}q|S)zSReads the page items. Returns: int: Total items read. FrNr)rrrCrJrBr:)rrFrr2rrr_page_in_itemss   zBufferingResult._page_in_itemscCs0d}|jD] }||kr"|S|d7}q dS)zeReturns the index of the column. Returns: int: The index of the column. rrr )r?r)rZcol_namercolrrrr s    zBufferingResult.index_ofcCs|jr dS|dS)zuFetch one item. Returns: :class:`mysqlx.Row` or :class:`mysqlx.DbDoc`: one result item. NF)rrJrrrr fetch_oneszBufferingResult.fetch_onecCs|sq q|jS)zFetch all items. Returns: `list`: The list of items of :class:`mysqlx.DbDoc` or :class:`mysqlx.Row`. )rKrBrrrr fetch_all!szBufferingResult.fetch_allcCs ||_dS)zdSets if result has data. Args: flag (bool): `True` if result has data. Nr@r(rrr set_has_data-szBufferingResult.set_has_datacCs ||_dS)zfSets if has more results. Args: flag (bool): `True` if has more results. N)rAr(rrrset_has_more_results5sz$BufferingResult.set_has_more_results)rTrUrVrrrrrFrErJrKrrMrNrPrQr=rrr8rr>s      r>c@s$eZdZdZeddZddZdS) RowResultzAllows traversing the Row objects returned by a Table.select operation. Args: connection (mysqlx.connection.Connection): The Connection object. cCs|jS)z`list`: The list of columns.r?rrrrcolumnsEszRowResult.columnscCs|jS)zReturns the list of columns. Returns: `list`: The list of columns. .. versionadded:: 8.0.12 rSrrrr get_columnsJszRowResult.get_columnsN)rTrUrVrrrTrUrrrrrR>s rRc@s(eZdZdZddZddZddZdS) SqlResultzRepresents a result from a SQL statement. Args: connection (mysqlx.connection.Connection): The Connection object. cCs|jS)zReturns the identifier for the last record inserted. Returns: str: The identifier of the last record inserted. r0rrrrr9\sz!SqlResult.get_autoincrement_valuecCs|jr dSd|_|dS)zjProcess the next result. Returns: bool: Returns `True` if the fetch is done. FT)rrArErrrr next_resultds zSqlResult.next_resultcCs|jS)zReturns True if result has data. Returns: bool: Returns `True` if result has data. .. versionadded:: 8.0.12 rOrrrrhas_datapszSqlResult.has_dataN)rTrUrVrr9rWrXrrrrrVUs rVcs eZdZdZfddZZS) DocResultzAllows traversing the DbDoc objects returned by a Collection.find operation. Args: connection (mysqlx.connection.Connection): The Connection object. cs(t|}|dkrdStt|dS)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.DbDoc`: A `DbDoc` object. Nr)r5rJrr)rrHrIr8rrrJs zDocResult._read_item)rTrUrVrrJr=rrr8rrY{srY)+rrAr"rrrZcharsetsrZdbdocrZhelpersrrrrr$r&r4r8r9r;rErMrQrRr rrrrrrrrrrr rr4r>rRrVrYrrrrsN  $!d' j,H/}&