File: //opt/alt/python37/lib/python3.7/site-packages/alembic/operations/__pycache__/batch.cpython-37.pyc
B
�M!Vc+ � @ s� d dl mZmZmZmZmZmZmZ d dl mZ d dl m
Z d dl m
Z
ddlmZ ddlmZmZ G dd � d e�ZG d
d � d e�Zd S )
� )�Table�MetaData�Index�select�Column�ForeignKeyConstraint�cast)�types)�schema)� OrderedDict� )�util)�_columns_for_constraint�_is_type_boundc @ s� e Zd Zdd� Zedd� �Zedd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd � Zd d � Zd S ) �BatchOperationsImplc C sh t jstd��|| _|| _|| _|dkr0td��|| _|| _|| _ t
|�| _ || _ | | _
|
| _g | _d S )Nz.batch mode requires SQLAlchemy 0.8 or greater.)�auto�always�neverz4recreate may be one of 'auto', 'always', or 'never'.)r
Zsqla_08�NotImplementedError�
operations�
table_namer
�
ValueError�recreate� copy_from�
table_args�dict� table_kwargs� reflect_args�reflect_kwargs�naming_convention�batch) �selfr r r
r r r r r r r � r" �I/opt/alt/python37/lib/python3.7/site-packages/alembic/operations/batch.py�__init__ s"
z BatchOperationsImpl.__init__c C s
| j jjS )N)r �impl�dialect)r! r" r" r# r& s zBatchOperationsImpl.dialectc C s | j jS )N)r r% )r! r" r" r# r% $ s zBatchOperationsImpl.implc C s. | j dkr| jj�| �S | j dkr&dS dS d S )Nr r TF)r r r% Zrequires_recreate_in_batch)r! r" r" r# �_should_recreate( s
z$BatchOperationsImpl._should_recreatec C s� | � � }|s<x�| jD ]"\}}}t| jj|�}|||� qW n�| jrPt| jd�}nt� }| jd k rh| j}n,t| j |f| j
�| j d| j� � d�| j
��}t|| j| j�}x(| jD ] \}}}t||�}|||� q�W |�| j� d S )N)r T)r
ZautoloadZ
autoload_with)r' r �getattrr r% r r r r r r r
Zget_bindr �ApplyBatchImplr r �_create) r! Zshould_recreate�opname�arg�kw�fn�m1Zexisting_tableZ
batch_implr" r" r# �flush0 s&
"
zBatchOperationsImpl.flushc O s | j �d||f� d S )N� alter_column)r �append)r! r, r- r" r" r# r1 O s z BatchOperationsImpl.alter_columnc O s | j �d||f� d S )N�
add_column)r r2 )r! r, r- r" r" r# r3 R s z BatchOperationsImpl.add_columnc O s | j �d||f� d S )N� drop_column)r r2 )r! r, r- r" r" r# r4 U s z BatchOperationsImpl.drop_columnc C s | j �d|fi f� d S )N�add_constraint)r r2 )r! �constr" r" r# r5 X s z"BatchOperationsImpl.add_constraintc C s | j �d|fi f� d S )N�drop_constraint)r r2 )r! r6 r" r" r# r7 [ s z#BatchOperationsImpl.drop_constraintc O s | j �d||f� d S )N� rename_table)r r2 )r! r, r- r" r" r# r8 ^ s z BatchOperationsImpl.rename_tablec C s | j �d|fi f� d S )N� create_index)r r2 )r! �idxr" r" r# r9 a s z BatchOperationsImpl.create_indexc C s | j �d|fi f� d S )N�
drop_index)r r2 )r! r: r" r" r# r; d s z BatchOperationsImpl.drop_indexc C s t d��d S )Nz Can't create table in batch mode)r )r! �tabler" r" r# � create_tableg s z BatchOperationsImpl.create_tablec C s t d��d S )Nz Can't drop table in batch mode)r )r! r< r" r" r# �
drop_tablej s z BatchOperationsImpl.drop_tableN)�__name__�
__module__� __qualname__r$ �propertyr& r% r' r0 r1 r3 r4 r5 r7 r8 r9 r; r= r> r" r" r" r# r
s r c @ s~ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd dd�Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd
S ) r) c C s< || _ || _|| _d | _tdd� | j jD ��| _| �� d S )Nc s s | ]}|j d |ifV qdS )�exprN)�name)�.0�cr" r" r# � <genexpr>u s z*ApplyBatchImpl.__init__.<locals>.<genexpr>)r< r r � new_tabler rF �column_transfers�_grab_table_elements)r! r< r r r" r" r# r$ o s zApplyBatchImpl.__init__c C s� | j j}t� | _x4| j jD ](}|j|d�}d |_|_|| j|j< qW i | _ g | _
i | _ i | _ x:| j j
D ].}t|�rvqh|jr�|| j |j< qh| j
�|� qhW x | j j D ]}|| j |j< q�W x&| j jD ]}| j�|| j j| � q�W d S )N)r
F)r< r
r �columnsrF �copy�unique�indexrD �named_constraints�unnamed_constraints�indexes� new_indexes� constraintsr r2 �kwargsr �
setdefault)r! r
rF Zc_copyr6 r: �kr" r" r# rJ y s( z#ApplyBatchImpl._grab_table_elementsc C s� | j d kstd��t� }| jj}td|ft| j�� �t| j � �d|i| j
�� | _ }xnt| j �� �| j D ]V}t
dd� t|�D ��}|�| j�s�ql|j||d�}t|t�r�| �||� |�|� qlW d S )Nz Can only create new table once�_alembic_batch_tempr
c S s g | ]
}|j �qS r" )�key)rE rF r" r" r# �
<listcomp>� s zBApplyBatchImpl._transfer_elements_to_new_table.<locals>.<listcomp>)r
Z target_table)rH �AssertionErrorr r< r
r �listrK �valuesr r rO rP �setr �issubsetrI rL �
isinstancer �_setup_referentZappend_constraint)r! �mr
rH r6 Z
const_columnsZ
const_copyr" r" r# � _transfer_elements_to_new_table� s$
z.ApplyBatchImpl._transfer_elements_to_new_tablec sd g }|� � j�� � xJ� j�� D ]<}|�t|jf� fdd�|j�� D ��d|j i|j
��� q W |S )Nc s g | ]}� j j| �qS r" )rH rF )rE �col)r! r" r# rY � s zCApplyBatchImpl._gather_indexes_from_both_tables.<locals>.<listcomp>rM ) �extendrQ r\ rR r2 r rD rK �keysrM rT )r! r: rN r" )r! r# � _gather_indexes_from_both_tables� s z/ApplyBatchImpl._gather_indexes_from_both_tablesc C s� |j d �� }|�d�}|d }t|�dkr6|d }nd }|dkr�t�||�}||jkr�|j| }xl|j D ]4} | �� �d�d }
|j�|
�sj|� t
|
t j �� qjW n,t
||fdd� d d� |j D �D ��d
|i� d S ) Nr �.���� rW ���c S s g | ]}t |tj��qS r" )r �sqltypes�NULLTYPE)rE �nr" r" r# rY � s z2ApplyBatchImpl._setup_referent.<locals>.<listcomp>c S s g | ]}|� � �d �d �qS )rg rj )� _get_colspec�split)rE �elemr" r" r# rY � s r
)�elementsrn ro �len�
sql_schemaZ_get_table_keyZtablesrF Zcontains_columnZ
append_columnr rk rl r ) r! �metadataZ
constraint�spec�partsZtnameZreferent_schemarX �trp Zcolnamer" r" r# r` � s*
z ApplyBatchImpl._setup_referentc C s� | � � |�| j� |�| j� yT|�| jjdd��tdd� | j �
� D ��t dd� | j � � D ���� |�
| j� W n |�
| j� � Y nRX |jd| jj| jjd� | jj| j_z x| �� D ]}|�|� q�W W d d| j_X d S ) NT)�inlinec s s | ]\}}d |kr|V qdS )rC Nr" )rE rV �transferr" r" r# rG � s z)ApplyBatchImpl._create.<locals>.<genexpr>c S s g | ]}d |kr|d �qS )rC r" )rE ry r" r" r# rY � s z*ApplyBatchImpl._create.<locals>.<listcomp>rW )r
)rb Zprep_table_for_batchr< r= rH �_exec�insertZ from_selectr[ rI �itemsr r\ r> r8 rD r
rf r9 )r! Zop_implr: r" r" r# r* � s0 zApplyBatchImpl._createNFc K s� | j | } | j| }
|d k r2||kr2|| _||
d< |d k r\t�|�}|| _t|
d |�|
d< |d k rj|| _|dk rx|| _|d k r�t |�| _
d S )NrD rC F) rK rI rD rk Z to_instance�typer �nullable�server_default�bool�
autoincrement) r! r Z column_namer~ r rD �type_r� r- �existingZexisting_transferr" r" r# r1 � s
zApplyBatchImpl.alter_columnc K s( |j | jjd�| j|j< i | j|j< d S )N)r
)rL r< r
rK rD rI )r! r �columnr- r" r" r# r3 s zApplyBatchImpl.add_columnc K s | j |j= | j|j= d S )N)rK rD rI )r! r r� r- r" r" r# r4 s
zApplyBatchImpl.drop_columnc C sH |j std��t|tj�r8| jj| jkr8| j�| jj� || j |j < d S )NzConstraint must have a name)
rD r r_ rs ZPrimaryKeyConstraintr< Z primary_keyrP �removerO )r! r6 r" r" r# r5 s z ApplyBatchImpl.add_constraintc C sD |j std��y| j|j = W n" tk
r> td|j ��Y nX d S )NzConstraint must have a namezNo such constraint: '%s')rD r rO �KeyError)r! r6 r" r" r# r7 # s z ApplyBatchImpl.drop_constraintc C s || j |j< d S )N)rR rD )r! r: r" r" r# r9 + s zApplyBatchImpl.create_indexc C s6 y| j |j= W n" tk
r0 td|j ��Y nX d S )NzNo such index: '%s')rQ rD r� r )r! r: r" r" r# r; . s zApplyBatchImpl.drop_indexc O s t d��d S )NZTODO)r )r! r, r- r" r" r# r8 4 s zApplyBatchImpl.rename_table)NFNNN)r? r@ rA r$ rJ rb rf r` r* r1 r3 r4 r5 r7 r9 r; r8 r" r" r" r# r) n s$
$
r) N)Z
sqlalchemyr r r r r r r r rk r
rs Zsqlalchemy.utilr � r
Zutil.sqla_compatr r �objectr r) r" r" r" r# �<module> s $ d