U 4Je@spdZddlmZddlZddlZddlZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZddlmZddlmZdd lmZmZdd lmZmZd d d dddZd d dddZd ddddZd ddddZd ddddZd ddddZ d dddd Z!d d!dd"d#Z"d d$dd%d&Z#d d'dd(d)Z$d d*dd+d,Z%Gd-d.d.Z&Gd/d0d0Z'Gd1d2d2Z(Gd3d4d4Z)Gd5d6d6e)Z*Gd7d8d8e*Z+Gd9d:d:e*Z,Gd;d<dd>e*Z.Gd?d@d@e*Z/GdAdBdBe*Z0GdCd d Z1GdDdEdEZ2GdFdGdGZ3GdHdIdIe3Z4GdJdKdKe3Z5GdLdMdMe5Z6GdNdOdOe6Z7GdPdQdQe5Z8dS)Rz%Implementation of the Result classes.) annotationsN)datetime timedelta)AnyCallableDictListOptionalSequenceTupleUnion)MYSQL_CHARACTER_SETS)DbDoc)decode_from_bytes deprecated)ConnectionType FieldTypesColumnbytesr)columnpayloadreturnc Cst|dkrdS|tjkr.t|ddSztj||WStk r}z,t j |t j | dWYdSd}~XYnXdS)Nrhex) lenget_type ColumnTypeSTRINGrColumnProtoType converter_mapget_proto_typeKeyErrorsysstderrwriteencode)rrerrr(1/tmp/pip-unpacked-wheel-7_167w8m/mysqlx/result.py from_protobuf/s r*)rrcCs |ddS)Nrr(rr(r(r)bytes_from_protobuf>sr,floatcCs t|dksttd|dS)Nzd?Zd d d@dAZ dBd dCdDZ!dEd dFdGZ"dBd dHdIZ#dS)Kra_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 Nr;r Optional[int]r) rcatalogschematableoriginal_tablename original_namelength collationfractional_digitsflags content_typerc 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 r binaryZ_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_fillrrr9 _map_typerrrrrr)rrrrrrrrrrrrrinfor(r(r)rsH         zColumn.__init__rrcCst|j|j|jt|jdS)N)rrrr)rrrrrrr(r(r)rszColumn.__str__cCsj|jtjkrtj|_nD|jtjkr,tj|_n.|jtjkrBtj|_n|jrRtj|_ntj |_|j d@|_ dS)z Map bytes.r N) rrrrrrrrrrrrrr(r(r) _map_bytess       zColumn._map_bytescCsN|jdkrtj|_n6|jtj@dkr.tj|_n|jdkrBtj|_ntddS)z Map datetime.rMrrpz#Datetime mapping scenario unhandledN) rrrrrrrrr9rr(r(r) _map_datetimes     zColumn._map_datetimecCsb|jdkrtj|_nD|jdkr(tj|_n0|jdkrrWr(r)rs zBufferingResult.__init__r;zUnion[Row, DbDoc]r'cCs |j|Sr)ra)rr(r(r(r)r.szBufferingResult.__getitem__rcCs t|jS)zint: The total of items.)rrarr(r(r)countszBufferingResult.countcCsJ|j||_t|jdk|_g|_d|_d|_|j|jr@|nddS)zInitialize the result.rrqrN) r6Zget_column_metadatar^rZ_has_more_datararbrcZset_active_resultrr(r(r)rds zBufferingResult._init_resultrzOptional[Union[Row, DbDoc]]dumpingrcCsn|j|}|dkrdSdgt|d}|sdtt|dD]$}|j|}t||d|||<q>t||S)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.Row`: A `Row` object. Nfield)r6Zread_rowrrr^r*r )rrgrowrArrr(r(r) _read_items  zBufferingResult._read_itemrcCsL|jr dSd}t|jD].}|d}|dkr2qH|j||d7}q|S)zSReads the page items. Returns: int: Total items read. FrNr )r7rrbrjrarI)rrerrAr(r(r)_page_in_itemss   zBufferingResult._page_in_itemsr)col_namercCs0d}|jD] }||kr"|S|d7}q dS)zeReturns the index of the column. Returns: int: The index of the column. rr r)r^r)rrlr(colr(r(r)r+s    zBufferingResult.index_ofcCs|jr dS|dS)zuFetch one item. Returns: :class:`mysqlx.Row` or :class:`mysqlx.DbDoc`: one result item. NF)r7rjrr(r(r) fetch_one szBufferingResult.fetch_onezList[Union[Row, DbDoc]]cCs|sq q|jS)zFetch all items. Returns: `list`: The list of items of :class:`mysqlx.DbDoc` or :class:`mysqlx.Row`. )rkrarr(r(r) fetch_all+szBufferingResult.fetch_allrCcCs ||_dS)zdSets if result has data. Args: flag (bool): `True` if result has data. Nr_rDr(r(r) set_has_data7szBufferingResult.set_has_datacCs ||_dS)zfSets if has more results. Args: flag (bool): `True` if has more results. N)r`rDr(r(r)set_has_more_results?sz$BufferingResult.set_has_more_results)rcrdrerrr.rrerdrjrkr+rnrorqrrr\r(r(rWr)r]s     r]c@s0eZdZdZeddddZddddZdS) RowResultzAllows traversing the Row objects returned by a Table.select operation. Args: connection (mysqlx.connection.Connection): The Connection object. z List[Column]rcCs|jS)z`list`: The list of columns.r^rr(r(r)columnsOszRowResult.columnscCs|jS)zReturns the list of columns. Returns: `list`: The list of columns. .. versionadded:: 8.0.12 rtrr(r(r) get_columnsTszRowResult.get_columnsN)rcrdrerrrurvr(r(r(r)rsHsrsc@s:eZdZdZddddZddddZddd d Zd S) SqlResultzRepresents a result from a SQL statement. Args: connection (mysqlx.connection.Connection): The Connection object. r;rcCs|jS)zReturns the identifier for the last record inserted. Returns: str: The identifier of the last record inserted. rOrr(r(r)rXfsz!SqlResult.get_autoincrement_valuercCs|jr dSd|_|dS)zjProcess the next result. Returns: bool: Returns `True` if the fetch is done. FT)r7r`rdrr(r(r) next_resultns zSqlResult.next_resultcCs|jS)zReturns True if result has data. Returns: bool: Returns `True` if result has data. .. versionadded:: 8.0.12 rprr(r(r)has_datazszSqlResult.has_dataN)rcrdrerrXrxryr(r(r(r)rw_s rwcs(eZdZdZdddfdd ZZS) DocResultzAllows traversing the DbDoc objects returned by a Collection.find operation. Args: connection (mysqlx.connection.Connection): The Connection object. rrrfcs(t|}|dkrdStt|dS)zRead item. Args: dumping (bool): `True` for dumping. Returns: :class:`mysqlx.DbDoc`: A `DbDoc` object. Nr)rUrjrr)rrgrirWr(r)rjs zDocResult._read_item)rcrdrerrjr\r(r(rWr)rzsrz)9r __future__rrPr1r#rrtypingrrrrr r r r ZcharsetsrZdbdocrZhelpersrrtypesrrr*r,r3r5rCrGrHrJrTr\r`rarrrrrrrrrrrr r2rSr]rsrwrzr(r(r(r)sT (  $!d' j.H3}&