Edit File: compare.cpython-37.pyc
B �M!V|o � @ s� d dl mZmZ d dlmZ d dl mZ ddlm Z d dl Z ddl m Z ddl m Z dd l mZ d d lmZ d dlZd d lmZ d dlZd d lmZ e �e�Zdd� Ze jdd�Z dd� Z e � d�dd� �Z dd� Z dd� Z!dd � Z"d d � Z#ej$d d � �Z%G d!d"� d"e&�Z'G d#d$� d$e'�Z(G d%d&� d&e'�Z)G d'd(� d(e'�Z*e � d)�d*d+� �Z+e � d,�d-d.� �Z,e � d,�d/d0� �Z-d1d2� Z.e � d,�d3d4� �Z/e � d)�d5d6� �Z0dS )7� )�schema�types)� Inspector)�event� )�opsN)�util)�compat)� sqla_compat)� OrderedSet� )�_user_defined_render)�_fk_specc C s, |j d }|jd }t| |� |�|� d S )N���)Zupgrade_ops_listZdowngrade_ops_list�_produce_net_changesZ reverse_into)�autogen_contextZmigration_script� upgrade_opsZ downgrade_ops� r �M/opt/alt/python37/lib/python3.7/site-packages/alembic/autogenerate/compare.py�_populate_migration_script s r T)Zuselistc C sx | j }| j�dd�}t�|�}|jj}|rVt|�� �}|� d� |� d � |� |� nd g}t � d| jj �| ||� d S )N�include_schemasFZinformation_schemar )� connection�opts�getr Z from_engine�dialect�default_schema_name�setZget_schema_names�discard�add� comparators�dispatch�name)r r r r � inspector�default_schema�schemasr r r r s r r c C s� | j }| j}t� }| jj}| jj}xN|D ]F}t|j|d��} ||krT| �| jjg�} |�t |gt | � | �� q(W t dd� |j D ���||fg�} t || |||| � d S )N)r c S s g | ]}|j |jf�qS r )r r! )�.0�tabler r r � <listcomp>I s z'_autogen_for_tables.<locals>.<listcomp>)r" �metadatar �migration_context�version_table_schema� version_tableZget_table_names� difference�update�zip�lenr Z sorted_tables�_compare_tables) r r r$ r" r( �conn_table_namesr* r+ �s�tables�metadata_table_namesr r r �_autogen_for_tables4 s r5 c s� |j jj� t� fdd�|D ��}t�fdd�t||�D ��}|}x�|�| �D ]�\}} |rdd|| f n| } ||| f } |�| | ddd �rL|j� tj � | �� t � d| � tj| g |d �} t�d�|| || d | � | �� sL|j� | � qLW t�� } x�| �|�D ]�\}} t�| |�} | | jk}tj| | |d �}|�sRt�|d |jj�|�� |� |d � |�|| dd d �r�|j� tj � |�� t � d | � q�W | � |�}t�� }i }xn|D ]f\}} t�| |�} | |jk}tj| ||d �}|�s�t�|d |jj�|�� |� |d � |||| f< �q�W x�t |d d� d�D ]�\}} |�p0d }|�rDd|| f n| } ||| f } |j| }|�| | dd|��r tj| g |d �} t || || | ||�� t�d�|| || || � W d Q R X | �� �s |j� | � �q W d S )Nc s$ g | ] \}}|� kr|nd |f�qS )Nr )r% r �tname)r# r r r' [ s z#_compare_tables.<locals>.<listcomp>c 3 s, | ]$\}\}}|� j t�||� fV qd S )N)r3 � sa_schema�_get_table_key)r% Zno_dflt_schemar r6 )r( r r � <genexpr>c s z"_compare_tables.<locals>.<genexpr>z%s.%sr& FzDetected added table %r)r Zcolumn_reflectTzDetected removed table %rc S s | d p d| d fS )Nr � r r )�xr r r �<lambda>� � z!_compare_tables.<locals>.<lambda>)�key)!�bindr r r �dictr. r, � run_filtersr �appendZ CreateTableOpZ from_table�log�infoZModifyTableOpsr r Zis_emptyr7 ZMetaDatar8 r3 ZTabler �listenr) �implZ _compat_autogen_column_reflectZ reflecttableZ DropTableOp� intersection�sorted�_compare_columns)r1 r4 r" r( r r Z#metadata_table_names_no_dflt_schemaZtname_to_tabler2 r6 r! �metadata_table�modify_table_opsZremoval_metadata�exists�tZexisting_tablesZexisting_metadataZconn_column_info� conn_tabler )r# r( r r0 P s� r0 c s0 t j| d f� fdd�| d D ��d| d i�S )Nr! c s g | ]}� j | �qS r )�c)r% �cname)rN r r r' � s z _make_index.<locals>.<listcomp>� column_names�unique)r7 �Index)�paramsrN r )rN r � _make_index� s rU c s& t j� fdd�| d D �d| d i�S )Nc s g | ]}� j | �qS r )rO )r% rP )rN r r r' � s z+_make_unique_constraint.<locals>.<listcomp>rQ r! )r7 �UniqueConstraint)rT rN r )rN r �_make_unique_constraint� s rW 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_schemaz%s.%s�optionsc s g | ]}� j | �qS r )rO )r% rP )rN r r r' � s z%_make_foreign_key.<locals>.<listcomp>Zconstrained_columnsc s g | ]}d � |f �qS )z%s.%sr )r% �n)r6 r r r' � s Zreferred_columns�onupdate�ondelete� deferrable� initiallyr! )rZ r[ r\ r] r! )r r7 �ForeignKeyConstraintZappend_constraint)rT rN rX �constr )rN r6 r �_make_foreign_key� s r` c c s~ | rd| |f n|}t dd� |jD ��}t dd� |jD ��} tt|��} xR| �| �D ]D} |�|| | ddd �rT|j�tj� | ||| �� t � d|| � qTW xv| � | �D ]h} || } |j| }|�| | dd|�s�q�tj || | d�}t�d�||| || || � |�� r�|j�|� q�W d V x^t| ��| �D ]L} |�|j| | dd d ��r*|j�tj� | ||j| �� t � d || � �q*W d S ) Nz%s.%sc s s | ]}|j |fV qd S )N)r! )r% rO r r r r9 � s z#_compare_columns.<locals>.<genexpr>c s s | ]}|j |fV qd S )N)r! )r% rO r r r r9 � s �columnFz Detected added column '%s.%s')r Tz Detected removed column '%s.%s')r@ rO r rH r, rA r rB Z AddColumnOpZfrom_column_and_tablenamerC rD rG Z AlterColumnOpr r Z has_changesr Z DropColumnOp)r r6 rN rJ rK r r" r! Zmetadata_cols_by_nameZconn_col_namesZmetadata_col_namesrP Zcolname� metadata_col�conn_col�alter_column_opr r r rI � sF rI c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�_constraint_sigc C s | j |j kS )N)r_ )�self�otherr r r �__eq__ s z_constraint_sig.__eq__c C s | j |j kS )N)r_ )rf rg r r r �__ne__ s z_constraint_sig.__ne__c C s t | j�S )N)�hashr_ )rf r r r �__hash__ s z_constraint_sig.__hash__N)�__name__� __module__� __qualname__rh ri rk r r r r re s re c @ s( e Zd ZdZdZdd� Zedd� �ZdS )�_uq_constraint_sigFTc C s, || _ |j| _ttdd� |jD ���| _d S )Nc S s g | ] }|j �qS r )r! )r% �colr r r r' ' s z/_uq_constraint_sig.__init__.<locals>.<listcomp>)r_ r! �tuplerH �columns�sig)rf r_ r r r �__init__$ s z_uq_constraint_sig.__init__c C s dd� | j jD �S )Nc S s g | ] }|j �qS r )r! )r% rp r r r r' + s z3_uq_constraint_sig.column_names.<locals>.<listcomp>)r_ rr )rf r r r rQ ) s z _uq_constraint_sig.column_namesN)rl rm rn �is_index� is_uniquert �propertyrQ r r r r ro s ro c @ s$ e Zd ZdZdd� Zedd� �ZdS )�_ix_constraint_sigTc C s8 || _ |j| _ttdd� |jD ���| _t|j�| _d S )Nc S s g | ] }|j �qS r )r! )r% rp r r r r' 4 s z/_ix_constraint_sig.__init__.<locals>.<listcomp>) r_ r! rq rH rr rs �boolrR rv )rf r_ r r r rt 1 s z_ix_constraint_sig.__init__c C s t �| j�S )N)r Z_get_index_column_namesr_ )rf r r r rQ 7 s z _ix_constraint_sig.column_namesN)rl rm rn ru rt rw rQ r r r r rx . s rx c @ s e Zd Zddd�ZdS )�_fk_constraint_sigFc C s� || _ |j| _t|�\ | _| _| _| _| _| _}}}}| j| jt | j�| j| jt | j�f| _ |r�| j |r�|� � dkrzd q�|� � nd |r�|� � dkr�d q�|� � nd |r�|� � dkr�dn |r�dndf7 _ d S )Nz no actionZdeferredZinitially_deferrabler\ znot deferrable) r_ r! r � source_schema� source_table�source_columnsZ target_schemaZ target_table�target_columnsrq rs �lower)rf r_ �include_optionsrZ r[ r\ r] r r r rt = s ( z_fk_constraint_sig.__init__N)F)rl rm rn rt r r r r rz < s rz r& c s� � j }�d k�tdd� |jD ��}t|j�}t� } } d��d k r�t|d�rxy|j�|d�} d�W n tk rv Y nX y|j�|d�} W n tk r� Y nX t�fdd�| D ��} t�fdd�| D ��} � j j � | | ||� td d� |D ��}td d� |D ��}td d� | D ��} td d� | D ��} t d d� |� |�D ��} t dd� | D ���t dd� | D ���t dd� | � | �D ��} t ��fdd�t�����D ��}t dd� | D ��}t dd� |D ��}t dd� |D ��}t dd� |D ��}� ���fdd�}� ���fdd�}� ��fdd�}x,tt| ��| ��D ]}| | }||� �q.W x�tt| ��| ��D ]�}| | }||k�r�|| \}}|j�r�|} n|} n| | } | j|jk�r�|| � ||� n^g } | j|jk�r�| �d | j|jf � | j|jk�r| �d | j|jf � | �r\|| || � �q\W x�tt| ��| ��D ]p} | | } | j�sX| j|k�rX�q2nH| |k�r�| j|k�r�| j|k�r�|| \}}||� ||� n|| � �q2W x$|D ] } | |k�r�||| � �q�W d S ) Nc s s | ]}t |tj�r|V qd S )N)� isinstancer7 rV )r% �uqr r r r9 f s z/_compare_indexes_and_uniques.<locals>.<genexpr>F�get_unique_constraints)r Tc 3 s | ]}t |� �V qd S )N)rW )r% Zuq_def)rN r r r9 s c 3 s | ]}t |� �V qd S )N)rU )r% �ix)rN r r r9 � s c s s | ]}t |�V qd S )N)ro )r% r� r r r r9 � s c s s | ]}t |�V qd S )N)rx )r% r� r r r r9 � s c s s | ]}t |�V qd S )N)ro )r% r� r r r r9 � s c s s | ]}t |�V qd S )N)rx )r% r� r r r r9 � s c s s" | ]}|j d k r|j |fV qd S )N)r! )r% rO r r r r9 � s c s s | ]}|j |fV qd S )N)r! )r% rO r r r r9 � s c s s | ]}|j |fV qd S )N)r! )r% rO r r r r9 � s c s s" | ]}|j d k r|j |fV qd S )N)r! )r% rO r r r r9 � s c 3 s" | ]}|�| � | ffV qd S )Nr )r% r! )�conn_indexes_by_name�conn_uniques_by_namer r r9 � s c s s | ]}|j |fV qd S )N)rs )r% r� r r r r9 � s c s s | ]}|j |fV qd S )N)rs )r% r� r r r r9 � s c s s | ]}|j |fV qd S )N)rs )r% r� r r r r9 � s c s s" | ]}|j d kr|j|fV qd S )N)r! rs )r% r� r r r r9 � s c s� | j rR� �| j| jddd �r��j�tj�| j�� t� d| jd� d| j g�� nZ�sZd S �rbd S � �| j| jddd �r��j�tj � | j�� t� d| jd� d| j g�� d S )N�indexFz Detected added index '%s' on %sz, z'%s'�unique_constraintz+Detected added unique constraint '%s' on %s)ru rA r_ r! r rB � CreateIndexOp� from_indexrC rD �joinrQ �AddConstraintOp�from_constraint)�obj)r �is_create_table� modify_ops�supports_unique_constraintsr r � obj_added� s( z/_compare_indexes_and_uniques.<locals>.obj_addedc s� | j rR| jr�sd S � �| j| jddd �r��j�tj�| j�� t � d| j�� n<� �| j| jddd �r��j�tj � | j�� t � d| j�� d S )Nr� Tz#Detected removed index '%s' on '%s'r� z/Detected removed unique constraint '%s' on '%s') ru rv rA r_ r! r rB � DropIndexOpr� rC rD �DropConstraintOpr� )r� )r r� r� r6 r r � obj_removed� s z1_compare_indexes_and_uniques.<locals>.obj_removedc s� | j rd� �|j|jdd| j�r�t�d| j�d�|�� �j�tj � | j�� �j�tj � |j�� n\� �|j|jdd| j�r�t�d| j�d�|�� �j�tj � | j�� �j�tj� |j�� d S )Nr� Fz&Detected changed index '%s' on '%s':%sz, r� z2Detected changed unique constraint '%s' on '%s':%s)ru rA r_ r! rC rD r� r rB r� r� r� r� r� r� )�old�new�msg)r r� r6 r r � obj_changed� s& z1_compare_indexes_and_uniques.<locals>.obj_changedz unique=%r to unique=%rz columns %r to %r)r" r � constraints�indexes� frozenset�hasattrr� �NotImplementedErrorZ get_indexesr) rF Z correct_for_autogen_constraintsr@ �unionrG rH r, ru rv rB rs ) r r� r r6 rN rJ r" Zmetadata_unique_constraintsZmetadata_indexesZ conn_uniquesZ conn_indexesZconn_unique_constraintsZmetadata_namesZ conn_namesZdoubled_constraintsZconn_uniques_by_sigZmetadata_uniques_by_sigZmetadata_indexes_by_sigZunnamed_metadata_uniquesr� r� r� Z added_namer� Z existing_nameZ metadata_objZconn_uqZconn_idxZconn_objr� Z removed_nameZuq_sigr )r r� rN r� r� r� r� r6 r � _compare_indexes_and_uniques\ s� r� ra c C sF |j r d S |j}|j}||_||k rB||_t�d|r8dnd||� d S )Nz Detected %s on column '%s.%s'�NULLzNOT NULL)Z primary_keyZnullableZexisting_nullableZmodify_nullablerC rD ) r rd r r6 rP rc rb Zmetadata_col_nullableZconn_col_nullabler r r �_compare_nullableH 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') �typeZ existing_type�_type_affinity�sqltypesZNullTyperC rD r) � _compare_typeZ modify_type) r rd r r6 rP rc rb Z conn_typeZ metadata_type�isdiffr r r r� ^ s" r� c C s� t d| |�}|dk r|S t| tj�rNt| jtj�r:| j} nt| jj|j d��} t| tj�r�|j j t j kr~t�dd| �} t| �S | S nd S d S )N�server_defaultF)r z^'|'$r: )r r� r7 Z DefaultClause�argr � string_types�str�compiler r� r� r� �String�re�sub�repr)�metadata_defaultrb r �renderedr r r �"_render_server_default_for_comparey s r� c C st |j }|j }|d kr |d kr dS t||| �} |j r<|j jjnd } ||_| j�||| | �} | rp||_t� d||� d S )NFz)Detected server default on column '%s.%s') r� r� r� �textZexisting_server_defaultr) �_compare_server_defaultZmodify_server_defaultrC rD ) r rd r r6 rP rc rb r� Zconn_col_defaultZrendered_metadata_defaultZrendered_conn_defaultr� r r r r� � s" r� c s� �d kr d S � j }tdd� |jD ��}|j||d�}|oBd|d k�t�fdd�|D ��}� jj�||� t�fdd�|D ��}t�fdd�|D ��}td d� |D ��} td d� |D ��} td d� |D ��} td d� |D ��} � ��fd d�} � �fdd�}xPt| ��| �D ]>}| | �|| k�r �j | k�r:| �j j nd }|�|� �q W xPt| ��| �D ]>}| | �|| k�r^�j | k�r�| �j j nd }| �|� �q^W d S )Nc s s | ]}t |tj�r|V qd S )N)r� r7 r^ )r% �fkr r r r9 � s z(_compare_foreign_keys.<locals>.<genexpr>)r rX r c 3 s | ]}t |� �V qd S )N)r` )r% r_ )rN r r r9 � s c 3 s | ]}t |� d �V qdS ))r� N)rz )r% r� )�backend_reflects_fk_optionsr r r9 � s c 3 s | ]}t |� d �V qdS ))r� N)rz )r% r� )r� r r r9 � s c s s | ]}|j |fV qd S )N)rs )r% rO r r r r9 � s c s s | ]}|j |fV qd S )N)rs )r% rO r r r r9 � s c s s" | ]}|j d k r|j |fV qd S )N)r! )r% rO r r r r9 � s c s s" | ]}|j d k r|j |fV qd S )N)r! )r% rO r r r r9 � s c sd � � | j| jdd|�r`�j�tj��j�� t�dd� | j �d� | j �| j rVd| j nd| j � d S )N�foreign_key_constraintFz1Detected added foreign key (%s)(%s) on table %s%sz, z%s.r: )rA r_ r! r rB ZCreateForeignKeyOpr� rC rD r� r} r~ r{ r| )r� � compare_to)r r_ rK r r �_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%sz, z%s.r: )rA r_ r! r rB r� r� rC rD r� r} r~ r{ r| )r� r� )r rK r r � _remove_fk� s z)_compare_foreign_keys.<locals>._remove_fk) r" r r� Zget_foreign_keysr) rF Z correct_for_autogen_foreignkeysr@ r, r! r_ )r rK r r6 rN rJ r" Z metadata_fksZconn_fksZconn_fks_by_sigZmetadata_fks_by_sigZmetadata_fks_by_nameZconn_fks_by_namer� r� Z removed_sigr� Z added_sigr )r r� rN r_ rK r �_compare_foreign_keys� sH r� )1Z sqlalchemyr r7 r r� Z sqlalchemy.engine.reflectionr r Z operationsr �loggingr: r r r Zsqlalchemy.utilr r� Zrenderr � contextlibZalembic.ddl.baser � getLoggerrl rC r Z Dispatcherr r Z dispatch_forr5 r0 rU rW r` �contextmanagerrI �objectre ro rx rz r� r� r� r� r� r� r r r r �<module> sB l 2 m