관리-도구
편집 파일: compare.cpython-38.pyc
U 0?�f�� � @ s� U d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d d lm Z d d lm Z d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dl m Z ddl!m"Z" ddl#m$Z$ ddl#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,m.Z. d d"lm/Z/ d d#lm0Z0 d d$l1m2Z2 d d%l3m4Z4 d d&l5m6Z6 d d'l5m7Z7 d d(l5m8Z8 d d)l5m9Z9 dd*l#m:Z: e�;e<�Z=d+d,d-d.�d/d0�Z>e"j?d1d2�Z@d+d3d-d4�d5d6�ZAe@�Bd7�d+d3d8d-d9�d:d;��ZCd<d<d=d3d+d-d>�d?d@�ZDe"�EejFejGejHejIejHejIdA��ZJdBeKdC< dDdEdFdGdH�dIdJ�ZLdDdEdFdKdH�dLdM�ZMdEdFdNdO�dPdQ�ZNejOdRdSdFdFdTd+d=dUdV�dWdX��ZPedYee e e f dZ�ZQe@�Bd[�d+dTdRdSd\d\d-d]�d^d_��ZRd`da� ZSe@�Bdb�d+dcdRdSdSddddd-de�dfdg��ZTe@�Bdb�d+dcdRdSdhddddd-de�didj��ZUe@�Bdb�d+dcdRdSdSddddd-de�dkdl��ZVdmd+dRdn�dodp�ZWdqdqdr�dsdt�ZXd+dcdRdqdqddddd-de�dudv�ZYdqdqd-dw�dxdy�ZZdzd{� Z[e@�Bdb�d+dcdRdSdSddddd|de�d}d~��Z\e@�Bdb�d+dcdRdSdhdddddde�d�d���Z]e@�Bd[�d+dTdRdSdFdFd-d��d�d���Z^e@�Bd[�d+dTdRdSd\d\d-d��d�d���Z_dS )�� )� annotationsN)�Any)�cast)�Dict)�Iterator)�Mapping)�Optional)�Set)�Tuple)� TYPE_CHECKING)�TypeVar)�Union)�event)�inspect��schema)�text)�types)� expression)�ForeignKeyConstraint)�Index)�UniqueConstraint)� OrderedSet� )�util)� is_index_sig)� is_uq_sig)�ops)� sqla_compat)�Literal)� Inspector)� quoted_name)� TextClause)�Column)�Table)�AutogenContext)� DefaultImpl)� AlterColumnOp)�MigrationScript)�ModifyTableOps)� UpgradeOps)�_constraint_sigr% r( �None)�autogen_context�migration_script�returnc C s, |j d }|jd }t| |� |�|� d S )N���)Zupgrade_ops_listZdowngrade_ops_list�_produce_net_changesZ reverse_into)r- r. � upgrade_opsZ downgrade_ops� r3 �J/opt/hc_python/lib/python3.8/site-packages/alembic/autogenerate/compare.py�_populate_migration_script; s r5 T)Zuselistr* )r- r2 r/ c s� � j }|d k st�� j�dd�}t|�}|jj}|r`t|�� �}|� d� |� |� |� d � nd h}� fdd�|D �}� jd k s�t�t � d� jj �� ||� d S )N�include_schemasFZinformation_schemac s h | ]}� � |d i �r|�qS r ��run_name_filters)�.0�s�r- r3 r4 � <setcomp>] s z'_produce_net_changes.<locals>.<setcomp>r )� connection�AssertionError�opts�getr �dialect�default_schema_name�setZget_schema_names�discard�add� comparators�dispatch�name)r- r2 r= r6 � inspector�default_schema�schemasr3 r; r4 r1 H s( � �r1 r z$Union[Set[None], Set[Optional[str]]])r- r2 rK r/ c s� � j }t� }� jj}� jj}|D ]F�t|j�d��}�|krL|�� jjg�}|�� �fdd�|D �� q tdd� � j D ���||fg�}t ||||� � d S )Nr c 3 s( | ] }� � |d d�i�r�|fV qdS )�table� schema_nameNr7 )r9 �tname�r- rM r3 r4 � <genexpr>} s ��z&_autogen_for_tables.<locals>.<genexpr>c S s g | ]}|j |jf�qS r3 )r rH )r9 rL r3 r3 r4 � <listcomp>� s z'_autogen_for_tables.<locals>.<listcomp>) rI rC �migration_context�version_table_schema� version_tableZget_table_names� difference�updater Z sorted_tables�_compare_tables) r- r2 rK rI �conn_table_namesrS rT �tables�metadata_table_namesr3 rO r4 �_autogen_for_tablesg s4 ��� ���r[ rC r )rX rZ rI r2 r- r/ c s |j jj�t�fdd�|D ��}� fdd�t||�D �}|}|�| �D ]�\}}|r^d||f n|} |||f } � �| |ddd �rF|j�tj � | �� t � d| � tj |g |d �} t�d�� | ||d | � | �� sF|j�| � qFt�� } | �|�D ]�\}}t�||�} | | jk} tj|| |d �}| �sHt�|d � jj�|�� t� ||� � �||dd d �r�tj |g |d �} t�d�� | |||d � | �� �s�|j�| � |j�tj � |�� t � d | � q�| � |�}t�� }i }|D ]f\}}t�||�} | |jk} tj|||d �}| �s,t�|d � jj�|�� t� ||� ||||f< �q�t |d d� d�D ]�\}}|�p\d }|�rpd||f n|} |||f } |j| }� �| |dd|��rLtj |g |d �} t |||| | � |�� t�d�� | |||| � W 5 Q R X | �� �sL|j�| � �qLd S )Nc s$ g | ] \}}|� kr|nd |f�qS �Nr3 )r9 r rN )rJ r3 r4 rQ � s �z#_compare_tables.<locals>.<listcomp>c s( i | ] \}\}}|� j t�||� �qS r3 )Ztable_key_to_table� sa_schema�_get_table_key)r9 Zno_dflt_schemar rN r; r3 r4 � <dictcomp>� s � �z#_compare_tables.<locals>.<dictcomp>�%s.%srL FzDetected added table %rr Zcolumn_reflectTzDetected removed table %rc S s | d p d| d fS )Nr � � r3 )�xr3 r3 r4 �<lambda>� � z!_compare_tables.<locals>.<lambda>)�key)!�bindrA rB r �ziprU �run_object_filtersr �appendZ CreateTableOpZ from_table�log�infor) rF rG Zis_emptyr] ZMetaDatar^ rY r$ r �listenrR �implZ _compat_autogen_column_reflectr Z_reflect_tableZ DropTableOp� intersection�sorted�_compare_columns)rX rZ rI r2 r- Z#metadata_table_names_no_dflt_schemaZtname_to_tabler: rN rH �metadata_table�modify_table_opsZremoval_metadata�exists�tZexisting_tablesZexisting_metadataZconn_column_info� conn_tabler3 )r- rJ r4 rW � s� �� �� � � � �� � �� �� � rW )�asc�descZ nulls_firstZ nulls_last� nullsfirst� nullslastzMapping[str, Any]�_IndexColumnSortingOpsr& zDict[str, Any]r$ zOptional[Index])rn �paramsrv r/ c C s� g }|� d�}t|d �D ]x\}}|d krLd|ks6t�|d | }t|�}n|}|j| }|r�||kr�|| D ]} | tkrnt| |�}qn|�|� qtj|d f|�|d |d�| � |d���} d|kr�|d | j d< | S ) NZcolumn_sorting� column_namesZ expressionsrH �unique)r~ �_table�indexZduplicates_constraint) r@ � enumerater> r �cr{ rj r] r � adjust_reflected_dialect_optionsrl ) rn r| rv ZexprsZsorting�numZcol_namerH �item�operator�ixr3 r3 r4 � _make_index( s6 ��� �r� r c sL t j� fdd�|d D �d|d i| �|d���}d|krH|d |jd< |S )Nc s g | ]}� j | �qS r3 �r� �r9 �cname�rv r3 r4 rQ L s z+_make_unique_constraint.<locals>.<listcomp>r} rH �unique_constraint�duplicates_index)r] r r� rl )rn r| rv �uqr3 r� r4 �_make_unique_constraintH s �� �r� r )r| rv r/ c s� | d �| d r d| d �f �| � di �}tj� fdd�| d D ��fdd�| d D �|� d �|� d �|� d �|� d �| d d�}� �|� |S )NZreferred_tableZreferred_schemar` �optionsc s g | ]}� j | �qS r3 r� r� r� r3 r4 rQ ` s z%_make_foreign_key.<locals>.<listcomp>Zconstrained_columnsc s g | ]}d � |f �qS )r` r3 )r9 �n)rN r3 r4 rQ a s Zreferred_columns�onupdate�ondelete� deferrable� initiallyrH )r� r� r� r� rH )r@ r] r Zappend_constraint)r| rv r� �constr3 )rv rN r4 �_make_foreign_keyV s � r� z Optional[str]zUnion[quoted_name, str]r) zIterator[None])r rN rv rr rs r- rI r/ c # sz �rd��f n�}t dd� |jD ��}dd� |jD �} � ��fdd�|jD �} |�| �D ]D} � �| | | ddd �rZ|j�tj���| | �� t� d || � qZ|� | �D ]h} | | } |j| }� �| | dd|�s�q�tj �| �d �}t � d�� |��| || � |�� r�|j�|� q�d V t| ��|�D ]L} � �|j| | dd d ��r(|j�tj���|j| �� t� d || � �q(d S ) Nr` c s s | ]}|j s|jV qd S r\ ��systemrH �r9 r� r3 r3 r4 rP x s z#_compare_columns.<locals>.<genexpr>c S s i | ]}|j s|j|�qS r3 r� r� r3 r3 r4 r_ { s z$_compare_columns.<locals>.<dictcomp>c s* i | ]"}� � |jd ��d��r|j|�qS )�column�Z table_namerM )r8 rH r� �r- r rN r3 r4 r_ s �� r� Fz Detected added column '%s.%s'r Tz Detected removed column '%s.%s')r r� rU ri r rj Z AddColumnOpZfrom_column_and_tablenamerk rl ro r' rF rG Z has_changesrC Z DropColumnOp)r rN rv rr rs r- rI rH Zmetadata_col_namesZmetadata_cols_by_nameZconn_col_namesr� Zcolname� metadata_col�conn_col�alter_column_opr3 r� r4 rq m s� ��� � �� �� � ��rq �_C)�boundrL zOptional[Table])r- � modify_opsr rN rv rr r/ c $ sd � j }�d k�|d k�� jj�|d k rBdd� |jD �}t|j�}n t� }t� }t� } } d� d} �d k �rTz|j� � d�} d� W n& tk r� Y nB t k r� Y n0X � � � fdd�| D �} | D ]} | � d�r�d} q�z|j � � d�} W n tk r� Y nX � � � fd d�| D �} ��r"t� } n��fd d�| D �} d d� ��fd d �| D �D �} | �rnt | | ||� j �� ��| | ||� �fdd�|D �} �fdd�|D �}�fdd�| D �}�fdd�| D �}� fdd�| �|�D �}dd� |D ��dd� |D ��dd� |�|�D �}��fdd�t�����D �}dd� |D �}dd� | D �}� fdd�| D �}dd �� ���� fd d � }dd �� ���� � fd d � }ddd!d"�� �� fd#d$� }tt|��|��D ]~}|| }t|��r�|j|k�r�q�nT||k�r0|| \} �t�fd%d �|D ���r8| j|k�r8|| � |�� n||� �q�tt|��|��D ]�} || } | |k�r�|| \} �t| ��r��}n| }n|| }t|�t| �k�r�||� || � n^| �|�} | j�r�||| | j� n<| j�rt|��r�d&nd'} t � d(| |j | j� n | j �sNt ��qNtt|��|��D ]}!||! }"||"� �q&|D ] }#|#|k�rB|||# � �qBd S ))Nc S s h | ]}t |tj�r|�qS r3 )� isinstancer] r �r9 r� r3 r3 r4 r< � s �z/_compare_indexes_and_uniques.<locals>.<setcomp>Fr Tc s( g | ] }� � |d d��d��r|�qS )rH r� r� r7 r� r� r3 r4 rQ � s ��z0_compare_indexes_and_uniques.<locals>.<listcomp>r� c s( g | ] }� � |d d��d��r|�qS )rH r� r� r7 �r9 r� r� r3 r4 rQ � s ��c s h | ]}t �|� ��qS r3 )r� )r9 Zuq_def�rv rn r3 r4 r< s �c S s h | ]}|d k r|�qS r\ r3 )r9 r� r3 r3 r4 r< s �c 3 s | ]}t �|� �V qd S r\ )r� r� r� r3 r4 rP s z/_compare_indexes_and_uniques.<locals>.<genexpr>c s h | ]}� � |��qS r3 �� _create_metadata_constraint_sigr� �rn r3 r4 r< 3 s �c s h | ]}� � |��qS r3 r� r� r� r3 r4 r< 8 s c s h | ]}� � |��qS r3 �� _create_reflected_constraint_sigr� r� r3 r4 r< < s c s h | ]}� � |��qS r3 r� r� r� r3 r4 r< @ s c s$ i | ] }|j rtt|�� ��|�qS r3 )Zis_namedr �strZmd_name_to_sql_namer� r; r3 r4 r_ E s � z0_compare_indexes_and_uniques.<locals>.<dictcomp>c S s i | ] }|j |�qS r3 �rH r� r3 r3 r4 r_ N s c S s i | ] }|j |�qS r3 r� r� r3 r3 r4 r_ O s c S s i | ]}t �|j�r|j|�qS r3 )r Zconstraint_name_stringrH r� r3 r3 r4 r_ P s � c s i | ]}|�| � | f�qS r3 r3 )r9 rH )�conn_indexes_by_name�conn_uniques_by_namer3 r4 r_ V s � c S s i | ] }|j |�qS r3 ��unnamedr� r3 r3 r4 r_ _ s c S s i | ] }|j |�qS r3 r� r� r3 r3 r4 r_ ` s c s$ i | ] }t �|j� j�s|j|�qS r3 )r Z_constraint_is_namedr� rA r� r� r; r3 r4 r_ c s �� r+ ��objc s� t | �rH� �| j| jddd �r��j�tj�| j�� t� d| j| j � ndt | �r��sXd S �s`�rdd S � �| j| jddd �r��j�tj � | j�� t� d| j| j � nds�t�d S )Nr� Fz!Detected added index '%r' on '%s'r� z+Detected added unique constraint %r on '%s')r ri r� rH r rj � CreateIndexOp� from_indexrk rl r} r �AddConstraintOp�from_constraintr> r� )r- �is_create_table� is_drop_tabler� �supports_unique_constraintsr3 r4 � obj_addeds sF �� � ��z/_compare_indexes_and_uniques.<locals>.obj_addedc s� t | �rT| jr�sd S � �| j| jddd �r��j�tj�| j�� t � d| j�� nZt | �r��sd�rhd S � �| j| jddd �r��j�tj � | j�� t � d| j�� nds�t�d S )Nr� Tz Detected removed index %r on %rr� z+Detected removed unique constraint %r on %rF)r Z is_uniqueri r� rH r rj � DropIndexOpr� rk rl r �DropConstraintOpr� r> r� )r- r� r� r� r� rN r3 r4 � obj_removed� s> � � ��z1_compare_indexes_and_uniques.<locals>.obj_removedr� ��old�new�msgc s� t | �rlt |�st�� �|j|jdd| j�r�t�d| j�|� �j�tj � | j�� �j�tj � |j�� ntt | �r�t |�s�t�� �|j|jdd| j�r�t�d| j�|� �j�tj �| j�� �j�tj�|j�� nds�t�d S )Nr� Fz#Detected changed index %r on %r: %sr� z/Detected changed unique constraint %r on %r: %s)r r> ri r� rH rk rl r rj r� r� r� r r� r� r� r� )r- r� rN r3 r4 � obj_changed� sN � � �� � �z1_compare_indexes_and_uniques.<locals>.obj_changedc 3 s | ]}� j |j kV qd S r\ r� )r9 Zmeta_idx)�conn_idxr3 r4 rP � s �r� zunique constraintz5Cannot compare %s %r, assuming equal and skipping. %s)!rI rR rn � constraintsrC Zindexes� frozensetZget_unique_constraints�NotImplementedError� TypeErrorr@ Z get_indexes� _correct_for_uq_duplicates_uixrA Z correct_for_autogen_constraints�unionro rp rU r r� �allr �typeZcompare_to_reflectedZ is_different�messageZis_skiprk rl rH Zis_equalr> )$r- r� r rN rv rr rI �metadata_unique_constraints�metadata_indexesZ conn_uniques� conn_indexesZ+unique_constraints_duplicate_unique_indexesr� Z metadata_unique_constraints_sigZmetadata_indexes_sig�conn_unique_constraintsZconn_indexes_sigZmetadata_namesZ conn_namesZdoubled_constraintsZconn_uniques_by_sigZmetadata_uniques_by_sigZunnamed_metadata_uniquesr� r� r� Z removed_nameZconn_objZconn_uqZ existing_nameZ metadata_objZ comparison�thingZ added_namer� Zuq_sigr3 ) r- r� r� rv r� rn r� r� r� r r� rN r4 � _compare_indexes_and_uniques� s* � �� � � � �� � � � � � � � � �� � �! %�� ��� �� r� c s� � fdd�|D �}dd� |D �}�fdd�|D �}� fdd�|D �} dd� |D �} d d� | D �} | D ]F} | |kr��� | | �j|kr�| �| | � qd| | krd|�| | � qdd S ) Nc s g | ]}t �|� �|f�qS r3 )r �_get_constraint_final_name�r9 �cons�rA r3 r4 rQ . s �z2_correct_for_uq_duplicates_uix.<locals>.<listcomp>c S s h | ]\}}|d k r|�qS r\ r3 �r9 rH r� r3 r3 r4 r< 3 s z1_correct_for_uq_duplicates_uix.<locals>.<setcomp>c s$ h | ] \}}|d kr� � |�j�qS r\ )r� r� r� r� r3 r4 r< 7 s �c s h | ]}|j rt�|� ��qS r3 )r~ r r� r� r� r3 r4 r< = s �c S s i | ]}|j r|j|�qS r3 )r~ rH r� r3 r3 r4 r_ E s z2_correct_for_uq_duplicates_uix.<locals>.<dictcomp>c S s i | ]}|j d r|j|�qS )r� )rl rH r� r3 r3 r4 r_ G s � )r� r� rD ) r� r� r� r� rA rn Zmetadata_cons_namesZmetadata_uq_namesZunnamed_metadata_uqsZmetadata_ix_namesZ conn_ix_namesZuqs_dupe_indexesZoverlapr3 )rA rn r4 r� s6 �� � �����r� r� r' z Column[Any])r- r� r rN r� r� r� r/ c C sv |j }|j }||_||k rrt�|j|j�r4t�|�sDt�|j|j�rTt�d||� n ||_ t�d|rhdnd||� d S )Nz3Ignoring nullable change on identity column '%s.%s'z Detected %s on column '%s.%s'ZNULLzNOT NULL) ZnullableZexisting_nullabler �_server_default_is_computed�server_defaultZ_nullability_might_be_unset�_server_default_is_identityrk rl Zmodify_nullable) r- r� r rN r� r� r� Zmetadata_col_nullableZconn_col_nullabler3 r3 r4 �_compare_nullableZ s8 ��� �� � �r� r! c C sF |j j|krd|jd< n*|jdkr.d|jd< n|jdkrBd|jd< d S )NT� autoincrementF)rL Z_autoincrement_column�kwr� )r- r� r rN r� r� r� r3 r3 r4 �_setup_autoincrement� s r� c C s| |j }||_|j }|jtjkr0t�d||� d S |jtjkrNt�d||� d S | j�||�} | rx||_ t�d||||� d S )Nz3Couldn't determine database type for column '%s.%s'z:Column '%s.%s' has no type within the model; can't comparez-Detected type change from %r to %r on '%s.%s') r� Z existing_typeZ_type_affinity�sqltypesZNullTyperk rl rR � _compare_typeZ modify_type) r- r� r rN r� r� r� Z conn_typeZ metadata_typeZisdiffr3 r3 r4 r� � s< � � ��r� z Optional[Any])�metadata_defaultr- r/ c C sP t | tj�r:t | jt�r | j} nt| jj|jddid��} t | t�rH| S d S d S )N� literal_bindsT�rA �compile_kwargs)r� r] Z DefaultClause�argr� �compilerA )r� r- r3 r3 r4 �"_render_server_default_for_compare� s �� r� r� )�sqltextr/ c C s t �dd| ��� S )z�we want to warn if a computed sql expression has changed. however we don't want false positives and the warning is not that critical. so filter out most forms of variability from the SQL text. z[ \(\)'\"`\[\]\t\r\n]ra )�re�sub�lower)r� r3 r3 r4 �_normalize_computed_default� s r� c C s� t ttj|j�jj| jddid��}t|�}t |jtj�rnt |jjj| jddid��}|d krdd}qrt|�}nd}||kr�t ||� d S )Nr� Tr� ra ) r� r r] ZComputedr� r� r� rA r� r� � _warn_computed_not_supported) r- r� r rN r� r� r� �rendered_metadata_default�rendered_conn_defaultr3 r3 r4 �_compare_computed_default� s0 �� ����r� )rN r� r/ c C s t �d| |f � d S )Nz,Computed default on %s.%s cannot be modified)r �warn)rN r� r3 r3 r4 r� s r� c C s& | j j}|�|j|j�\}} } || fS r\ )rR rn �_compare_identity_defaultr� ) r- r� r rN r� r� r� rn �diffZ ignored_attr�is_alterr3 r3 r4 r� s � r� zOptional[bool]c C s |j }|j }|d kr |d kr dS t�|�rHtjs4dS t| ||||||�S t�|�r`t||� dS t�||�r�||_t| ||||||�\} } | r�||_ | r�t � d||t | �� nRt || �} |r�tt|�jjnd } ||_| j�||| | �} | �r ||_ t � d||� d S )NFzIDetected server default on column '%s.%s': identity options attributes %sz)Detected server default on column '%s.%s')r� r r� Zhas_computed_reflectionr� r� r� Zexisting_server_defaultr� Zmodify_server_defaultrk rl rp r� r r r� r rR �_compare_server_default)r- r� r rN r� r� r� r� Zconn_col_defaultr� r� r� r� Zis_diffr3 r3 r4 r� sx �� �� � ���r� zOptional[Literal[False]]c C s` | j d k st�| j jsd S |j}|j}|d kr:|d kr:dS ||_||kr\||_t�d||� d S )NFz Detected column comment '%s.%s')rA r> �supports_comments�comment�existing_commentZmodify_commentrk rl ) r- r� r rN r� r� r� Zmetadata_commentZconn_col_commentr3 r3 r4 �_compare_column_commentv s r� )r- rs r rN rv rr r/ c s� �d ks|d krd S � j }dd� |jD �}� ��fdd�|j��d�D �}�fdd�|D �} � jj�� jj�| |� �fdd�|D �} �fdd�| D �} |r�d |d kr�d d � | D �} d d � | D �} n dd � | D �} dd � | D �} dd � | D �}dd � | D �}� ��fdd�}� �fdd�}t| ��| �D ]>}| | �|| k�r*�j|k�rX|�j j nd }|�|� �q*t| ��| �D ]>}| | �|| k�rx�j|k�r�|�j j nd }|�|� �qxd S )Nc S s h | ]}t |tj�r|�qS r3 )r� r] r �r9 Zfkr3 r3 r4 r< � s �z(_compare_foreign_keys.<locals>.<setcomp>c s( g | ] }� � |d d��d��r|�qS )rH �foreign_key_constraintr� r7 r� r� r3 r4 rQ � s ��z)_compare_foreign_keys.<locals>.<listcomp>r c s h | ]}t |� ��qS r3 )r� )r9 r� r� r3 r4 r< � s �c s h | ]}� � |��qS r3 r� r� r� r3 r4 r< � s c s h | ]}� � |��qS r3 r� r� r� r3 r4 r< � s r� r c S s i | ] }|j |�qS r3 r� r� r3 r3 r4 r_ � s z)_compare_foreign_keys.<locals>.<dictcomp>c S s i | ] }|j |�qS r3 r� r� r3 r3 r4 r_ � s c S s i | ] }|j |�qS r3 �Zunnamed_no_optionsr� r3 r3 r4 r_ � s c S s i | ] }|j |�qS r3 r� r� r3 r3 r4 r_ � s c S s i | ]}|j d k r|j |�qS r\ r� r� r3 r3 r4 r_ � s c S s i | ]}|j d k r|j |�qS r\ r� r� r3 r3 r4 r_ � s c sd � � | j| jdd|�r`�j�tj��j�� t�dd� | j �d� | j �| j rVd| j nd| j � d S )Nr� Fz1Detected added foreign key (%s)(%s) on table %s%s�, �%s.ra )ri r� rH r rj ZCreateForeignKeyOpr� rk rl �join�source_columns�target_columns� source_schema� source_table�r� � compare_to)r- r� rs r3 r4 �_add_fk� s" � � �z&_compare_foreign_keys.<locals>._add_fkc sd � � | j| jdd|�r`�j�tj�| j�� t�dd� | j �d� | j �| j rVd| j nd| j � d S )Nr� Tz3Detected removed foreign key (%s)(%s) on table %s%sr r ra )ri r� rH r rj r� r� rk rl r r r r r r )r- rs r3 r4 � _remove_fk� s" � � �z)_compare_foreign_keys.<locals>._remove_fk) rI r� Zget_foreign_keysrR rn Z correct_for_autogen_foreignkeysrC rU rH r� )r- rs r rN rv rr rI Z metadata_fksZ conn_fks_listZconn_fksZmetadata_fks_sigZ conn_fks_sigZconn_fks_by_sigZmetadata_fks_by_sigZmetadata_fks_by_nameZconn_fks_by_namer r Z removed_sigr Z added_sigr3 )r- rv r� rn rs r rN r4 �_compare_foreign_keys� sh � � � � � ��� �� ��r c C s� | j d k st�| j jsd S |d ks*|d kr.d S |jd krF|jd krFd S |jd krv|jd k rv|j�tj||j|d�� n*|j|jkr�|j�tj||j|j|d�� d S )N)r� r )rA r> r� r� r rj ZDropTableCommentOpZCreateTableCommentOp)r- rs r rN rv rr r3 r3 r4 �_compare_table_comment s0 �� ��r )`� __future__r � contextlib�loggingr� �typingr r r r r r r r r r r Z sqlalchemyr r r r] r r r� Zsqlalchemy.sqlr Zsqlalchemy.sql.schemar r r Zsqlalchemy.utilr ra r Z ddl._autogenr r � operationsr r r Z sqlalchemy.engine.reflectionr Zsqlalchemy.sql.elementsr! r" r# r$ Zalembic.autogenerate.apir% Zalembic.ddl.implr& Zalembic.operations.opsr' r( r) r* r+ � getLogger�__name__rk r5 Z DispatcherrF r1 Z dispatch_forr[ rW Z immutabledictrw rx ry rz r{ �__annotations__r� r� r� �contextmanagerrq r� r� r� r� r� r� r� r� r� r� r� r� r� r r r3 r3 r3 r4 �<module> s� * �� H c? ( * , W {