Edit File: 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