File: //opt/alt/python37/lib/python3.7/site-packages/alembic/ddl/__pycache__/postgresql.cpython-37.pyc
B
�M!V� � @ s� d dl Z ddlmZ ddlmZ ddlmZmZmZmZ ddl m
Z
d dl m Z m
Z
d d lmZmZmZ ejr|d d
lmZ n d d lmZ d dlZe�e�ZG d d
� d
e
�Zeed�dd� �ZdS )� N� )�compat)�util� )�compiles� alter_table�format_table_name� RenameTable)� DefaultImpl)�INTEGER�BIGINT)�text�Numeric�Column)�UnaryExpression)�_UnaryExpressionc @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) �PostgresqlImpl�
postgresqlTc C s( x"|j D ]}|jd k r| �|� qW d S )N)� constraints�nameZdrop_constraint)�self�tableZ
constraint� r �G/opt/alt/python37/lib/python3.7/site-packages/alembic/ddl/postgresql.py�prep_table_for_batch s
z#PostgresqlImpl.prep_table_for_batchc C s� |j r||jjkrdS |}d ||fkr.||kS |jd k rnt|jjtj�rnt� d|�snt|j
t �snt� dd|�}| j
�d||f � S )NFz^'.+'$z ^u?'?|'?$�'zSELECT %s = %s)Z primary_keyr Z_autoincrement_columnZserver_default�
isinstance�argr � string_types�re�match�typer �subZ
connectionZscalar)r Zinspector_columnZmetadata_columnZrendered_metadata_defaultZrendered_inspector_defaultZconn_col_defaultr r r �compare_server_default s"
z%PostgresqlImpl.compare_server_defaultc C s� |� d�r�t|d ttf�r�t�d|d �}|r�|jjtd�|� d�d��
� }|r�|\}}||d kr�t � d||j
|f � |d= d S ) N�defaultr! z nextval\('(.+?)'::regclass\)a+ select c.relname, a.attname from pg_class as c join pg_depend d on d.objid=c.oid and d.classid='pg_class'::regclass and d.refclassid='pg_class'::regclass join pg_class t on t.oid=d.refobjid join pg_attribute a on a.attrelid=t.oid and a.attnum=d.refobjsubid where c.relkind='S' and c.relname=:seqnamer )�seqnamer z^Detected sequence named '%s' as owned by integer column '%s(%s)', assuming SERIAL and omitting)�getr r r r r �bind�executer
�group�first�log�infor )r Z inspectorr Z column_infoZ seq_matchr, r% Zcolnamer r r �autogen_column_reflect< s
z%PostgresqlImpl.autogen_column_reflectc s� t dd� |D ���t dd� |D ��� t � �fdd�t���� �D ��}x"|�� D ]\}\}}|�|� qPW xft|�D ]Z} | j� kr�qttjr�| j }
n| j
}
x4|
D ],} t | t t
f�s�t�d| j � |�| � q�W qtW d S )Nc s s | ]}|j |fV qd S )N)r )�.0�cr r r � <genexpr>_ s zAPostgresqlImpl.correct_for_autogen_constraints.<locals>.<genexpr>c s s | ]}|j |fV qd S )N)r )r. r/ r r r r0 a s c 3 s" | ]}|�| � | ffV qd S )Nr )r. r )�conn_indexes_by_name�conn_uniques_by_namer r r0 f s zQautogenerate skipping functional index %s; not supported by SQLAlchemy reflection)�dict�set� intersection�items�remove�listr r �sqla_08Z expressions�columnsr r r r �warn�discard) r Zconn_unique_constraintsZ conn_indexesZmetadata_unique_constraintsZmetadata_indexesZdoubled_constraintsr ZuqZix�idx�exprs�exprr )r1 r2 r � correct_for_autogen_constraintsZ s,
z.PostgresqlImpl.correct_for_autogen_constraintsN) �__name__�
__module__� __qualname__Z __dialect__Ztransactional_ddlr r# r- r@ r r r r r s r r c K s" dt || j| j�t|| jd �f S )Nz%s RENAME TO %s)r Z
table_nameZschemar Znew_table_name)�element�compiler�kwr r r �visit_rename_table} s rG ) r r r � �baser r r r �implr
Z sqlalchemy.dialects.postgresqlr r Z
sqlalchemyr
r r r9 Zsqlalchemy.sql.expressionr r �logging� getLoggerrA r+ r rG r r r r �<module> s
i