Edit File: pygresql.cpython-37.pyc
B ��4]� � @ s\ d Z ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd lm Z dd l m Z dd l mZ dd l mZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ ddlmZ G dd� de�ZG dd� de �ZG dd� de�ZG dd� de�Z G d d � 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+pygresql :name: pygresql :dbapi: pgdb :connectstring: postgresql+pygresql://user:password@host:port/dbname[?key=value&key=value...] :url: http://www.pygresql.org/ .. note:: The pygresql dialect is **not tested as part of SQLAlchemy's continuous integration** and may have unresolved issues. The recommended PostgreSQL dialect is psycopg2. � N� )�_DECIMAL_TYPES)� _FLOAT_TYPES)� _INT_TYPES)� PGCompiler)� PGDialect)�PGIdentifierPreparer)�UUID)�HSTORE)�JSON)�JSONB� )�exc)� processors)�util)�Null)�Numericc @ s e Zd Zdd� Zdd� ZdS )� _PGNumericc C s d S )N� )�self�dialectr r �Z/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/dialects/postgresql/pygresql.py�bind_processor, s z_PGNumeric.bind_processorc C s� t |t�s|j}| jrR|tkr.t�tj| j �S |t ks>|t krBd S t � d| ��n0|tkr^d S |t ksn|t krttjS t � d| ��d S )NzUnknown PG numeric type: %d)� isinstance�intZoidZ asdecimalr r Z to_decimal_processor_factory�decimalZDecimalZ _effective_decimal_return_scaler r r ZInvalidRequestErrorZto_float)r r �coltyper r r �result_processor/ s z_PGNumeric.result_processorN)�__name__� __module__� __qualname__r r r r r r r + s r c s( e Zd Z� fdd�Z� fdd�Z� ZS )� _PGHStorec s. |j stt| ��|�S |jj� � fdd�}|S )Nc s t | t�r� | �S | S )N)r �dict)�value)�hstorer r �processP s z)_PGHStore.bind_processor.<locals>.process)�has_native_hstore�superr! r �dbapiZHstore)r r r% )� __class__)r$ r r K s z_PGHStore.bind_processorc s |j stt| ��||�S d S )N)r& r' r! r )r r r )r) r r r W s z_PGHStore.result_processor)r r r r r � __classcell__r r )r) r r! J s r! c s( e Zd Z� fdd�Z� fdd�Z� ZS )�_PGJSONc s0 |j stt���|�S |jj� � �fdd�}|S )Nc sN | �j krd } n t| t�s(| d kr,�jr,d S | d ksBt| ttf�rJ� | �S | S )N)�NULLr r � none_as_nullr" �list)r# )�jsonr r r r% b s z'_PGJSON.bind_processor.<locals>.process)�has_native_jsonr' r+ r r( �Json)r r r% )r) )r/ r r r ] s z_PGJSON.bind_processorc s |j stt| ��||�S d S )N)r0 r' r+ r )r r r )r) r r r o s z_PGJSON.result_processor)r r r r r r* r r )r) r r+ \ s r+ c s( e Zd Z� fdd�Z� fdd�Z� ZS )�_PGJSONBc s0 |j stt���|�S |jj� � �fdd�}|S )Nc sN | �j krd } n t| t�s(| d kr,�jr,d S | d ksBt| ttf�rJ� | �S | S )N)r, r r r- r" r. )r# )r/ r r r r% z s z(_PGJSONB.bind_processor.<locals>.process)r0 r' r2 r r( r1 )r r r% )r) )r/ r r r u s z_PGJSONB.bind_processorc s |j stt| ��||�S d S )N)r0 r' r2 r )r r r )r) r r r � s z_PGJSONB.result_processor)r r r r r r* r r )r) r r2 t s r2 c s( e Zd Z� fdd�Z� fdd�Z� ZS )�_PGUUIDc s. |j stt| ��|�S |jj� � fdd�}|S )Nc sP | d kr d S t | ttf�r8t| �dkr0� | d�S � | �S t | t�rL� | d�S | S )N� )�bytes)r )r �strr5 �lenr )r# )�uuidr r r% � s z'_PGUUID.bind_processor.<locals>.process)�has_native_uuidr' r3 r r( ZUuid)r r r% )r) )r8 r r � s z_PGUUID.bind_processorc s. |j stt| ��||�S | js*dd� }|S d S )Nc S s | d k rt | �S d S )N)r6 )r# r r r r% � s z)_PGUUID.result_processor.<locals>.process)r9 r' r3 r Zas_uuid)r r r r% )r) r r r � s z_PGUUID.result_processor)r r r r r r* r r )r) r r3 � s r3 c @ s e Zd Zdd� Zdd� ZdS )� _PGCompilerc K s$ | j |jf|�d | j |jf|� S )Nz %% )r% �left�right)r �binary�operator�kwr r r �visit_mod_binary� s z _PGCompiler.visit_mod_binaryc C s |� dd�S )N�%z%%)�replace)r �textr r r �post_process_text� s z _PGCompiler.post_process_textN)r r r r@ rD r r r r r: � s r: c @ s e Zd Zdd� ZdS )�_PGIdentifierPreparerc C s |� | j| j�}|� dd�S )NrA z%%)rB Z escape_quoteZescape_to_quote)r r# r r r �_escape_identifier� s z(_PGIdentifierPreparer._escape_identifierN)r r r rF r r r r rE � s rE c sj e Zd ZdZeZeZedd� �Z e � e j eeeeeeeeeeeei�Z � fdd�Zdd� Zdd � Z� Z S ) �PGDialect_pygresqlZpygresqlc C s dd l }|S )Nr )�pgdb)�clsrH r r r r( � s zPGDialect_pygresql.dbapic s� t t| �jf |� y4| jj}t�d|�}t|�d��t|�d��f}W n t t t fk rd d}Y nX || _ |dk r�d } }}|dkr�t �d� nd| _d| _d } }}|| _|| _|| _d S ) Nz (\d+)\.(\d+)r � )r r )� r FzAPyGreSQL is only fully supported by SQLAlchemy since version 5.0.T)r' rG �__init__r( �version�re�matchr �group�AttributeError� ValueError� TypeErrorZ dbapi_versionr �warnZsupports_unicode_statementsZsupports_unicode_bindsr& r0 r9 )r �kwargsrM �mr& r0 r9 )r) r r rL � s& zPGDialect_pygresql.__init__c C sR |j dd�}d|kr>d|�dd��dd�d |�d�f |d< |�|j� g |fS ) N�user)�username�portz%s:%s�host� �:r r )Ztranslate_connect_args�get�rsplit�pop�update�query)r �url�optsr r r �create_connect_args� s z&PGDialect_pygresql.create_connect_argsc C sd t || jj�r`|sdS y |j}W n tk r4 Y n X |s>dS y|jS tk r^ |jd kS X dS )NF)r r( �Error� connectionrQ �closedZ_cnx)r �erf Zcursorr r r � is_disconnect� s z PGDialect_pygresql.is_disconnect) r r r Zdriverr: Zstatement_compilerrE Zpreparer� classmethodr( r Z update_copyr Zcolspecsr r r r! r1 r+ r r r2 r r3 rL rd ri r* r r )r) r rG � s rG )"�__doc__r rN �baser r r r r r r r$ r r/ r r r[ r r r Z sql.elementsr �typesr1 r r r! r+ r2 r3 r: rE rG r r r r r �<module> s6 M