Current 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" � /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�fnZm1Zexisting_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# r0 O s z BatchOperationsImpl.alter_columnc O s | j �d||f� d S )N�
add_column)r r1 )r! r, r- r" r" r# r2 R s z BatchOperationsImpl.add_columnc O s | j �d||f� d S )N� drop_column)r r1 )r! r, r- r" r" r# r3 U s z BatchOperationsImpl.drop_columnc C s | j �d|fi f� d S )N�add_constraint)r r1 )r! �constr" r" r# r4 X s z"BatchOperationsImpl.add_constraintc C s | j �d|fi f� d S )N�drop_constraint)r r1 )r! r5 r" r" r# r6 [ s z#BatchOperationsImpl.drop_constraintc O s | j �d||f� d S )N� rename_table)r r1 )r! r, r- r" r" r# r7 ^ s z BatchOperationsImpl.rename_tablec C s | j �d|fi f� d S )N� create_index)r r1 )r! �idxr" r" r# r8 a s z BatchOperationsImpl.create_indexc C s | j �d|fi f� d S )N�
drop_index)r r1 )r! r9 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' r/ r0 r2 r3 r4 r6 r7 r8 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 rE �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 �columnsrE �copy�unique�indexrC �named_constraints�unnamed_constraints�indexes� new_indexesZ constraintsr r1 �kwargsr �
setdefault)r! r
rE Zc_copyr5 r9 �kr" r" r# rI 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)rD rE r" r" r# �
<listcomp>� s zBApplyBatchImpl._transfer_elements_to_new_table.<locals>.<listcomp>)r
Z target_table)rG �AssertionErrorr r; r
r �listrJ �valuesr r rN rO �setr �issubsetrH rK �
isinstancer �_setup_referentZappend_constraint)r! �mr
rG r5 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" )rG rE )rD �col)r! r" r# rW � s zCApplyBatchImpl._gather_indexes_from_both_tables.<locals>.<listcomp>rL ) �extendrP rZ rQ r1 r rC rJ �keysrL rR )r! r9 rM 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 �.���� rU ���c S s g | ]}t |tj��qS r" )r �sqltypes�NULLTYPE)rD �nr" r" r# rW � s z2ApplyBatchImpl._setup_referent.<locals>.<listcomp>c S s g | ]}|� � �d �d �qS )re rh )� _get_colspec�split)rD �elemr" r" r# rW � s r
)�elementsrl rm �len�
sql_schemaZ_get_table_keyZtablesrE Zcontains_columnZ
append_columnr ri rj r ) r! ZmetadataZ
constraint�spec�partsZtnameZreferent_schemarV �trn 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)Zinlinec s s | ]\}}d |kr|V qdS )rB Nr" )rD rT �transferr" r" r# rF � s z)ApplyBatchImpl._create.<locals>.<genexpr>c S s g | ]}d |kr|d �qS )rB r" )rD ru r" r" r# rW � s z*ApplyBatchImpl._create.<locals>.<listcomp>rU )r
)r` Zprep_table_for_batchr; r< rG �_exec�insertZ from_selectrY rH �itemsr rZ r= r7 rC r
rd r8 )r! Zop_implr9 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 )NrC rB F) rJ rH rC ri Z to_instance�typer �nullable�server_default�bool�
autoincrement) r! r Z column_namerz r{ rC Ztype_r} r- ZexistingZexisting_transferr" r" r# r0 � s
zApplyBatchImpl.alter_columnc K s( |j | jjd�| j|j< i | j|j< d S )N)r
)rK r; r
rJ rC rH )r! r �columnr- r" r" r# r2 s zApplyBatchImpl.add_columnc K s | j |j= | j|j= d S )N)rJ rC rH )r! r r~ r- r" r" r# r3 s
zApplyBatchImpl.drop_columnc C sH |j std��t|tj�r8| jj| jkr8| j�| jj� || j |j < d S )NzConstraint must have a name)
rC r r] rq ZPrimaryKeyConstraintr; Z primary_keyrO �removerN )r! r5 r" r" r# r4 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')rC r rN �KeyError)r! r5 r" r" r# r6 # s z ApplyBatchImpl.drop_constraintc C s || j |j< d S )N)rQ rC )r! r9 r" r" r# r8 + s zApplyBatchImpl.create_indexc C s6 y| j |j= W n" tk
r0 td|j ��Y nX d S )NzNo such index: '%s')rP rC r� r )r! r9 r" r" r# r: . s zApplyBatchImpl.drop_indexc O s t d��d S )NZTODO)r )r! r, r- r" r" r# r7 4 s zApplyBatchImpl.rename_table)NFNNN)r> r? r@ r$ rI r` rd r^ r* r0 r2 r3 r4 r6 r8 r: r7 r" r" r" r# r) n s$
$
r) N)Z
sqlalchemyr r r r r r r r ri r
rq Zsqlalchemy.utilr � r
Zutil.sqla_compatr r �objectr r) r" r" r" r# �<module> s $ d
Mini Shell