Current 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)Z constraints�nameZdrop_constraint)�self�tableZ
constraint� r �/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 Z 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 ZbindZexecuter
�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 r, a s c 3 s" | ]}|�| � | ffV qd S )Nr )r* r )�conn_indexes_by_name�conn_uniques_by_namer r r, 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�idxZexprs�exprr )r- r. 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)ZelementZcompiler�kwr r r �visit_rename_table} s r@ ) r r r � �baser r r r �implr
Z sqlalchemy.dialects.postgresqlr r Z
sqlalchemyr
r r r5 Zsqlalchemy.sql.expressionr r ZloggingZ getLoggerr< r' r r@ r r r r �<module> s
i
Mini Shell