Edit File: batch.cpython-37.opt-1.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�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