Edit File: compare.cpython-37.opt-1.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 )� connectionZopts�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 r4 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)r2 � sa_schema�_get_table_key)r$ Zno_dflt_schemar r5 )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)!Zbindr r r �dictr- r+ � run_filtersr �appendZ CreateTableOpZ from_table�log�infoZModifyTableOpsr r Zis_emptyr6 ZMetaDatar7 r2 ZTabler Zlistenr( �implZ _compat_autogen_column_reflectZ reflecttableZ DropTableOp� intersection�sorted�_compare_columns)r0 r3 r! r' r r Z#metadata_table_names_no_dflt_schemaZtname_to_tabler1 r5 r �metadata_table�modify_table_opsZremoval_metadata�exists�tZexisting_tablesZexisting_metadataZconn_column_info� conn_tabler )r" r' r r/ P s� r/ c s0 t j| d f� fdd�| d D ��d| d i�S )Nr c s g | ]}� j | �qS r )�c)r$ �cname)rK r r r& � s z _make_index.<locals>.<listcomp>� column_names�unique)r6 ZIndex)�paramsrK r )rK r � _make_index� s rQ c s& t j� fdd�| d D �d| d i�S )Nc s g | ]}� j | �qS r )rL )r$ rM )rK r r r& � s z+_make_unique_constraint.<locals>.<listcomp>rN r )r6 �UniqueConstraint)rP rK r )rK r �_make_unique_constraint� s rS 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 )rL )r$ rM )rK r r r& � s z%_make_foreign_key.<locals>.<listcomp>Zconstrained_columnsc s g | ]}d � |f �qS )z%s.%sr )r$ �n)r5 r r r& � s Zreferred_columns�onupdate�ondelete� deferrable� initiallyr )rV rW rX rY r )r r6 �ForeignKeyConstraintZappend_constraint)rP rK rT �constr )rK r5 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$ rL r r r r8 � s z#_compare_columns.<locals>.<genexpr>c s s | ]}|j |fV qd S )N)r )r$ rL r r r r8 � s �columnFz Detected added column '%s.%s')r Tz Detected removed column '%s.%s')r> rL r rE r+ r? r r@ Z AddColumnOpZfrom_column_and_tablenamerA rB rD Z AlterColumnOpr r Z has_changesr Z DropColumnOp)r r5 rK rG rH r r! r Zmetadata_cols_by_nameZconn_col_namesZmetadata_col_namesrM Zcolname� metadata_col�conn_col�alter_column_opr r r rF � sF rF 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[ )rb rc r r r �__ne__ s z_constraint_sig.__ne__c C s t | j�S )N)�hashr[ )rb r r r �__hash__ s z_constraint_sig.__hash__N)�__name__� __module__� __qualname__rd re rg r r r r ra s ra 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 �tuplerE �columns�sig)rb 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$ rl r r r r& + s z3_uq_constraint_sig.column_names.<locals>.<listcomp>)r[ rn )rb r r r rN ) s z _uq_constraint_sig.column_namesN)rh ri rj �is_index� is_uniquerp �propertyrN r r r r rk s rk 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$ rl r r r r& 4 s z/_ix_constraint_sig.__init__.<locals>.<listcomp>) r[ r rm rE rn ro �boolrO rr )rb r[ r r r rp 1 s z_ix_constraint_sig.__init__c C s t �| j�S )N)r Z_get_index_column_namesr[ )rb r r r rN 7 s z _ix_constraint_sig.column_namesN)rh ri rj rq rp rs rN r r r r rt . s rt 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_deferrablerX znot deferrable) r[ r r � source_schema� source_table�source_columnsZ target_schemaZ target_table�target_columnsrm ro �lower)rb r[ �include_optionsrV rW rX rY r r r rp = s ( z_fk_constraint_sig.__init__N)F)rh ri rj rp r r r r rv < s rv 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)� isinstancer6 rR )r$ �uqr r r r8 f s z/_compare_indexes_and_uniques.<locals>.<genexpr>F�get_unique_constraints)r Tc 3 s | ]}t |� �V qd S )N)rS )r$ Zuq_def)rK r r r8 s c 3 s | ]}t |� �V qd S )N)rQ )r$ �ix)rK r r r8 � s c s s | ]}t |�V qd S )N)rk )r$ r~ r r r r8 � s c s s | ]}t |�V qd S )N)rt )r$ r� r r r r8 � s c s s | ]}t |�V qd S )N)rk )r$ r~ r r r r8 � s c s s | ]}t |�V qd S )N)rt )r$ r� r r r r8 � s c s s" | ]}|j d k r|j |fV qd S )N)r )r$ rL r r r r8 � s c s s | ]}|j |fV qd S )N)r )r$ rL r r r r8 � s c s s | ]}|j |fV qd S )N)r )r$ rL r r r r8 � s c s s" | ]}|j d k r|j |fV qd S )N)r )r$ rL r r r r8 � s c 3 s" | ]}|�| � | ffV qd S )Nr )r$ r )�conn_indexes_by_name�conn_uniques_by_namer r r8 � s c s s | ]}|j |fV qd S )N)ro )r$ r~ r r r r8 � s c s s | ]}|j |fV qd S )N)ro )r$ r~ r r r r8 � s c s s | ]}|j |fV qd S )N)ro )r$ r� r r r r8 � s c s s" | ]}|j d kr|j|fV qd S )N)r ro )r$ r~ r r r r8 � 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)rq r? r[ r r r@ � CreateIndexOp� from_indexrA rB �joinrN �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') rq rr r? r[ r r r@ � DropIndexOpr� rA rB �DropConstraintOpr� )r� )r r� r� r5 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)rq r? r[ r rA rB r� r r@ r� r� r� r� r� r� )�old�new�msg)r r� r5 r r � obj_changed� s& z1_compare_indexes_and_uniques.<locals>.obj_changedz unique=%r to unique=%rz columns %r to %r)r! r � constraintsZindexes� frozenset�hasattrr �NotImplementedErrorZ get_indexesr( rC Z correct_for_autogen_constraintsr> �unionrD rE r+ rq rr r@ ro ) r r� r r5 rK rG 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� rK r� r� r� r� r5 r � _compare_indexes_and_uniques\ s� r� r] c C sF |j r d S |j}|j}||_||k rB||_t�d|r8dnd||� d S )Nz Detected %s on column '%s.%s'ZNULLzNOT NULL)Z primary_keyZnullableZexisting_nullableZmodify_nullablerA rB ) r r` r r5 rM r_ r^ 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�sqltypesZNullTyperA rB r( � _compare_typeZ modify_type) r r` r r5 rM r_ r^ 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^'|'$r9 )r r} r6 Z DefaultClause�argr Z string_types�str�compiler r� r� r� �String�re�sub�repr)�metadata_defaultr^ r Zrenderedr 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_defaultrA rB ) r r` r r5 rM r_ r^ 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} r6 rZ )r$ �fkr r r r8 � s z(_compare_foreign_keys.<locals>.<genexpr>)r rT r c 3 s | ]}t |� �V qd S )N)r\ )r$ r[ )rK r r r8 � s c 3 s | ]}t |� d �V qdS ))r| N)rv )r$ r� )�backend_reflects_fk_optionsr r r8 � s c 3 s | ]}t |� d �V qdS ))r| N)rv )r$ r� )r� r r r8 � s c s s | ]}|j |fV qd S )N)ro )r$ rL r r r r8 � s c s s | ]}|j |fV qd S )N)ro )r$ rL r r r r8 � s c s s" | ]}|j d k r|j |fV qd S )N)r )r$ rL r r r r8 � s c s s" | ]}|j d k r|j |fV qd S )N)r )r$ rL r r r r8 � 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.r9 )r? r[ r r r@ ZCreateForeignKeyOpr� rA rB r� ry rz rw rx )r� � compare_to)r r[ rH 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.r9 )r? r[ r r r@ r� r� rA rB r� ry rz rw rx )r� r� )r rH r r � _remove_fk� s z)_compare_foreign_keys.<locals>._remove_fk) r! r r� Zget_foreign_keysr( rC Z correct_for_autogen_foreignkeysr> r+ r r[ )r rH r r5 rK rG 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� rK r[ rH r �_compare_foreign_keys� sH r� )1Z sqlalchemyr r6 r r� Z sqlalchemy.engine.reflectionr r Z operationsr Zloggingr9 r r r Zsqlalchemy.utilr r� Zrenderr � contextlibZalembic.ddl.baser Z getLoggerrh rA r Z Dispatcherr r Z dispatch_forr4 r/ rQ rS r\ �contextmanagerrF �objectra rk rt rv r� r� r� r� r� r� r r r r �<module> sB l 2 m