관리-도구
편집 파일: batch.cpython-38.pyc
U 0?�f?i � @ s d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d l m Z d d l m Z d d l m Z d d l mZ d d l mZ d dl mZ d dl mZ d dl mZ d dl mZ d dlmZ d dlmZ d dlmZ ddlm Z ddl m Z ddl m Z ddl m Z ddl m!Z! ddl m"Z" dd l m#Z# dd l m$Z$ dd l m%Z% dd l m&Z& dd l m'Z' dd!l m(Z( dd"l m)Z) dd#l m*Z* e�r�d d$lm+Z+ d d%l,m-Z- d d&l.m/Z/ d d'l.m0Z0 d d(l1m2Z2 d d)lm3Z3 d d*l4m5Z5 dd+l6m7Z7 G d,d-� d-�Z8G d.d/� d/�Z9d0S )1� )� annotations)�Any)�Dict)�List)�Optional)�Tuple)� TYPE_CHECKING)�Union)�CheckConstraint)�Column)�ForeignKeyConstraint)�Index)�MetaData)�PrimaryKeyConstraint��schema)�Table)�types)�SchemaEventTarget)� OrderedDict)� topological� )�exc)�_columns_for_constraint)�_copy)�_copy_expression)�_ensure_scope_for_ddl)�_fk_is_self_referential)� _idx_table_bound_expressions)�_insert_inline)�_is_type_bound)� _remove_column_from_collection)�_resolve_for_variant)�_select)�constraint_name_defined)�constraint_name_string)�Literal)�Dialect)� ColumnClause)� quoted_name)�Function)� Constraint)� TypeEngine)� DefaultImplc @ s� e Zd Zdd� Zedd�dd��Zedd�dd ��Zd d�d d �Zd d�dd�Zd d�dd�Z d d�dd�Z d d�dd�Z dd d�dd�Z dd d�dd�Z d d � Zd d d d �d!d"�Zd d d d �d#d$�Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/S )0�BatchOperationsImplc C s� || _ || _|| _|dkr"td��|| _|| _|| _t|�| _|| _ t| �| _ | j � dt | j � dd��� | j d �d|jjf� | | _| | _g | _d S )N)�auto�always�neverz4recreate may be one of 'auto', 'always', or 'never'.Z listeners� Zcolumn_reflect)� operations� table_namer � ValueError�recreate� copy_from� table_args�dict� table_kwargs� reflect_args�reflect_kwargs� setdefault�list�get�append�implZautogen_column_reflect�naming_convention�partial_reordering�batch) �selfr3 r4 r r6 r7 r8 r: r; r< rB rC r2 r2 �F/opt/hc_python/lib/python3.8/site-packages/alembic/operations/batch.py�__init__8 s. � � �z BatchOperationsImpl.__init__r' ��returnc C s | j jjS �N)r3 rA �dialect�rE r2 r2 rF rK ] s zBatchOperationsImpl.dialectr- c C s | j jS rJ )r3 rA rL r2 r2 rF rA a s zBatchOperationsImpl.impl�boolc C s. | j dkr| jj�| �S | j dkr&dS dS d S )Nr/ r0 TF)r6 r3 rA Zrequires_recreate_in_batchrL r2 r2 rF �_should_recreatee s z$BatchOperationsImpl._should_recreate�Nonec C s. | � � }t| jj��� |sH| jD ]"\}}}t| jj|�}|||� q"n�| jr\t| jd�}nt� }| j d k rx| j }d}nZ| jj j r�t � d| jj jj� d| j� d���t| j|f| j�| j| j�� d�| j��}d}t| j|| j| j|| jd�} | jD ] \}}}t| |�}|||� q�| �| j� W 5 Q R X d S ) N)rB FzEThis operation cannot proceed in --sql mode; batch mode with dialect zF requires a live database connection with which to reflect the table "z�". To generate a batch SQL migration script using table "move and copy", a complete Table object should be passed to the "copy_from" argument of the batch_alter_table() method so that table reflection can be skipped.)r Z autoload_withT)rC )rN r rA � connectionrD �getattrr3 rB r r7 Zmigration_contextZas_sqlr � CommandErrorrK �namer4 r r; r Zget_bindr< �ApplyBatchImplr8 r: rC �_create) rE Zshould_recreate�opname�arg�kw�fn�m1Zexisting_table� reflectedZ batch_implr2 r2 rF �flushm sP ������ zBatchOperationsImpl.flushc O s | j �d||f� d S )N� alter_column�rD r@ �rE rW rX r2 r2 rF r] � s z BatchOperationsImpl.alter_columnc O s8 d|ksd|kr"| � � s"t�d��| j�d||f� d S )N� insert_before� insert_afterz^Can't specify insert_before or insert_after when using ALTER; please specify recreate='always'� add_column)rN r rR rD r@ r_ r2 r2 rF rb � s ����z BatchOperationsImpl.add_columnc O s | j �d||f� d S )N� drop_columnr^ r_ r2 r2 rF rc � s z BatchOperationsImpl.drop_columnr+ ��constrI c C s | j �d|fi f� d S )N�add_constraintr^ �rE re r2 r2 rF rf � s z"BatchOperationsImpl.add_constraintc C s | j �d|fi f� d S )N�drop_constraintr^ rg r2 r2 rF rh � s z#BatchOperationsImpl.drop_constraintc O s | j �d||f� d S )N� rename_tabler^ r_ r2 r2 rF ri � s z BatchOperationsImpl.rename_tabler r )�idxrX rI c K s | j �d|f|f� d S )N� create_indexr^ �rE rj rX r2 r2 rF rk � s z BatchOperationsImpl.create_indexc K s | j �d|f|f� d S )N� drop_indexr^ rl r2 r2 rF rm � s z BatchOperationsImpl.drop_indexc C s | j �d|fi f� d S )N�create_table_commentr^ �rE �tabler2 r2 rF rn � s z(BatchOperationsImpl.create_table_commentc C s | j �d|fi f� d S )N�drop_table_commentr^ ro r2 r2 rF rq � s z&BatchOperationsImpl.drop_table_commentc C s t d��d S )Nz Can't create table in batch mode��NotImplementedErrorro r2 r2 rF � create_table� s z BatchOperationsImpl.create_tablec C s t d��d S )Nz Can't drop table in batch moderr ro r2 r2 rF � drop_table� s z BatchOperationsImpl.drop_tablec C s | j �d|fi f� d S )N�create_column_commentr^ �rE �columnr2 r2 rF rv � s z)BatchOperationsImpl.create_column_commentN)�__name__� __module__� __qualname__rG �propertyrK rA rN r\ r] rb rc rf rh ri rk rm rn rq rt ru rv r2 r2 r2 rF r. 7 s( %9 r. c @ sT e Zd ZdLdddddddd�d d �Zed d d �dd��Zdd�dd�Zdd�dd�Zdd�dd�Zdd�dd�Z dddd �d d �Z ddd �d d!�Z dMd d d$d%d&d'd(d)dd*� d+d,�Z d d&d&dd-�d.d/�Z dNd d0d&d&dd1�d2d3�Zd d4dd5�d6d7�Zd8d9� Zd:d;� Zd<d=� Zd>dd?�d@dA�Zd>dd?�dBdC�ZdDddE�dFdG�ZdDddE�dHdI�ZdJdK� Zd"S )OrT r2 r- r �tuplezDict[str, Any]rM rO )rA rp r8 r: r[ rC rI c C sn || _ || _|| _|| _| �|j�| _d | _|| _d| _ t dd� | jj D ��| _ t | j �| _|| _| �� d S )Nr2 c s s | ]}|j d |ifV qdS ��exprN)rS ��.0�cr2 r2 rF � <genexpr>� s z*ApplyBatchImpl.__init__.<locals>.<genexpr>)rA rp r8 r: �_calc_temp_namerS �temp_table_name� new_tablerC �add_col_orderingr r� �column_transfersr> �existing_orderingr[ �_grab_table_elements)rE rA rp r8 r: r[ rC r2 r2 rF rG � s �� zApplyBatchImpl.__init__zUnion[quoted_name, str]�str)� tablenamerI c C s d| dd� S )Nz_alembic_tmp_%sr �2 r2 )�clsr� r2 r2 rF r� � s z ApplyBatchImpl._calc_temp_namerH c C sP | j j}t� | _| j jD ]D}t||d�}d |_|_t|j t �rP|j |j k sPt �|| j|j < qi | _ g | _i | _i | _i | _| j jD ]L}t|�r�q�q�| jr�t|t�r�|j s�q�t|j �r�|| j |j < q�| j�|� q�| j�s | j jD ]&}|jD ]}|j r�||f| j|j < q�q�| j jD ]}|| j|j < �q| j jD ] }| j�|| j j| � �q.d S )Nr F)rp r r �columnsr� r �unique�index� isinstance�typer �AssertionErrorrS �named_constraints�unnamed_constraints�col_named_constraints�indexes� new_indexes� constraintsr r[ r r% r@ �kwargsr: r= )rE r r� Zc_copyre �colrj �kr2 r2 rF r� � sF ��� z#ApplyBatchImpl._grab_table_elementsc s� t � }t� j�}� jrT� jD ]4}t|�D ]&\}}|dkr(|�||d |f� q(q n2t� j�D ]&\}}|dkr^|�||d |f� q^|�� j� dd� |D �}tt j ||dd��}t � fdd�|D ��� _t � fd d�|D ��� _ d S ) Nr � c S s g | ]}|d |d kr|�qS )r r� r2 )r� �pr2 r2 rF � <listcomp>9 s zNApplyBatchImpl._adjust_self_columns_for_partial_reordering.<locals>.<listcomp>T)Zdeterministic_orderc 3 s | ]}|� j | fV qd S rJ )r� �r� r� rL r2 rF r� > s zMApplyBatchImpl._adjust_self_columns_for_partial_reordering.<locals>.<genexpr>c 3 s | ]}|� j | fV qd S rJ )r� r� rL r2 rF r� ? s ) �setr> r� rC � enumerate�addr� �updater� r �sortr r� )rE �pairsZ col_by_idx�tuple_r� �elemZ pairs_listZsorted_r2 rL rF �+_adjust_self_columns_for_partial_reordering% s& � �z:ApplyBatchImpl._adjust_self_columns_for_partial_reorderingc C s | j d kstd��t� }| jj}| js,| jr4| �� t| j |ft | j � � �t | j � �d|i| j�� | _ }t | j� � �| j D ]�}dd� t|�D �}|�| j�s�q�t|t�r�t|�r�t||| jd�}q�t||d�}nt|||d�}t|t�r�| �||� |�|� q�d S )Nz Can only create new table oncer c S s h | ] }|j �qS r2 )�keyr� r2 r2 rF � <setcomp>W s zAApplyBatchImpl._transfer_elements_to_new_table.<locals>.<setcomp>)r Z target_tabler )r� r� r rp r rC r� r� r r� r> r� �valuesr8 r: r� r� r �issubsetr� r� r r r �_setup_referentZappend_constraint)rE �mr r� re Z const_columnsZ const_copyr2 r2 rF � _transfer_elements_to_new_tableC sH ���� � � � z.ApplyBatchImpl._transfer_elements_to_new_tablez List[Index]c s� � j d k st�g }� j�� D ]J}|jr(q t|jf� fdd�t|�D ��|j� j d�|j ��}|� |� q � j �� D ]<}|� t|jf� fdd�|j � � D ��d|ji|j ��� qr|S )Nc s g | ]}t |� j��qS r2 )r r� )r� r rL r2 rF r� � s �zCApplyBatchImpl._gather_indexes_from_both_tables.<locals>.<listcomp>)r� �_tablec s g | ]}� j j| �qS r2 )r� r� )r� r� rL r2 rF r� � s r� )r� r� r� r� Z _column_flagr rS r r� r� r@ r� r� �keys)rE rj Z idx_existingZidx_copyr� r2 rL rF � _gather_indexes_from_both_tablesu s> � ���� �����z/ApplyBatchImpl._gather_indexes_from_both_tablesr r )�metadata� constraintrI c s� |j d �� }|�d�}|d }t|�dkr6|d }nd }|| jkr�t�||�}dd�dd�� ||jkr�|j| }|j D ]2} � | ��d�d } | |jkrx|� t | t j �� qxn0t ||fd d � � fd d �|j D �D ��d |i� d S )Nr �.���� r �r� c S s | � � S rJ )� _get_colspecr� r2 r2 rF �colspec� s z/ApplyBatchImpl._setup_referent.<locals>.colspec���c S s g | ]}t |tj��qS r2 )r �sqltypes�NULLTYPE)r� �nr2 r2 rF r� � s �z2ApplyBatchImpl._setup_referent.<locals>.<listcomp>c s g | ]}� |�� d �d �qS )r� r� )�split)r� r� �r� r2 rF r� � s �r )�elementsr� r� �lenr� � sql_schemaZ_get_table_keyZtablesr� � append_columnr r� r� r ) rE r� r� �spec�partsZtnameZreferent_schemar� �tr� �colnamer2 r� rF r� � s6 � ��� �z ApplyBatchImpl._setup_referent)�op_implrI c C s� | � � |�| | j� | jd k s$t�|�| j� zP|�t| j��t dd� | j � � D ��t dd� | j � � D �� �� |�| j� W n |�| j� � Y nRX |j| j| jj| jjd� | jj| j_z | �� D ]}|�|� q�W 5 | j| j_X d S )Nc s s | ]\}}d |kr|V qdS r~ r2 )r� r� �transferr2 r2 rF r� � s �z)ApplyBatchImpl._create.<locals>.<genexpr>c S s g | ]}d |kr|d �qS )r r2 )r� r� r2 r2 rF r� � s �z*ApplyBatchImpl._create.<locals>.<listcomp>r )r� Zprep_table_for_batchrp r� r� rt �_execr Z from_selectr> r� �itemsr# r� ru ri r� rS r r� rk )rE r� rj r2 r2 rF rU � s> ����� � zApplyBatchImpl._createNFzOptional[bool]z)Optional[Union[Function[Any], str, bool]]z Optional[str]zOptional[TypeEngine]z&Optional[Union[bool, Literal['auto']]]zUnion[str, Literal[False]]) r4 � column_name�nullable�server_defaultrS �type_� autoincrement�commentrI c K s | j | } | j| } |d k rt||krt|| _|| d< | �dd �} | rtt| d | jj�} t| t�rt| jrt| j � | jd � |d k r�t � |�}t| j t�r�d | j _| j _| j�| | |� || _ |d k r�|| _|dk r�|d kr�d | _nt�|��| � |d k �rt|�| _|dk �r|| _d S )NrS � existing_typeF)r� r� rS r? r" rA rK r� r r� �popr� Z to_instancer� Z_create_eventsZcreate_constraintZcast_for_batch_migrater� r� r� Z DefaultClauseZ _set_parentrM r� r� )rE r4 r� r� r� rS r� r� r� rX �existingZexisting_transferr� Zresolved_existing_typer2 r2 rF r] � sZ ���� � ��� zApplyBatchImpl.alter_column)r� r` ra rI c C s� | j }dd� t|�D �}| js�|r^|s^||krP|| d }|t|�k r^|| }nt| j�| }|r�|s�||kr�|| d }|dkr�|| }ndd� | jD �| }|r�| j||ff7 _|r�| j||ff7 _| js�|s�|s�|r�| j|d |ff7 _d S )Nc S s i | ]\}}||�qS r2 r2 )r� �irS r2 r2 rF � <dictcomp>4 s zEApplyBatchImpl._setup_dependencies_for_add_column.<locals>.<dictcomp>r� r c S s i | ]\}}||�qS r2 r2 )r� �a�br2 r2 rF r� L s r� )r� r� rC r� r9 r� )rE r� r` ra Z index_colsZ col_indexesrj r2 r2 rF �"_setup_dependencies_for_add_column- sH � ������z1ApplyBatchImpl._setup_dependencies_for_add_columnz Column[Any])r4 rx r` ra rI c K s8 | � |j||� t|| jjd�| j|j< i | j|j< d S )Nr )r� rS r rp r r� r� )rE r4 rx r` ra rX r2 r2 rF rb ] s �zApplyBatchImpl.add_columnz%Union[ColumnClause[Any], Column[Any]])r4 rx rI c K sz |j | jjjkr t| jjj|� | j|j = | j|j = | j�|j � d|krvt|d t �rv|d j rv| j � |d j d � d S )Nr� ) rS rp � primary_keyr� r! r� r� �remover� r r� r� )rE r4 rx rX r2 r2 rF rc m s$ � � �� �zApplyBatchImpl.drop_columnc C s dS )a^ the batch table creation function will issue create_column_comment on the real "impl" as part of the create table process. That is, the Column object will have the comment on it already, so when it is received by add_column() it will be a normal part of the CREATE TABLE and doesn't need an extra step here. Nr2 rw r2 r2 rF rv � s z$ApplyBatchImpl.create_column_commentc C s dS )z�the batch table creation function will issue create_table_comment on the real "impl" as part of the create table process. Nr2 ro r2 r2 rF rn � s z#ApplyBatchImpl.create_table_commentc C s dS )z�the batch table creation function will issue drop_table_comment on the real "impl" as part of the create table process. Nr2 ro r2 r2 rF rq � s z!ApplyBatchImpl.drop_table_commentr+ rd c C sd t |j�std��t|tj�r<| jj| jkr<| j� | jj� t |j�rT|| j |j< n | j� |� d S )N�Constraint must have a name) r$ rS r5 r� r� r rp r� r� r� r% r� r@ rg r2 r2 rF rf � s z ApplyBatchImpl.add_constraintc C s� |j std��z�|j | jkrh| j�|j �\}}t| j|j j�D ]$}|j |j kr@| j|j j�|� q@n0t|j �r�| j �|j �}n|| j kr�| j �|� W n0 t k r� t |�r�Y d S td|j ��Y n&X t |t�r�|jD ]}d| j|j _q�d S )Nr� zNo such constraint: '%s'F)rS r5 r� r� r> r� r� r� r% r� r� �KeyErrorr r� r r� )rE re r� Z col_constr2 r2 rF rh � s& z ApplyBatchImpl.drop_constraintr )rj rI c C s || j |j< d S rJ )r� rS �rE rj r2 r2 rF rk � s zApplyBatchImpl.create_indexc C s6 z| j |j= W n" tk r0 td|j ��Y nX d S )NzNo such index: '%s')r� rS r� r5 r� r2 r2 rF rm � s zApplyBatchImpl.drop_indexc O s t d��d S )N�TODOrr r_ r2 r2 rF ri � s zApplyBatchImpl.rename_table)r2 )NFNNNF)NN)ry rz r{ rG � classmethodr� r� r� r� r� r� rU r] r� rb rc rv rn rq rf rh rk rm ri r2 r2 r2 rF rT � s< � . 2!%+ � K4 � rT N):� __future__r �typingr r r r r r r Z sqlalchemyr r r r r r r r� r r r� Zsqlalchemy.sql.schemar Zsqlalchemy.utilr r �utilr Zutil.sqla_compatr r r r r r r r r! r" r# r$ r% r& Zsqlalchemy.enginer' Zsqlalchemy.sql.elementsr( r) Zsqlalchemy.sql.functionsr* r+ Zsqlalchemy.sql.type_apir, Zddl.implr- r. rT r2 r2 r2 rF �<module> sZ