관리-도구
편집 파일: connection.cpython-38.pyc
U )�e�@ � @ s& d Z zddlZdZW n dZY nX ddlZddlZddlmZ ddlmZ ddl m Z m Z m Z dd l mZ dd lmZmZ dd lmZmZmZ dd lmZmZ d ZdZG dd� de�Z dd� Z G dd� de�Z G dd� de �Z G dd� de �Z G dd� de�Z!G dd � d e!�Z"G d d � d e!�Z#dS ) z4Implementation of communication for MySQL X servers.� NTF��wraps� )�MySQL41AuthPlugin)�InterfaceError�OperationalError�ProgrammingError��Schema)�Protocol�MessageReaderWriter)�Result� RowResult� DocResult)� SqlStatement� AddStatementz DROP DATABASE IF EXISTS `{0}`z#CREATE DATABASE IF NOT EXISTS `{0}`c @ s@ e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zi fd d �Zd S )� SocketStreamc C s d | _ d| _d S )NF)�_socket�_is_ssl��self� r �f/opt/hc_python/lib64/python3.8/site-packages/../../../lib/python3.8/site-packages/mysqlx/connection.py�__init__2 s zSocketStream.__init__c C s6 t |t�rtjntj}t�|tj�| _| j�|� d S �N)� isinstance�tuple�socket�AF_INET�AF_UNIX� SOCK_STREAMr �connect)r �paramsZs_typer r r r! 6 s zSocketStream.connectc C s\ | j d krtd��g }|dkrR| j �|�}|dkr:td��|�|� |t|�8 }qd�|�S )N� MySQLx Connection not availabler � zUnexpected connection close)r r �recv� RuntimeError�append�len�join)r �count�buf�datar r r �read; s zSocketStream.readc C s" | j d krtd��| j �|� d S )Nr# )r r �sendall�r r, r r r r. G s zSocketStream.sendallc C s | j s d S | j �� d | _ d S r )r �closer r r r r0 L s zSocketStream.closec C sL t s| �� td��t�tj�}|�� d|krvz|�|d � tj|_ W n* t tj fk rt | �� t d��Y nX d|kr�z|�|d � tj |_W n* t tj fk r� | �� t d��Y nX d|k�rz |�|d |�dd �� W n, t tj fk �r | �� t d��Y nX nd|k�r4| �� t d ��|�| j�| _d | _d S ) Nz'Python installation has no SSL support.zssl-cazInvalid CA certificate.zssl-crlz Invalid CRL.zssl-certzssl-keyz Invalid Client Certificate/Key.z Client Certificate not provided.T)� SSL_AVAILABLEr0 r& �ssl� SSLContext�PROTOCOL_SSLv23�load_default_certs�load_verify_locations� CERT_REQUIRED� verify_mode�IOError�SSLErrorr �VERIFY_CRL_CHECK_CHAIN� verify_flags�load_cert_chain�get� wrap_socketr r )r Zssl_opts�contextr r r �set_sslS s@ � zSocketStream.set_sslN) �__name__� __module__� __qualname__r r! r- r. r0 rA r r r r r 1 s r c s t � �� fdd��}|S )Nc sB z� | f|�|�W S t jtfk r< | �� td��Y nX d S )NzCannot connect to host.)r �errorr& � disconnectr )r �args�kwargs��funcr r �wrapperx s z(catch_network_exception.<locals>.wrapperr )rJ rK r rI r �catch_network_exceptionw s rL c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Ze d d� �Z e dd� �Z e dd� �Z e dd� �Z e dd� �Ze d+dd��Ze dd� �Ze d d � �Ze d d � �Ze d d!� �Ze d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*S ),� Connectionc C sH |� d�| _|� d�| _|� d�| _d | _|| _t� | _d | _d | _ d S )N�user�password�schema) r> �_user� _passwordZ_schema�_active_result�settingsr �stream� reader_writer�protocol�r rT r r r r � s zConnection.__init__c C s | j d k r| j �� d | _ d S r )rS � fetch_allr r r r �fetch_active_result� s z Connection.fetch_active_resultc C s: d| j kr"| j d | j �dd�fS d| j kr6| j d S dS )N�host�port�$� r )� localhostr] )rT r> r r r r �_connection_params� s z Connection._connection_paramsc C s< | j �| �� � t| j �| _t| j�| _| �� | �� d S r ) rU r! r_ r rV r rW �_handle_capabilities� _authenticater r r r r! � s zConnection.connectc C s� | j �� j}|r$|d j�� dksJndsJ| j�dd�rF| �� td��d S t j dk rv| j�dd�rr| �� t d��d S | j j dd � | j �| j� d S ) Nr �tlsFz ssl-enablezSSL not enabled at server.)� � � z=The support for SSL is not available for this Python version.T)rb )rW Zget_capabilitesZ capabilities�name�lowerrT r> r0 r �sys� version_infor& Zset_capabilitiesrU rA r/ r r r r` � s z Connection._handle_capabilitiesc C sH t | j| j�}| j�|�� � | j�� }| j�|�|�� | j� � d S r ) r rQ rR rW Zsend_auth_startZ auth_nameZread_auth_continueZsend_auth_continueZ build_authentication_responseZ read_auth_ok)r �plugin� extra_datar r r ra � s �zConnection._authenticatec G s | j �d||� d S )N�sql)rW �send_execute_statement)r rl rG r r r �send_sql� s zConnection.send_sqlc C s* | j �|� d }t|t�r |j}t| |�S r )rW � send_insertr r Z_idsr )r � statementZidsr r r ro � s zConnection.send_insertc C s" | j �|� |jrt| �S t| �S r )rW Z send_findZ _doc_basedr r �r rp r r r �find� s zConnection.findc C s | j �|� t| �S r )rW Z send_deleter rq r r r �delete� s zConnection.deletec C s | j �|� t| �S r )rW Z send_updater rq r r r �update� s zConnection.updateTc G s | j �|||� t| �S r )rW rm r )r � namespace�cmdZ raise_on_failrG r r r �execute_nonquery� s zConnection.execute_nonqueryc G s> | j �d||� t| �}|�� |jdkr2td��|d d S )Nrl r z No data found)rW rm r rY r* r )r rl rG �resultr r r �execute_sql_scalar� s z Connection.execute_sql_scalarc G s | j �d||� t| �S )NZxplugin)rW rm r )r rv rG r r r �get_row_result� s zConnection.get_row_resultc C s | j �|�S r )rW �read_row�r rx r r r r{ � s zConnection.read_rowc C s | j �|� d S r )rW � close_resultr| r r r r} � s zConnection.close_resultc C s | j �|�S r )rW �get_column_metadatar| r r r r~ � s z Connection.get_column_metadatac C s | j jd k S r )rU r r r r r �is_open� s zConnection.is_openc C s | � � s d S | j�� d S r )r rU r0 r r r r rF � s zConnection.disconnectc C sB | � � s d S | jd k r | j�� | j�� | j�� | j�� d S r )r rS rY rW Z send_close�read_okrU r0 r r r r r0 � s zConnection.closeN)T)rB rC rD r rZ r_ r! r` ra rL rn ro rr rs rt rw ry rz r{ r} r~ r rF r0 r r r r rM � s> rM c s\ e Zd Z� fdd�Zdd� Z� fdd�Z� fdd�Zd d � Z� fd d �Z� fd d�Z � Z S )� XConnectionc s� t t| ��|� g | _|�dg �| _d|krT|d rT| j�|�d�|�dd �d�� d| _d| _| � � | jj dd� dd � d S ) NZroutersr[ r\ )r[ r\ ���Tc S s | d S )N�priorityr )�xr r r �<lambda> r$ z&XConnection.__init__.<locals>.<lambda>)�key�reverse) �superr� r �dependent_connections�pop�_routersr' � _cur_router� _can_failover�_ensure_priorities�sortrX �� __class__r r r s �zXConnection.__init__c C s� d}d}| j D ]D}|�dd �}|d kr8|d7 }||d< n|dkrJtdd��|d8 }qd| k rnt| j �k r|n n tdd��d S ) Nr �d r� r z(The priorities must be between 0 and 100i� z\You must either assign no priority to any of the routers or give a priority for every routeri� )r� r> r r( )r Zpriority_countr� �router�prir r r r� s � �z XConnection._ensure_prioritiesc s� | j sd| _tt| ��� S | jr*| jdkrJd| _d| _| j D ] }d|d< q<| jd7 _| j | j d }| j | j d }| jdkr�d| j | jd d< | jt| j �d kr�d| _||fS ) NFr� T� availabler r[ r\ r )r� r� r� r� r_ r� r( )r r� r[ r\ r� r r r_ ) s z XConnection._connection_paramsc sp d }| j rFztt| ��� W S tjk rB } z|}W 5 d }~X Y qX qt| j�dkrbtd� |���tdd��d S )Nr �Cannot connect to host: {0}z(Failed to connect to any of the routers.i� ) r� r� r� r! r rE r( r� r �format)r rE �errr� r r r! @ s zXConnection.connectc C s | j �|� d S r )r� r' )r � connectionr r r �bind_connectionM s zXConnection.bind_connectionc s( | j r| j �� �� q tt| ��� d S r )r� r� r0 r� r� r r� r r r0 P s zXConnection.closec s( | j r| j �� �� q tt| ��� d S r )r� r� rF r� r� r r� r r rF U s zXConnection.disconnect) rB rC rD r r� r_ r! r� r0 rF � __classcell__r r r� r r� s r� c s( e Zd Z� fdd�Z� fdd�Z� ZS )�NodeConnectionc s t t| ��|� d S r )r� r� r rX r� r r r \ s zNodeConnection.__init__c sJ zt t| ��� W n2 tjk rD } ztd�|���W 5 d }~X Y nX d S )Nr� )r� r� r! r rE r r� )r r� r� r r r! _ s zNodeConnection.connect)rB rC rD r r! r� r r r� r r� [ s r� c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )� BaseSessiona� Base functionality for Session classes through the X Protocol. This class encloses the core functionality to be made available on both the XSession and NodeSession classes, such functionality includes: - Accessing available schemas. - Schema management operations. - Enabling/disabling warning generation. - Retrieval of connection information. Args: settings (dict): Connection data used to connect to the database. c C s || _ d S r )� _settingsrX r r r r t s zBaseSession.__init__c C s | j jjd k S r )� _connectionrU r r r r r r w s zBaseSession.is_openc C s t | |�S )z�Retrieves a Schema object from the current session by it's name. Args: name (string): The name of the Schema object to be retrieved. Returns: mysqlx.Schema: The Schema object with the given name. r �r rf r r r � get_schemaz s zBaseSession.get_schemac C s, | j j�d�r t| | j jd �S td��dS )aK Retrieves a Schema object from the current session by the schema name configured in the connection settings. Returns: mysqlx.Schema: The Schema object with the given name at connect time. Raises: ProgrammingError: If default schema not provided. rP zDefault schema not providedN)r� rT r> r r r r r r �get_default_schema� s z BaseSession.get_default_schemac C s | j �dt�|�d� dS )z�Drops the schema with the specified name. Args: name (string): The name of the Schema object to be retrieved. rl TN)r� rw �_DROP_DATABASE_QUERYr� r� r r r � drop_schema� s �zBaseSession.drop_schemac C s | j �dt�|�d� t| |�S )z�Creates a schema on the database and returns the corresponding object. Args: name (string): A string value indicating the schema name. rl T)r� rw �_CREATE_DATABASE_QUERYr� r r� r r r � create_schema� s �zBaseSession.create_schemac C s | j �ddd� dS )z4Starts a transaction context on the server. rl zSTART TRANSACTIONTN�r� rw r r r r �start_transaction� s z BaseSession.start_transactionc C s | j �ddd� dS )zXCommits all the operations executed after a call to startTransaction(). rl ZCOMMITTNr� r r r r �commit� s zBaseSession.commitc C s | j �ddd� dS )zYDiscards all the operations executed after a call to startTransaction(). rl ZROLLBACKTNr� r r r r �rollback� s zBaseSession.rollbackc C s | j �� d S r )r� r0 r r r r r0 � s zBaseSession.closeN)rB rC rD �__doc__r r r� r� r� r� r� r� r� r0 r r r r r� f s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�XSessionaW Enables interaction with a X Protocol enabled MySQL Product. The functionality includes: - Accessing available schemas. - Schema management operations. - Enabling/disabling warning generation. - Retrieval of connection information. Args: settings (dict): Connection data used to connect to the database. c s* t t| ��|� t| j�| _| j�� d S r )r� r� r r� r� r� r! rX r� r r r � s zXSession.__init__c C s, | � � std��t| j�}| j�|j� |S )Nz$XSession is not connected to a farm.)r r � NodeSessionr� r� r� )r Znsessr r r �bind_to_default_shard� s z XSession.bind_to_default_shard)rB rC rD r� r r� r� r r r� r r� � s r� c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r� at Enables interaction with a X Protocol enabled MySQL Server. The functionality includes: - Accessing available schemas. - Schema management operations. - Enabling/disabling warning generation. - Retrieval of connection information. - Includes SQL Execution. Args: settings (dict): Connection data used to connect to the database. c s* t t| ��|� t| j�| _| j�� d S r )r� r� r r� r� r� r! rX r� r r r � s zNodeSession.__init__c C s t | j|�S )z}Creates a :class:`mysqlx.SqlStatement` object to allow running the SQL statement on the target MySQL Server. )r r� )r rl r r r rl � s zNodeSession.sql)rB rC rD r� r rl r� r r r� r r� � s r� )$r� r2 r1 rh r � functoolsr Zauthenticationr �errorsr r r Zcrudr rW r r rx r r r rp r r r� r� �objectr rL rM r� r� r� r� r� r r r r �<module> s2 F U W