관리-도구
편집 파일: pg8000.cpython-38.pyc
U -?�f�H � @ s$ d Z ddlZddlZddlmZ ddlmZ ddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z ddl!m"Z" G dd � d e j#�Z$G d d � d e j%�Z&G d d � d e&e j'�Z(G d!d"� d"e&�Z)G d#d$� d$e�Z*G d%d&� d&e�Z+G d'd(� d(e jj,�Z-G d)d*� d*e jj.�Z/G d+d,� d,e jj0�Z1G d-d.� d.e�Z2G d/d0� d0e �Z3G d1d2� d2e �Z4G d3d4� d4e j5�Z6G d5d6� d6e j7�Z8G d7d8� d8e j9�Z:G d9d:� d:e j;�Z<G d;d<� d<e j=�Z>G d=d>� d>e j?�Z@G d?d@� d@e jA�ZBG dAdB� dBe jC�ZDG dCdD� dDe�ZEG dEdF� dFee�ZFG dGdH� dHejG�ZHG dIdJ� dJejI�ZJe �K� ZLG dKdL� dLe�ZMG dMdN� dN�ZNG dOdP� dPe�ZOG dQdR� dRe�ZPG dSdT� dTe�ZQeQZRdS )Ua� .. dialect:: postgresql+pg8000 :name: pg8000 :dbapi: pg8000 :connectstring: postgresql+pg8000://user:password@host:port/dbname[?key=value&key=value...] :url: https://pypi.org/project/pg8000/ .. versionchanged:: 1.4 The pg8000 dialect has been updated for version 1.16.6 and higher, and is again part of SQLAlchemy's continuous integration with full feature support. .. _pg8000_unicode: Unicode ------- pg8000 will encode / decode string values between it and the server using the PostgreSQL ``client_encoding`` parameter; by default this is the value in the ``postgresql.conf`` file, which often defaults to ``SQL_ASCII``. Typically, this can be changed to ``utf-8``, as a more useful default:: #client_encoding = sql_ascii # actually, defaults to database # encoding client_encoding = utf8 The ``client_encoding`` can be overridden for a session by executing the SQL: SET CLIENT_ENCODING TO 'utf8'; SQLAlchemy will execute this SQL on all new connections based on the value passed to :func:`_sa.create_engine` using the ``client_encoding`` parameter:: engine = create_engine( "postgresql+pg8000://user:pass@host/dbname", client_encoding='utf8') .. _pg8000_ssl: SSL Connections --------------- pg8000 accepts a Python ``SSLContext`` object which may be specified using the :paramref:`_sa.create_engine.connect_args` dictionary:: import ssl ssl_context = ssl.create_default_context() engine = sa.create_engine( "postgresql+pg8000://scott:tiger@192.168.0.199/test", connect_args={"ssl_context": ssl_context}, ) If the server uses an automatically-generated certificate that is self-signed or does not match the host name (as seen from the client), it may also be necessary to disable hostname checking:: import ssl ssl_context = ssl.create_default_context() ssl_context.check_hostname = False ssl_context.verify_mode = ssl.CERT_NONE engine = sa.create_engine( "postgresql+pg8000://scott:tiger@192.168.0.199/test", connect_args={"ssl_context": ssl_context}, ) .. _pg8000_isolation_level: pg8000 Transaction Isolation Level ------------------------------------- The pg8000 dialect offers the same isolation level settings as that of the :ref:`psycopg2 <psycopg2_isolation_level>` dialect: * ``READ COMMITTED`` * ``READ UNCOMMITTED`` * ``REPEATABLE READ`` * ``SERIALIZABLE`` * ``AUTOCOMMIT`` .. seealso:: :ref:`postgresql_isolation_level` :ref:`psycopg2_isolation_level` � N� )�ranges)�ARRAY)�_DECIMAL_TYPES)� _FLOAT_TYPES)� _INT_TYPES)�ENUM��INTERVAL)� PGCompiler)� PGDialect)�PGExecutionContext)�PGIdentifierPreparer)�JSON)�JSONB)� JSONPathType)� _SpaceVector)� OIDVECTOR)�CITEXT� )�exc)�util)� processors)�sqltypes)� quoted_namec @ s e Zd ZdZdS )� _PGStringTN��__name__� __module__� __qualname__�render_bind_cast� r! r! �U/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/postgresql/pg8000.pyr y s r c @ s e Zd ZdZdd� ZdS )� _PGNumericTc C sv | j rB|tkr t�tj| j�S |tks.|tkr2d S t � d| ��n0|tkrNd S |tks^|tkrdtj S t � d| ��d S )NzUnknown PG numeric type: %d) Z asdecimalr r Z to_decimal_processor_factory�decimal�DecimalZ _effective_decimal_return_scaler r r ZInvalidRequestErrorZto_float��self�dialect�coltyper! r! r" �result_processor� s$ ���z_PGNumeric.result_processorN�r r r r r* r! r! r! r" r# } s r# c @ s e Zd ZdZdZdS )�_PGFloat�floatTN�r r r Z__visit_name__r r! r! r! r" r, � s r, c @ s e Zd Zdd� ZdS )�_PGNumericNoBindc C s d S �Nr! )r' r( r! r! r" �bind_processor� s z _PGNumericNoBind.bind_processorN)r r r r1 r! r! r! r" r/ � s r/ c @ s e Zd ZdZdd� ZdS )�_PGJSONTc C s d S r0 r! r& r! r! r" r* � s z_PGJSON.result_processorNr+ r! r! r! r" r2 � s r2 c @ s e Zd ZdZdd� ZdS )�_PGJSONBTc C s d S r0 r! r& r! r! r" r* � s z_PGJSONB.result_processorNr+ r! r! r! r" r3 � s r3 c @ s e Zd Zdd� ZdS )�_PGJSONIndexTypec C s t d��d S )Nzshould not be here)�NotImplementedError�r' �dbapir! r! r" �get_dbapi_type� s z _PGJSONIndexType.get_dbapi_typeN�r r r r8 r! r! r! r" r4 � s r4 c @ s e Zd ZdZdZdS )�_PGJSONIntIndexTypeZjson_int_indexTNr. r! r! r! r" r: � s r: c @ s e Zd ZdZdZdS )�_PGJSONStrIndexTypeZjson_str_indexTNr. r! r! r! r" r; � s r; c @ s e Zd ZdS )�_PGJSONPathTypeN�r r r r! r! r! r" r< � s r< c @ s e Zd Zdd� ZdS )�_PGEnumc C s |j S r0 )�UNKNOWNr6 r! r! r" r8 � s z_PGEnum.get_dbapi_typeNr9 r! r! r! r" r> � s r> c @ s$ e Zd ZdZdd� Zedd� �ZdS )� _PGIntervalTc C s |j S r0 r r6 r! r! r" r8 � s z_PGInterval.get_dbapi_typec K s t |jd�S )N)� precision)r@ Zsecond_precision)�cls�interval�kwr! r! r" �adapt_emulated_to_native� s z$_PGInterval.adapt_emulated_to_nativeN)r r r r r8 � classmethodrE r! r! r! r" r@ � s r@ c @ s e Zd ZdZdS )� _PGTimeStampTNr r! r! r! r" rG � s rG c @ s e Zd ZdZdS )�_PGDateTNr r! r! r! r" rH � s rH c @ s e Zd ZdZdS )�_PGTimeTNr r! r! r! r" rI � s rI c @ s e Zd ZdZdS )� _PGIntegerTNr r! r! r! r" rJ � s rJ c @ s e Zd ZdZdS )�_PGSmallIntegerTNr r! r! r! r" rK � s rK c @ s e Zd ZdS )� _PGNullTypeNr= r! r! r! r" rL � s rL c @ s e Zd ZdZdS )� _PGBigIntegerTNr r! r! r! r" rM � s rM c @ s e Zd ZdZdS )� _PGBooleanTNr r! r! r! r" rN � s rN c @ s e Zd ZdZdS )�_PGARRAYTNr r! r! r! r" rO � s rO c @ s e Zd ZdS )� _PGOIDVECTORNr= r! r! r! r" rP � s rP c @ s e Zd Zdd� Zdd� ZdS )� _Pg8000Rangec s |j j� � fdd�}|S )Nc s& t | tj�r"� | j| j| j| j�} | S r0 )� isinstancer �Range�lower�upper�bounds�empty��value�Z pg8000_Ranger! r" �to_range s �z-_Pg8000Range.bind_processor.<locals>.to_range�r7 rS )r' r( r[ r! rZ r" r1 s z_Pg8000Range.bind_processorc C s dd� }|S )Nc S s& | d k r"t j| j| j| j| jd�} | S )N�rV rW �r rS rT rU rV Zis_emptyrX r! r! r" r[ s �z/_Pg8000Range.result_processor.<locals>.to_ranger! )r' r( r) r[ r! r! r" r* s z _Pg8000Range.result_processorN�r r r r1 r* r! r! r! r" rQ s rQ c @ s e Zd Zdd� Zdd� ZdS )�_Pg8000MultiRangec s |j j� � fdd�}|S )Nc sX t | t�rPg }| D ]8}t |tj�r@|�� |j|j|j|j�� q|�|� q|S | S d S r0 ) rR �listr rS �appendrT rU rV rW )rY �mr�vrZ r! r" � to_multirange s � z7_Pg8000MultiRange.bind_processor.<locals>.to_multiranger\ )r' r( re r! rZ r" r1 s z _Pg8000MultiRange.bind_processorc C s dd� }|S )Nc S s$ | d kr d S t �dd� | D ��S d S )Nc s s( | ] }t j|j|j|j|jd �V qdS )r] Nr^ )�.0rd r! r! r" � <genexpr>4 s � �zL_Pg8000MultiRange.result_processor.<locals>.to_multirange.<locals>.<genexpr>)r Z MultiRangerX r! r! r" re 0 s �z9_Pg8000MultiRange.result_processor.<locals>.to_multiranger! )r' r( r) re r! r! r" r* / s z"_Pg8000MultiRange.result_processorNr_ r! r! r! r" r` s r` c @ s e Zd Zdd� Zdd� ZdS )�PGExecutionContext_pg8000c C s: dt t| ��dd � t t� �dd � f }t| j�� |�S )Nzc_%s_%s� )�hex�id�_server_side_id�ServerSideCursorZ_dbapi_connection�cursor)r' �identr! r! r" �create_server_side_cursorB s *z3PGExecutionContext_pg8000.create_server_side_cursorc C s | j s d S d S r0 )�compiled�r' r! r! r" �pre_execF s z"PGExecutionContext_pg8000.pre_execN)r r r rp rs r! r! r! r" rh A s rh c @ s� e Zd ZdZdd� Zedd� �Zedd� �Zedd � �Zd d d �Z dd� Z dd� Z d dd�Z dd� Z dd� Zdd� Zd dd�Zd S ) rm Tc C s || _ || _d S r0 )ro rn )r' rn ro r! r! r" �__init__N s zServerSideCursor.__init__c C s | j jS r0 )rn � connectionrr r! r! r" ru R s zServerSideCursor.connectionc C s | j jS r0 )rn �rowcountrr r! r! r" rv V s zServerSideCursor.rowcountc C s | j jS r0 )rn � descriptionrr r! r! r" rw Z s z ServerSideCursor.descriptionr! Nc C s( d| j d | }| jj|||d� | S )NzDECLARE z NO SCROLL CURSOR FOR )�stream)ro rn �execute)r' � operation�argsrx �opr! r! r" ry ^ s zServerSideCursor.executec C s | j �||� | S r0 )rn � executemany)r' rz Z param_setsr! r! r" r} c s z ServerSideCursor.executemanyc C s | j �d| j � | j �� S )NzFETCH FORWARD 1 FROM )rn ry ro �fetchonerr r! r! r" r~ g s zServerSideCursor.fetchonec C s@ |d kr| � � S | j�dtt|�� d | j � | j� � S d S )NzFETCH FORWARD z FROM )�fetchallrn ry �str�intro )r' �numr! r! r" � fetchmanyk s �zServerSideCursor.fetchmanyc C s | j �d| j � | j �� S )NzFETCH FORWARD ALL FROM )rn ry ro r rr r! r! r" r t s zServerSideCursor.fetchallc C s | j �d| j � | j �� d S )NzCLOSE )rn ry ro �closerr r! r! r" r� x s zServerSideCursor.closec G s | j j|� d S r0 )rn � setinputsizes)r' �sizesr! r! r" r� | s z ServerSideCursor.setinputsizesc C s d S r0 r! )r' �size�columnr! r! r" � setoutputsize s z ServerSideCursor.setoutputsize)r! N)N)N)r r r � server_sidert �propertyru rv rw ry r} r~ r� r r� r� r� r! r! r! r" rm K s rm c @ s e Zd Zdd� ZdS )�PGCompiler_pg8000c K s$ | j |jf|�d | j |jf|� S )Nz %% )�process�left�right)r' �binary�operatorrD r! r! r" �visit_mod_binary� s ���z"PGCompiler_pg8000.visit_mod_binaryN)r r r r� r! r! r! r" r� � s r� c @ s e Zd Zdd� ZdS )�PGIdentifierPreparer_pg8000c O s t j| f|�|� d| _d S )NF)r rt Z_double_percents)r' r{ �kwargsr! r! r" rt � s z$PGIdentifierPreparer_pg8000.__init__N)r r r rt r! r! r! r" r� � s r� c K @ s� e Zd ZdZdZdZdZdZdZe Z e Z e ZdZdZdZe�ejejeejeeje ej e ej e ej!e"e#e$e%e%ej j&e'ej j(e)ej j*e+ej j,e-ej.e/e0e/ej1e2ej1e2ej3e4ej5e6ej7e8ej9e:ej;e<ej=e>ej?e@eAeBeCjDeEeCjFeEeCjGeEeCjHeEeCjIeEeCjJeEeCjKeLeCjMeLeCjNeLeCjOeLeCjPeLeCjQeLi$�Zd,dd�ZRejSdd� �ZTeUd d � �ZVd d � ZWd d� ZXdd� ZYdd� ZZdd� Z[dd� Z\dd� Z]dd� Z^dd � Z_d d � Z`d d � Zad-d"d#�Zbd.d$d%�Zcd&d'� Zdd(d)� ZeejSd*d+� �ZfdS )/�PGDialect_pg8000�pg8000T�formatNc K s8 t j| f|� || _| jdk r&td��| jr4td��d S )N)r � � z$pg8000 1.16.6 or greater is requiredznThe pg8000 dialect does not fully implement ipaddress type handling; INET is supported by default, CIDR is not)r rt �client_encoding�_dbapi_versionr5 �_native_inet_types)r' r� r� r! r! r" rt � s �zPGDialect_pg8000.__init__c C s8 | j r0t| j d�r0tdd� t�d| j j�D ��S dS d S )N� __version__c S s g | ] }t |��qS r! )r� )rf �xr! r! r" � <listcomp>� s �z3PGDialect_pg8000._dbapi_version.<locals>.<listcomp>z(\d+)(?:[-\.]?|$))�c r� r� )r7 �hasattr�tuple�re�findallr� rr r! r! r" r� � s ��� z PGDialect_pg8000._dbapi_versionc C s t d�S )Nr� )� __import__)rB r! r! r" � import_dbapi� s z PGDialect_pg8000.import_dbapic C s8 |j dd�}d|kr$t|d �|d< |�|j� g |fS )N�user)�username�port)Ztranslate_connect_argsr� �update�query)r' �url�optsr! r! r" �create_connect_args� s z$PGDialect_pg8000.create_connect_argsc C s* t || jj�r dt|�kr dS dt|�kS )Nz network errorTzconnection is closed)rR r7 ZInterfaceErrorr� )r' �eru rn r! r! r" � is_disconnect� s �z PGDialect_pg8000.is_disconnectc C s dS )N)� AUTOCOMMITzREAD COMMITTEDzREAD UNCOMMITTEDzREPEATABLE READZ SERIALIZABLEr! )r' �dbapi_connectionr! r! r" �get_isolation_level_values s z+PGDialect_pg8000.get_isolation_level_valuesc C sP |� dd�}|dkr d|_n0d|_|�� }|�d|� �� |�d� |�� d S )N�_� r� TFz;SET SESSION CHARACTERISTICS AS TRANSACTION ISOLATION LEVEL �COMMIT)�replaceZ autocommitrn ry r� )r' r� �levelrn r! r! r" �set_isolation_level s � z$PGDialect_pg8000.set_isolation_levelc C s< |� � }z$|�d|rdnd � |�d� W 5 |�� X d S )N�-SET SESSION CHARACTERISTICS AS TRANSACTION %sz READ ONLYz READ WRITEr� �rn r� ry �r' ru rY rn r! r! r" � set_readonly s ��z PGDialect_pg8000.set_readonlyc C s6 |� � }z|�d� |�� d }W 5 |�� X |dkS )Nzshow transaction_read_onlyr �on�rn r� ry r~ �r' ru rn �valr! r! r" � get_readonly& s z PGDialect_pg8000.get_readonlyc C s< |� � }z$|�d|rdnd � |�d� W 5 |�� X d S )Nr� Z DEFERRABLEzNOT DEFERRABLEr� r� r� r! r! r" �set_deferrable0 s ��z PGDialect_pg8000.set_deferrablec C s6 |� � }z|�d� |�� d }W 5 |�� X |dkS )Nzshow transaction_deferrabler r� r� r� r! r! r" �get_deferrable; s z PGDialect_pg8000.get_deferrablec C s8 |� � }|�d|�dd�� d�� |�d� |�� d S )NzSET CLIENT_ENCODING TO '�'z''r� )rn ry r� r� )r' r� r� rn r! r! r" �_set_client_encodingE s �� z%PGDialect_pg8000._set_client_encodingc C s |j �d|df� d S �Nr � )ru Z tpc_begin�r' ru �xidr! r! r" �do_begin_twophaseO s z"PGDialect_pg8000.do_begin_twophasec C s |j �� d S r0 )ru Z tpc_preparer� r! r! r" �do_prepare_twophaseR s z$PGDialect_pg8000.do_prepare_twophaseFc C s |j �d|df� d S r� )ru Z tpc_rollback�r' ru r� Z is_preparedZrecoverr! r! r" �do_rollback_twophaseU s z%PGDialect_pg8000.do_rollback_twophasec C s |j �d|df� d S r� )ru Z tpc_commitr� r! r! r" �do_commit_twophaseZ s z#PGDialect_pg8000.do_commit_twophasec C s dd� |j �� D �S )Nc S s g | ] }|d �qS )r r! )rf �rowr! r! r" r� ` s z8PGDialect_pg8000.do_recover_twophase.<locals>.<listcomp>)ru Z tpc_recover)r' ru r! r! r" �do_recover_twophase_ s z$PGDialect_pg8000.do_recover_twophasec s� g � dd� }� � |� �jd k r6�fdd�}� � |� �jdkrRdd� }� � |� �jrn�fdd�}� � |� t� �dkr�� fdd�}|S d S d S ) Nc S s | j t | j t< d S r0 )Zpy_typesr� r ��connr! r! r" � on_connecte s z/PGDialect_pg8000.on_connect.<locals>.on_connectc s � � | � j� d S r0 )r� r� r� rr r! r" r� l s Fc S s$ | � ddd� � | � ddd� � d S )Nie c S s | S r0 r! ��sr! r! r" �<lambda>u � zAPGDialect_pg8000.on_connect.<locals>.on_connect.<locals>.<lambda>i� c S s | S r0 r! r� r! r! r" r� x r� )�register_in_adapterr� r! r! r" r� s s c s | � d� j� | � d� j� d S )N�r i� )r� �_json_deserializerr� rr r! r" r� ~ s r c s � D ] }|| � qd S r0 r! )r� �fn)�fnsr! r" r� � s )rb r� r� r� �len)r' r� r! )r� r' r" r� b s zPGDialect_pg8000.on_connectc C s dS )N�;r! rr r! r! r" � _dialect_specific_select_one� s z-PGDialect_pg8000._dialect_specific_select_one)N)TF)TF)gr r r ZdriverZsupports_statement_cacheZsupports_unicode_statementsZsupports_unicode_bindsZdefault_paramstyleZ supports_sane_multi_rowcountrh Zexecution_ctx_clsr� Zstatement_compilerr� �preparerZ supports_server_side_cursorsr Zdescription_encodingr Z update_copyr Zcolspecsr �Stringr �Numericr/ �Floatr, r r2 �BooleanrN �NullTyperL r r3 r r r< � JSONIndexTyper4 �JSONIntIndexTyper: �JSONStrIndexTyper; ZIntervalr@ r �DateTimerG �DaterH �TimerI �IntegerrJ � SmallIntegerrK � BigIntegerrM �Enumr> r rO r rP r Z INT4RANGErQ Z INT8RANGEZNUMRANGEZ DATERANGEZTSRANGEZ TSTZRANGEZINT4MULTIRANGEr` ZINT8MULTIRANGEZ NUMMULTIRANGEZDATEMULTIRANGEZ TSMULTIRANGEZTSTZMULTIRANGErt Zmemoized_propertyr� rF r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r! r! r! r" r� � s� ��* � � /r� )S�__doc__r$ r� r� r �arrayr ZPGARRAY�baser r r r r r r r r �jsonr r r Z pg_catalogr r �typesr r r Zenginer �sqlr Z sql.elementsr r� r r� r# r� r, r/ r2 r3 r� r4 r� r: r� r; r< r> r@ r� rG r� rH r� rI r� rJ r� rK r� rL r� rM r� rN rO rP ZAbstractSingleRangeImplrQ ZAbstractMultiRangeImplr` �counterrl rh rm r� r� r� r( r! r! r! r" �<module> sr U # 8