Edit File: pg8000.cpython-37.pyc
B ��4]�% � @ sp d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd lm Z dd lm Z dd l m Z d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ydd lm Z W n ek r� dZY nX G dd� dej�ZG dd� de�ZG dd� de �Z G dd� de �Z G dd � d e �Z G d d � d e�Z G d d � d e �Z G d!d"� d"e�Z!e!Z"dS )#a� .. dialect:: postgresql+pg8000 :name: pg8000 :dbapi: pg8000 :connectstring: postgresql+pg8000://user:password@host:port/dbname[?key=value&key=value...] :url: https://pythonhosted.org/pg8000/ .. note:: The pg8000 dialect is **not tested as part of SQLAlchemy's continuous integration** and may have unresolved issues. The recommended PostgreSQL dialect is psycopg2. .. _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:`.create_engine` using the ``client_encoding`` parameter:: engine = create_engine( "postgresql+pg8000://user:pass@host/dbname", client_encoding='utf8') .. _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`` .. versionadded:: 0.9.5 support for AUTOCOMMIT isolation level when using pg8000. .. seealso:: :ref:`postgresql_isolation_level` :ref:`psycopg2_isolation_level` � N� )�_DECIMAL_TYPES)� _FLOAT_TYPES)� _INT_TYPES)� PGCompiler)� PGDialect)�PGExecutionContext)�PGIdentifierPreparer)�UUID)�JSON� )�exc)� processors)�types)�util)� quoted_namec @ s e Zd Zdd� ZdS )� _PGNumericc 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�decimalZDecimalZ _effective_decimal_return_scaler r r ZInvalidRequestErrorZto_float)�self�dialect�coltype� r �X/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/dialects/postgresql/pg8000.py�result_processor` s z_PGNumeric.result_processorN)�__name__� __module__� __qualname__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_processorz s z _PGNumericNoBind.bind_processorN)r r r r r r r r r y s r c s e Zd Z� fdd�Z� ZS )�_PGJSONc s$ |j dkrd S tt| ��||�S d S )N)r � r )�_dbapi_version�superr r )r r r )� __class__r r r s z_PGJSON.result_processor)r r r r � __classcell__r r )r# r r ~ s r c @ s e Zd Zdd� Zdd� ZdS )�_PGUUIDc C s | j sdd� }|S d S )Nc S s | d k rt | �} | S )N)� _python_UUID)�valuer r r �process� s z'_PGUUID.bind_processor.<locals>.process)�as_uuid)r r r( r r r r � s z_PGUUID.bind_processorc C s | j sdd� }|S d S )Nc S s | d k rt | �} | S )N)�str)r' r r r r( � s z)_PGUUID.result_processor.<locals>.process)r) )r r r r( r r r r � s z_PGUUID.result_processorN)r r r r r r r r r r% � s r% c @ s e Zd ZdS )�PGExecutionContext_pg8000N)r r r r r r r r+ � s r+ c @ s e Zd Zdd� Zdd� ZdS )�PGCompiler_pg8000c K s$ | j |jf|�d | j |jf|� S )Nz %% )r( �left�right)r �binary�operator�kwr r r �visit_mod_binary� s z"PGCompiler_pg8000.visit_mod_binaryc C s d|krt �d� |�dd�S )Nz%%z^The SQLAlchemy postgresql dialect now automatically escapes '%' in text() expressions to '%%'.�%)r �warn�replace)r �textr r r �post_process_text� s z#PGCompiler_pg8000.post_process_textN)r r r r2 r7 r r r r r, � s r, c @ s e Zd Zdd� ZdS )�PGIdentifierPreparer_pg8000c C s |� | j| j�}|� dd�S )Nr3 z%%)r5 Z escape_quoteZescape_to_quote)r r' r r r �_escape_identifier� s z.PGIdentifierPreparer_pg8000._escape_identifierN)r r r r9 r r r r r8 � s r8 c s� e Zd ZdZdZdZdZdZeZ e Z e Z dZe�ejejeejeeeejeeei�Zd#dd�Z � fdd �Z ej d d � �Z e d d � �Z!dd� Z"dd� Z#dd� Z$dd� Z%dd� Z&dd� Z'd$dd �Z(d%d d �Z)d d � Z*d!d"� Z+� Z,S )&�PGDialect_pg8000�pg8000T�formatZ use_encodingNc K s t j| f|� || _d S )N)r �__init__�client_encoding)r r> �kwargsr r r r= � s zPGDialect_pg8000.__init__c s | j dk| _tt| ��|� d S )N)r � � )r! � supports_sane_multi_rowcountr" r: � initialize)r � connection)r# r r rC � s zPGDialect_pg8000.initializec 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 )�int)�.0�xr r r � <listcomp>� s z3PGDialect_pg8000._dbapi_version.<locals>.<listcomp>z(\d+)(?:[-\.]?|$))�c rJ rJ )�dbapi�hasattr�tuple�re�findallrE )r r r r r! � s z PGDialect_pg8000._dbapi_versionc C s t d�S )Nr; )� __import__)�clsr r r rK � s zPGDialect_pg8000.dbapic C s8 |j dd�}d|kr$t|d �|d< |�|j� g |fS )N�user)�username�port)Ztranslate_connect_argsrF �update�query)r �url�optsr r r �create_connect_args� s z$PGDialect_pg8000.create_connect_argsc C s dt |�kS )Nzconnection is closed)r* )r �erD �cursorr r r � is_disconnect� s z PGDialect_pg8000.is_disconnectc C s� |� dd�}t|d�r |j}|dkr,d|_nX|| jkrfd|_|�� }|�d| � |�d� |�� n t� d || j d � | j�f ��d S ) N�_� rD Z AUTOCOMMITTFz=SET SESSION CHARACTERISTICS AS TRANSACTION ISOLATION LEVEL %s�COMMITzZInvalid value '%s' for isolation_level. Valid isolation levels for %s are %s or AUTOCOMMITz, ) r5 rL rD Z autocommitZ_isolation_lookupr[ �execute�closer � ArgumentError�name�join)r rD �levelr[ r r r �set_isolation_level� s z$PGDialect_pg8000.set_isolation_levelc C s@ t |d�r|j}|�� }|�d| d � |�d� |�� d S )NrD zSET CLIENT_ENCODING TO '�'r_ )rL rD r[ r` ra )r rD r> r[ r r r �set_client_encoding s z$PGDialect_pg8000.set_client_encodingc C s |j �d|df� d S )Nr � )rD Z tpc_begin)r rD �xidr r r �do_begin_twophase s z"PGDialect_pg8000.do_begin_twophasec C s |j �� d S )N)rD Z tpc_prepare)r rD rj r r r �do_prepare_twophase s z$PGDialect_pg8000.do_prepare_twophaseFc C s |j �d|df� d S )Nr ri )rD Z tpc_rollback)r rD rj � is_prepared�recoverr r r �do_rollback_twophase s z%PGDialect_pg8000.do_rollback_twophasec C s |j �d|df� d S )Nr ri )rD Z tpc_commit)r rD rj rm rn r r r �do_commit_twophase# s z#PGDialect_pg8000.do_commit_twophasec C s dd� |j �� D �S )Nc S s g | ] }|d �qS )r r )rG �rowr r r rI ) s z8PGDialect_pg8000.do_recover_twophase.<locals>.<listcomp>)rD Z tpc_recover)r rD r r r �do_recover_twophase( s z$PGDialect_pg8000.do_recover_twophasec sz g � dd� }� � |� �jd k r6�fdd�}� � |� �jd k rV�fdd�}� � |� t� �dkrr� fdd�}|S d S d S )Nc S s | j tj | j t< d S )N)Zpy_typesr Z text_typer )�connr r r � on_connect. s z/PGDialect_pg8000.on_connect.<locals>.on_connectc s � � | � j� d S )N)rh r> )rs )r r r rt 5 s c s � � | � j� d S )N)rf �isolation_level)rs )r r r rt < s r c s x� D ] }|| � qW d S )Nr )rs �fn)�fnsr r rt C s )�appendr> ru �len)r rt r )rw r r rt + s zPGDialect_pg8000.on_connect)N)TF)TF)-r r r ZdriverZsupports_unicode_statementsZsupports_unicode_bindsZdefault_paramstylerB r+ Zexecution_ctx_clsr, Zstatement_compilerr8 ZpreparerZdescription_encodingr Z update_copyr Zcolspecs�sqltypes�Numericr ZFloatr r r r r% r= rC Zmemoized_propertyr! � classmethodrK rY r\ rf rh rk rl ro rp rr rt r$ r r )r# r r: � s< r: )#�__doc__r rN �baser r r r r r r r �jsonr ri r r r rz r Z sql.elementsr �uuidr& � ImportErrorr{ r r r r% r+ r, r8 r: r r r r r �<module>E s<