관리-도구
편집 파일: result.cpython-38.pyc
U )�e+\ � @ s� d Z ddlZddlZddlZddlmZmZ ddlmZ ddlm Z ddl m Z dId d �Z d d � Z d d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � ZG d!d"� d"e�ZG d#d$� d$e�ZG d%d&� d&e�ZG d'd(� d(e�Z G d)d*� d*e �Z G d+d,� d,e �Z G d-d.� d.e �Z G d/d0� d0e �Z G d1d2� d2e �Z!G d3d4� d4e �Z"G d5d6� d6e �Z#G d7d8� d8e�Z$G d9d:� d:e�Z%G d;d<� d<e�Z&G d=d>� d>e�Z'G d?d@� d@e'�Z(G dAdB� dBe'�Z)G dCdD� dDe)�Z*G dEdF� dFe*�Z+G dGdH� dHe)�Z,dS )Jz)Implementation of the Result Set classes.� N)�datetime� timedelta� )�DbDoc)�MYSQL_CHARACTER_SETS)� STRING_TYPES�utf-8c C s t | t�r| �|�S | S �N)� isinstance�bytes�decode)�value�encoding� r �b/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysqlx/result.py�decode_from_bytes% s r c C s� t |�dkrd S zPtjttjttjttjt tj t tj t tjttjttjttjttjti | |�W S tk r� } z4tj�d�|�� tj�d�|�d��� W Y �d S d }~X Y nX d S )Nr �{0}�hex) �len�ColumnProtoType�SINT�varsint_from_protobuf�UINT�varint_from_protobuf�BYTES�bytes_from_protobuf�DATETIME�datetime_from_protobuf�TIME�time_from_protobuf�FLOAT�float_from_protobuf�DOUBLE�double_from_protobuf�BIT�SET�set_from_protobuf�ENUM�DECIMAL�decimal_from_protobuf�KeyError�sys�stderr�write�format�encode)�col_type�payload�er r r � from_protobuf) sD � � � r3 c C s | d d� S )N���r �r1 r r r r A s r c C s t | �dkst�t�d| �S )N� z<f�r �AssertionError�struct�unpackr5 r r r r! F s r! c C s t | �dkst�t�d| �S )N� z<dr7 r5 r r r r# K s r# c C s� t | �dkrtd��d}d}d}| D ]j}t|t�r6|nt|�}|d@ dk}|d@ }||K }||O }|r~|| |d d � f S |d7 }|d7 }q$td��d S )Nr zPayload is empty� � r � zPayload too short)r � ValueErrorr �int�ord�EOFError)r1 �cur�i�shift�c�chZeosZcur_bitsr r r �varint_from_protobuf_streamP s rH c C s$ t | �\}} t| �dkr td��|S )Nr �Payload too long�rH r r? �r1 rD r r r r f s r c C sL t | �\}} t| �dkr td��|d@ r@| }|d? }|dO }n|d? }|S )Nr rI r l rJ rK r r r r n s r c C s� g }zxt | �\}} t| �|k rLt| �dkrD|dkrDt|�dkrDg W S td��|�| d |� � | |d � } t| �dkrzW q�W q tk r� Y q�Y qX q|S )Nr r zInvalid Set encoding)rH r r? �append)r1 �sZ field_lenr r r r& ~ s r& c C s� g }d }t | d �}| dd � } | D ]�}t |�}|d@ d? }|d@ }|dk r�|�|� |dk rj|�|� q�|dkr|d} q�q�|dkr�d} q�q�td ��q$|dkr�d}|dks�t� q�q$|dkr�d}|dks�t� q�q$td �|���q$t�||| f�S ) Nr r �� r6 � � � � z Invalid BCDzInvalid BCD: {0})rA rL r? r8 r. �decimal�Decimal)r1 �digits�sign�scalerF rG Zhigh_bcdZlow_bcdr r r r) � s: r) c C s� d}d}d}d}t | �\}} t | �\}} t | �\}} z4t | �\}} t | �\}} t | �\}} t | �\}} W n tk r| Y nX t|||||||�S �Nr )rH r? r )r1 �hour�minutes�seconds�useconds�year�month�dayr r r r � s r c C s� d}d}d}d}| d dk}| dd � } z4t | �\}} t | �\}} t | �\}} t | �\}} W n tk rp Y nX |r�|r�|d9 }n(|r�|d9 }n|r�|d9 }n |r�|d9 }t||||d�S )Nr r r4 )�hoursrZ r[ � microseconds)rH r? r )r1 rY rZ r[ r\ �negater r r r � s2 �r c @ s e Zd ZdZdS )� Collations�! N)�__name__� __module__� __qualname__ZUTF8_GENERAL_CIr r r r rc � s rc c @ s� e Zd ZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Z!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&d0d1� �Z-e&d2d3� �Z.d4S )5� ColumnTyper � � r6 � � r> r; � rP � rQ rR � rO � � � � � � � � � � � � � � � � � rd �" �# c C s* t | ��� D ]\}}||kr | S q d S r )�vars�items)�cls�needle�keyr r r r � to_string s zColumnType.to_stringc C s t | |�� d �S r )�getattr�upper)r� r� r r r � from_string s zColumnType.from_stringc C s || j | jfkS r )�CHAR�VARCHAR�r� r0 r r r �is_char s zColumnType.is_charc C s || j | jfkS r )�BINARY� VARBINARYr� r r r � is_binary# s zColumnType.is_binaryc C s || j | j| j| jfkS r )�TEXT�TINYTEXT� MEDIUMTEXT�LONGTEXTr� r r r �is_text' s �zColumnType.is_textc C s || j | j| j| j| jfkS r )�REALr" r r( �NUMERICr� r r r � is_decimals, s �zColumnType.is_decimalsc C s || j | j| j| j| j| jfkS r )r$ �TINYINT�SMALLINT� MEDIUMINT�INT�BIGINTr� r r r � is_numeric1 s �zColumnType.is_numericc C s || j | jfkS r )r% r' r� r r r � is_finite_set6 s zColumnType.is_finite_setN)/re rf rg r$ r� r� r� r� r� r� r r( r� r" �JSON�STRINGr r �DATEr � TIMESTAMPr% r' �GEOMETRY�XMLZYEARr� r� r� r� ZTINYBLOBZBLOBZ MEDIUMBLOBZLONGBLOBr� r� r� r� � classmethodr� r� r� r� r� r� r� r� r r r r rh � sf rh c @ s8 e Zd ZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S ) r r ri rk rl r> rP rQ rO rp rq rr N)re rf rg r r r" r r r r r% r' r$ r( r r r r r ; s r c @ s6 e Zd Zdd� Zdd� Zedd� �Zejdd� �ZdS ) �Flagsc C sZ i | _ i | _| jj�� D ]6\}}|�d�r,qt|�tfkr|| j |< || j|< q|| _d S )N�__) Z_allowed_flags� _flag_names� __class__�__dict__r� � startswith�typer@ r )�selfr �k�vr r r �__init__J s zFlags.__init__c C s� d}g }| j }tdd�D ]l}|dK }||@ }|rz|�| j| � W q tk r� tj�d�| j�� tj�d�| j j �� Y qX qd� |�S )Nr r �? r �,) r �rangerL r� r* r+ r, r- r. r� r� �join)r� �maskZ flag_namesr �_�flagr r r �__str__U s z Flags.__str__c C s | j S r ��_value�r� r r r r g s z Flags.valuec C s || _ d S r r� )r� r� r r r r k s N)re rf rg r� r� �propertyr �setterr r r r r� I s r� c @ s e Zd ZdZdZdZdZdZdS )� ColumnFlagsrp r� �@ r< � N)re rf rg ZNOT_NULLZ PRIMARY_KEYZ UNIQUE_KEYZ MULTIPLE_KEYZAUTO_INCREMENTr r r r r� p s r� c @ s e Zd ZdZdS )�DatetimeColumnFlagsr N)re rf rg r� r r r r r� x s r� c @ s e Zd ZdZdS )�UIntColumnFlagsr N)re rf rg ZZEROFILLr r r r r� | s r� c @ s e Zd ZdZdS )�DoubleColumnFlagsr N�re rf rg �UNSIGNEDr r r r r� � s r� c @ s e Zd ZdZdS )�FloatColumnFlagsr Nr� r r r r r� � s r� c @ s e Zd ZdZdS )�BytesColumnFlagsr N)re rf rg Z RIGHT_PADr r r r r� � s r� c @ s e Zd ZdZdZdZdS )�BytesContentTyper ri rj N)re rf rg r� r� r� r r r r r� � s r� c @ s� e Zd Zd*dd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� ZdS )+�ColumnMetaDataNc C s t |�| _t |�| _t |�| _t |�| _t |�| _|| _d | _|| _|| _ | | _ | | _ | | _ | | _ d| _d| _d| _d| _d | _d | _| j dkr�| j tt�kr�td�| j ���t| j } | d | _| d | _d| jkp�d| jk| _| �� | jtjtjtj tj tj fk| _d S )NFr z"No mapping found for collation {0}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_namer r r? r. � _map_typerh r� r� r� r r� )r� r0 �catalog�schema�tableZoriginal_table�name� original_name�lengthZ collationZfractional_digits�flags� content_type�infor r r r� � sL � � �zColumnMetaData.__init__c C s t | j| j| jt | j�d��S )N)r0 r� r� r� )�strr� r� r� r� r� r r r r� � s �zColumnMetaData.__str__c C s | j S r )r� r� r r r �get_schema_name� s z ColumnMetaData.get_schema_namec C s | j p | jS r )r� r� r� r r r �get_table_name� s z ColumnMetaData.get_table_namec C s | j p | jS r )r� r� r� r r r �get_table_label� s z ColumnMetaData.get_table_labelc C s | j p | jS r )r� r� r� r r r �get_column_name� s z ColumnMetaData.get_column_namec C s | j p | jS r )r� r� r� r r r �get_column_label� s z ColumnMetaData.get_column_labelc C s | j S r )r� r� r r r �get_type� s zColumnMetaData.get_typec C s | j S r )r� r� r r r � get_length� s zColumnMetaData.get_lengthc C s | j S r )r� r� r r r �get_fractional_digits� s z$ColumnMetaData.get_fractional_digitsc C s | j S r )r� r� r r r �get_collation_name� s z!ColumnMetaData.get_collation_namec C s | j S r )r� r� r r r �get_character_set_name� s z%ColumnMetaData.get_character_set_namec C s | j S r )r� r� r r r �is_number_signed� s z ColumnMetaData.is_number_signedc C s | j S r )r� r� r r r � is_padded� s zColumnMetaData.is_paddedc C s | j S r )r� r� r r r �is_bytes� s zColumnMetaData.is_bytesc C sb | j dkrtj| _nD| j dkr(tj| _n0| j dkr<tj| _n | j dkrPtj| _ntj| _d| _d S )Nr6 rl rm rn T) r� rh r� r� r� r� r� r� r� r� r r r � _map_int_type� s z ColumnMetaData._map_int_typec C sh | j dkrtj| _nD| j dkr(tj| _n0| j dkr<tj| _n | j dkrPtj| _ntj| _| jd@ | _ d S )Nrj rk r; rP r ) r� rh r� r� r� r� r� r� r� Z _zero_fillr� r r r �_map_uint_type� s z ColumnMetaData._map_uint_typec C sj | j tjkrtj| _nD| j tjkr,tj| _n.| j tjkrBtj| _n| jrRtj| _ntj | _| j d@ | _ d S )Nr ) r� r� r� rh r� r� r� r� r r� r� r� r� r r r � _map_bytes s zColumnMetaData._map_bytesc C sN | j dkrtj| _n6| j dkr(tj| _n"| jtj@ dkrBtj| _ntd��d S )NrP rs r z#Datetime mapping scenario unhandled) r� rh r� r� r r� r� r� r? r� r r r � _map_datetime s z ColumnMetaData._map_datetimec C sF | j tjkr| �� �n*| j tjkr0| �� �n| j tjkrXtj| _| j t j @ dk| _ n�| j tj kr�tj | _| j t j @ dk| _ n�| j tjkr�tj| _| j t j @ dk| _ n�| j tjkr�| �� n�| j tjkr�tj| _nn| j tjkr�| �� nX| j tjk�rtj| _n@| j tjk�rtj| _n(| j tjk�r2tj| _ntd�| j ���d S )Nr zUnknown column type {0})r� r r r� r r� r rh r� r� r� r� Z_is_number_signedr( r" r r� r r r� r% r' r$ r? r. r� r r r r� s: � � � zColumnMetaData._map_type) NNNNNNNNNNN)re rf rg r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s: � & r� c @ s8 e Zd Zdd� Zedd� �Zedd� �Zedd� �Zd S ) �Warningc C s || _ || _|| _d S r )�_level�_code�_message)r� �level�code�msgr r r r� ; s zWarning.__init__c C s | j S r )r� r� r r r �Level@ s z Warning.Levelc C s | j S r )r� r� r r r �CodeD s z Warning.Codec C s | j S r )r� r� r r r �MessageH s zWarning.MessageN)re rf rg r� r� r r r r r r r r� : s r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �Rowz�Represents a row element returned from a SELECT query. Args: rs (mysqlx.Result): The Result set. fields (list): The list of fields. c C s || _ || _d S r )�_fields� _resultset)r� �rs�fieldsr r r r� T s z Row.__init__c C s8 t |t�r| j�|�}n|t| j�kr.td��| j| S )NzIndex out of range)r r r �index_ofr r � IndexError�r� �indexr r r � __getitem__X s zRow.__getitem__c C sF | j �|�}|t| j�kr"td��|dkr8td�|���t| j| �S )zpReturns the value if the index by string. Args: str_index (str): The string index. zArgument out of ranger4 zColumn name '{0}' not found)r r r r r r? r. r� )r� Z str_indexZ int_indexr r r � get_string_ s zRow.get_stringN)re rf rg �__doc__r� r r r r r r r M s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � BaseResultz�Provides base functionality for result objects. Args: connection (mysqlx.connection.Connection): The Connection object. c C sB || _ d| _d| _d| _g | _|d kr.d | _n|j| _|�� d S )NFr r4 )� _connection�_closed�_rows_affected� _generated_id� _warnings� _protocol�protocolZfetch_active_result�r� � connectionr r r r� s s zBaseResult.__init__c C s | j S )zXReturns the warnings. Returns: list: The list of warnings. )r r� r r r � get_warnings� s zBaseResult.get_warningsc C s t | j�S )zcReturns the number of warnings. Returns: int: The number of warnings. )r r r� r r r �get_warnings_count� s z BaseResult.get_warnings_countN)re rf rg r r� r r r r r r r m s r c sB e Zd ZdZd � fdd� Zdd� Zdd� Zd d � Zd d � Z� Z S )�Resultz�Allows retrieving information about non query operations performed on the database. Args: connection (mysqlx.connection.Connection): The Connection object. ids (list): A list of IDs. Nc s. t t| ��|� || _|d k r*| j�| � d S r )�superr r� �_idsr Z close_result)r� r Zids�r� r r r� � s zResult.__init__c C s | j S )z�Returns the number of affected items for the last operation. Returns: int: The number of affected items. )r r� r r r �get_affected_items_count� s z Result.get_affected_items_countc C s | j S )zjReturns the last insert id auto generated. Returns: int: The last insert id. �r r� r r r �get_autoincrement_value� s z Result.get_autoincrement_valuec C s* | j dkrdS t| j �dkr dS | j d S )zDReturns ID of the last document inserted into a collection. Nr )r r r� r r r �get_document_id� s zResult.get_document_idc C s | j S )z5Returns the list of generated documents IDs. )r r� r r r �get_document_ids� s zResult.get_document_ids)NN) re rf rg r r� r" r$ r% r& � __classcell__r r r! r r � s r c s` e Zd Z� fdd�Zdd� Zedd� �Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z � Z S )�BufferingResultc s t t| ��|� | �� d S r )r r( r� � _init_resultr r! r r r� � s zBufferingResult.__init__c C sN | j �| �| _t| j�dkr dnd| _g | _d| _d| _| jrB| nd | j _d S )Nr TFrt r4 ) r Zget_column_metadata�_columnsr Z_has_more_data�_items� _page_sizeZ _positionZ_active_resultr� r r r r) � s z BufferingResult._init_resultc C s t | j�S )z!int: The total of items. )r r+ r� r r r �count� s zBufferingResult.countc C s | j | S r )r+ r r r r r � s zBufferingResult.__getitem__c C s0 d}| j D ] }|�� |kr"| S |d }q dS )zeReturns the index of the column. Returns: int: The index of the column. r r r4 )r* r� )r� Zcol_namer �colr r r r � s zBufferingResult.index_ofc C s� | j �| �}|d krd S d gt|j� }|sztt|j��D ]>}| j| }|�� r^t|j| �n|j| }t|j |�||< q:t | |�S r ) r Zread_rowr �fieldr� r* r� r r3 r� r )r� �dumping�row�item�xr. r r r r � _read_item� s ��zBufferingResult._read_itemc C sL | j r dS d}t| j�D ].}| �d�}|d kr2 qH| j�|� |d7 }q|S )NFr r )r r� r, r4 r+ rL )r� r- rD r2 r r r �_page_in_items� s z BufferingResult._page_in_itemsc C s | j r dS | �d�S )zS Fetch one item. Returns: Row/DbDoc: one result item. NF)r r4 r� r r r � fetch_one� s zBufferingResult.fetch_onec C s | � � s q q | jS )zPFetch all items. Returns: list: The list of items. )r5 r+ r� r r r � fetch_all s zBufferingResult.fetch_all)re rf rg r� r) r� r- r r r4 r5 r6 r7 r' r r r! r r( � s r( c s, e Zd ZdZ� fdd�Zedd� �Z� ZS )� RowResultz�Allows traversing the Row objects returned by a Table.select operation. Args: connection (mysqlx.connection.Connection): The Connection object. c s t t| ��|� d S r )r r8 r� r r! r r r� s zRowResult.__init__c C s | j S )z#list: The list of columns. )r* r� r r r �columns s zRowResult.columns)re rf rg r r� r� r9 r' r r r! r r8 s r8 c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )� SqlResultz�Represents a result from a SQL statement. Args: connection (mysqlx.connection.Connection): The Connection object. c s t t| ��|� d| _d S )NF)r r: r� �_has_more_resultsr r! r r r� ) s zSqlResult.__init__c C s | j S )z=Returns the identifier for the last record inserted. r# r� r r r r$ - s z!SqlResult.get_autoincrement_valuec C s | j r dS d| _| �� dS )NFT)r r; r) r� r r r � next_result2 s zSqlResult.next_result)re rf rg r r� r$ r< r' r r r! r r: # s r: c s, e Zd ZdZ� fdd�Z� fdd�Z� ZS )� DocResultz�Allows traversing the DbDoc objects returned by a Collection.find operation. Args: connection (mysqlx.connection.Connection): The Connection object. c s t t| ��|� d S r )r r= r� r r! r r r� A s zDocResult.__init__c s, t t| ��|�}|d kr d S tt|d ��S rX )r r= r4 r r )r� r0 r1 r! r r r4 D s zDocResult._read_item)re rf rg r r� r4 r' r r r! r r= : s r= )r )-r rS r9 r+ r r Zdbdocr Zcharsetsr �compatr r r3 r r! r# rH r r r& r) r r �objectrc rh r r� r� r� r� r� r� r� r� r� r� r r r r( r8 r: r= r r r r �<module> sP $ K' ) $.T