Current File : //proc/227033/root/opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyo
�
�M!Vc @ s( d d l m Z m Z d d l m Z d d l m Z d d l m Z d d l
Z
d d l m Z d d l m
Z
d d l m Z d d
l m Z d d l Z d d l m Z d d l Z d d
l m Z e
j e � Z d � Z e j d e � Z d � Z e j d � d � � Z d � Z! d � Z" d � Z# d � Z$ e j% d � � Z& d e' f d � � YZ( d e( f d � � YZ) d e( f d � � YZ* d e( f d � � YZ+ e j d � d! � � Z, e j d" � d# � � Z- e j d" � d$ � � Z. d% � Z/ e j d" � d& � � Z0 e j d � d' � � Z1 d S(( i����( t schemat types( t Inspector( t eventi ( t opsN( t util( t compat( t sqla_compat( t
OrderedSeti ( t _user_defined_render( t _fk_specc C s8 | j d } | j d } t | | � | j | � d S( Ni����( t upgrade_ops_listt downgrade_ops_listt _produce_net_changest reverse_into( t autogen_contextt migration_scriptt upgrade_opst
downgrade_ops( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _populate_migration_script s
t uselistc C s� | j } | j j d t � } t j | � } | j j } | r{ t | j � � } | j
d � | j d � | j
| � n d g } t
j d | j j � | | | � d S( Nt include_schemast information_schemaR ( t
connectiont optst gett FalseR t from_enginet dialectt default_schema_namet sett get_schema_namest discardt addt Nonet comparatorst dispatcht name( R R R R t inspectort default_schemat schemas( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR
s
R c C s | j } | j } t � } | j j } | j j } xp | D]h } t | j d | � � } | | k r | j | j j g � } n | j t | g t
| � | � � q: Wt g | j D] }
|
j
|
j f ^ q� � j | | f g � } t | | | | | | � d S( NR ( R& t metadataR t migration_contextt version_table_schemat
version_tablet get_table_namest
differencet updatet zipt lenR t
sorted_tablesR R% t _compare_tables( R R R( R& R) t conn_table_namesR+ R, t st tablest tablet metadata_table_names( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _autogen_for_tables4 s
'+ c
s | j j j } t g | D]* \ } } | | k r7 | n d | f ^ q � } t � f d � t | | � D� � }
| } x� | j | � D]� \ } } | r� d | | f n | } |
| | f }
| j |
| d t d � r� | j
j t
j j
|
� � t j d | � t
j | g d | �} t j d � | | | | d |
� | j � sb| j
j | � qbq� q� Wt j � } x� | j | � D]� \ } } t j | | � } | | j k } t j | | d | �} | st j | d | j j j | � � | j | d � n | j | | d t d � r�| j
j t
j j
| � � t j d | � q�q�W| j! | � } t j � } i } x� | D]� \ } } t j | | � } | | j k } t j | | d | �} | s�t j | d | j j j | � � | j | d � n | | | | f <qxWxt" | d d � �D]� \ } } | p7d } | rPd | | f n | } |
| | f }
| j | } | j |
| d t | � r"t
j | g d | �} t# | | | |
| | | � �' t j d � | | | | | |
� Wd QX| j � s| j
j | � qq"q"Wd S(
Nc 3 s: | ]0 \ } \ } } | � j t j | | � f Vq d S( N( R6 t sa_schemat _get_table_key( t .0t no_dflt_schemaR t tname( R) ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>c s s %s.%sR7 s Detected added table %rR t column_reflects Detected removed table %rt keyc S s | d p
d | d f S( Ni t i ( ( t x( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt <lambda>� s ($ t bindR R R R" t dictR0 R. t run_filtersR R t appendt
CreateTableOpt
from_tablet logt infot ModifyTableOpsR# R$ t is_emptyR: t MetaDataR; R6 t TableR t listenR* t implt _compat_autogen_column_reflectt reflecttablet Truet DropTableOpt intersectiont sortedt _compare_columns( R4 R8 R& R) R R R' R R> t# metadata_table_names_no_dflt_schemat tname_to_tableR5 R% t metadata_tablet modify_table_opst removal_metadatat existst tt existing_tablest existing_metadatat conn_column_infot
conn_table( ( R) sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR3 P s� 7
"
c C s< t j | d d | d g | d D] } | j | ^ q" �S( NR% t uniquet column_names( R: t Indext c( t paramsRc t cname( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _make_index� s
c C s5 t j d | d g | d D] } | j | ^ q � S( NR% Re ( R: t UniqueConstraintRg ( Rh Rc Ri ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _make_unique_constraint� s c C s� | d } | d r+ d | d | f } n | j d i � } t j g | d D] } | j | ^ qN g | d D] } d | | f ^ qo d | j d � d | j d � d | j d � d
| j d
� d | d �} | j | � | S( Nt referred_tablet referred_schemas %s.%st optionst constrained_columnst referred_columnst onupdatet ondeletet
deferrablet initiallyR% ( R R: t ForeignKeyConstraintRg t append_constraint( Rh Rc R> Ro Ri t nt const( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _make_foreign_key� s
!'
c c s
| r d | | f n | } t d � | j D� � } t d � | j D� � } t t | � � }
xr |
j | � D]a } | j | | | d t d � rp | j j t j
j | | | | � � t j
d | | � qp qp Wx� |
j | � D]� } | | }
| j | } | j |
| d t | � s#q� n t j | | d | �} t j d � | | | | | | |
� | j � r� | j j | � q� q� Wd Vx~ t | � j |
� D]g } | j | j | | d t d � r�| j j t j j | | | j | � � t j
d | | � q�q�Wd S( Ns %s.%sc s s | ] } | j | f Vq d S( N( R% ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R% ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s t columns Detected added column '%s.%s'R s Detected removed column '%s.%s'( RE Rg R RW R. RF R R" R RG t AddColumnOpt from_column_and_tablenameRJ RK RV t
AlterColumnOpR# R$ t has_changesR RT t DropColumnOp( R R> Rc R[ R\ R R& R% t metadata_cols_by_namet conn_col_namest metadata_col_namesRi t colnamet metadata_colt conn_colt alter_column_op( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyRX � sF
t _constraint_sigc B s# e Z d � Z d � Z d � Z RS( c C s | j | j k S( N( Ry ( t selft other( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt __eq__ s c C s | j | j k S( N( Ry ( R� R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt __ne__ s c C s
t | j � S( N( t hashRy ( R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt __hash__ s ( t __name__t
__module__R� R� R� ( ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR� s t _uq_constraint_sigc B s, e Z e Z e Z d � Z e d � � Z RS( c C sG | | _ | j | _ t t g | j D] } | j ^ q% � � | _ d S( N( Ry R% t tupleRW t columnst sig( R� Ry t col( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt __init__$ s c C s g | j j D] } | j ^ q
S( N( Ry R� R% ( R� R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyRe ) s ( R� R� R t is_indexRT t is_uniqueR� t propertyRe ( ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR� s t _ix_constraint_sigc B s& e Z e Z d � Z e d � � Z RS( c C sY | | _ | j | _ t t g | j D] } | j ^ q% � � | _ t | j � | _ d S( N( Ry R% R� RW R� R� t boolRd R� ( R� Ry R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR� 1 s .c C s t j | j � S( N( R t _get_index_column_namesRy ( R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyRe 7 s ( R� R� RT R� R� R� Re ( ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR� . s t _fk_constraint_sigc B s e Z e d � Z RS( c C s- | | _ | j | _ t | � \
| _ | _ | _ | _ | _ | _ } } } } | j | j t | j � | j | j t | j � f | _
| r)| j
| r� | j � d k r� d q� | j � n d | r� | j � d k r� d q� | j � n d | r| j � d k rd n | rd n d f 7_
n d S( Ns no actiont deferredt initially_deferrableRt s not deferrable(
Ry R% R
t
source_schemat source_tablet source_columnst
target_schemat target_tablet target_columnsR� R� t lowerR" ( R� Ry t include_optionsRr Rs Rt Ru ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR� = s <!-- ( R� R� R R� ( ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR� < s R7 c s� � j } � d k � t d � | j D� � } t | j � } t � } }
t � � d k r t | d � r� y | j � d | �} t � Wq� t
k
r� q� Xn y | j � d | �}
Wn t
k
r� n Xt � f d � | D� � } t � f d � |
D� � }
n � j j
j | |
| | � t d � | D� � } t d � | D� � } t d � | D� � } t d � |
D� � }
t d
� | j | � D� � } t d � | D� � � t d � |
D� � � t d
� | j |
� D� � }
t � � f d � t � � j � � D� � } t d � | D� � } t d � | D� � } t d � | D� � } t d � | D� � } � � � � f d � } � � � � f d � } � � � f d � } x7 t t | � j |
� � D] } | | } | | � q�Wxt t | � j |
� � D]� } | | } | | k rO| | \ } } | j rF| } qY| } n
|
| } | j | j k r�| | � | | � qg } | j | j k r�| j d | j | j f � n | j | j k r�| j d | j | j f � n | r| | | | � qqWx� t t |
� j | � � D]� } |
| } | j rT| j | k rTq%q%| | k r�| j | k r�| j | k r�| | \ } } | | � | | � q�q%| | � q%Wx+ | D]# } | | k r�| | | � q�q�Wd S( Nc s s' | ] } t | t j � r | Vq d S( N( t
isinstanceR: Rk ( R< t uq( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>f s t get_unique_constraintsR c 3 s | ] } t | � � Vq d S( N( Rl ( R< t uq_def( Rc ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr> s c 3 s | ] } t | � � Vq d S( N( Rj ( R< t ix( Rc ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } t | � Vq d S( N( R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } t | � Vq d S( N( R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } t | � Vq d S( N( R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } t | � Vq d S( N( R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s- | ]# } | j d k r | j | f Vq d S( N( R% R" ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R% ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R% ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s- | ]# } | j d k r | j | f Vq d S( N( R% R" ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c 3 s) | ] } | � | � | f f Vq d S( N( ( R< R% ( t conn_indexes_by_namet conn_uniques_by_name( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s- | ]# } | j d k r | j | f Vq d S( N( R% R" R� ( R< R� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s� | j rx � j | j | j d t d � r� � j j t j j | j � � t
j d | j d j d | j
g � � q� n� � s� d S� r� d S� j | j | j d t d � r� � j j t j j | j � � t
j d | j d j d | j
g � � n d S( Nt indexs Detected added index '%s' on %ss , s '%s't unique_constraints+ Detected added unique constraint '%s' on %s( R� RF Ry R% R R" R RG t
CreateIndexOpt
from_indexRJ RK t joinRe t AddConstraintOpt from_constraint( t obj( R t is_create_tablet
modify_opst supports_unique_constraints( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt obj_added� s( c s� | j ry | j r � r d S� j | j | j d t d � r� � j j t j j
| j � � t j d | j � � q� nY � j | j | j d t d � r� � j j t j
j | j � � t j d | j � � n d S( NR� s# Detected removed index '%s' on '%s'R� s/ Detected removed unique constraint '%s' on '%s'( R� R� RF Ry R% RT R" R RG t DropIndexOpR� RJ RK t DropConstraintOpR� ( R� ( R R� R� R> ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt obj_removed� s c s | j r� � j | j | j d t | j � rt j d | j � d j | � � � j j t j
j | j � � � j j t j j | j � � qn� � j | j | j d t | j � rt j d | j � d j | � � � j j t j
j | j � � � j j t j j | j � � n d S( NR� s& Detected changed index '%s' on '%s':%ss , R� s2 Detected changed unique constraint '%s' on '%s':%s( R� RF Ry R% R RJ RK R� R RG R� R� R� R� R� R� ( t oldt newt msg( R R� R> ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt obj_changed� s& s unique=%r to unique=%rs columns %r to %r( R& R" R t constraintst indexest frozensetR t hasattrR� RT t NotImplementedErrort get_indexesR* RQ t correct_for_autogen_constraintsRE t unionRV RW R. R� R� RG R� ( R R� R R> Rc R[ R& t metadata_unique_constraintst metadata_indexest conn_uniquest conn_indexest conn_unique_constraintst metadata_namest
conn_namest doubled_constraintst conn_uniques_by_sigt metadata_uniques_by_sigt metadata_indexes_by_sigt unnamed_metadata_uniquesR� R� R� t
added_nameR� t
existing_namet metadata_objt conn_uqt conn_idxt conn_objR� t removed_namet uq_sig( ( R R� Rc R� R� R� R� R> sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _compare_indexes_and_uniques\ s�
"
"
"
R{ c C sf | j r
d S| j } | j } | | _ | | k rb | | _ t j d | rR d n d | | � n d S( Ns Detected %s on column '%s.%s't NULLs NOT NULL( t primary_keyt nullablet existing_nullablet modify_nullableRJ RK ( R R� R R> Ri R� R� t metadata_col_nullablet conn_col_nullable( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _compare_nullableH s c
C s� | j } | | _ | j } | j t j k rD t j d | | � d S| j t j k rm t j d | | � d S| j j | | � } | r� | | _ t j d | | | | � n d S( Ns3 Couldn't determine database type for column '%s.%s's: Column '%s.%s' has no type within the model; can't compares- Detected type change from %r to %r on '%s.%s'(
t typet
existing_typet _type_affinityt sqltypest NullTypeRJ RK R* t
_compare_typet modify_type(
R R� R R> Ri R� R� t conn_typet
metadata_typet isdiff( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR� ^ s"
c C s� t d | | � } | t k r" | St | t j � rv t | j t j � rU | j } qv t | j j d | j
� � } n t | t j � r� | j j t
j k r� t j d d | � } t | � S| Sn d Sd S( Nt server_defaultR s ^'|'$RA ( R R R� R: t
DefaultClauset argR t string_typest strt compileR R� R� R� t Stringt ret subt reprR" ( t metadata_defaultR� R t rendered( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt" _render_server_default_for_comparey s
c C s� | j } | j } | d k r. | d k r. t St | | | � } | j rX | j j j n d }
| | _ | j j | | | |
� } | r� | | _ t
j d | | � n d S( Ns) Detected server default on column '%s.%s'( R� R" R R� R� t textt existing_server_defaultR* t _compare_server_defaultt modify_server_defaultRJ RK ( R R� R R> Ri R� R� R� t conn_col_defaultt rendered_metadata_defaultt rendered_conn_defaultR� ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyR � s" c s � d k r d S� j } t d � | j D� � } | j | d | �} | oZ d | d k � t � f d � | D� � } � j j j | | � t � f d � | D� � } t � f d � | D� � } t d � | D� � } t d � | D� � }
t d
� | D� � } t d � | D� � } � � � f d � }
� � f d
� } xh t | � j |
� D]Q } | | � | |
k r\� j
| k r�| � j
j n d } | � | � q\q\Wxh t |
� j | � D]Q } |
| � | | k r�� j
| k r| � j
j n d } |
� | � q�q�Wd S( Nc s s' | ] } t | t j � r | Vq d S( N( R� R: Rv ( R< t fk( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s R Ro i c 3 s | ] } t | � � Vq d S( N( Rz ( R< Ry ( Rc ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c 3 s! | ] } t | d � �Vq d S( R� N( R� ( R< R ( t backend_reflects_fk_options( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c 3 s! | ] } t | d � �Vq d S( R� N( R� ( R< R ( R ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R� ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s | ] } | j | f Vq d S( N( R� ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s- | ]# } | j d k r | j | f Vq d S( N( R% R" ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s s- | ]# } | j d k r | j | f Vq d S( N( R% R" ( R< Rg ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pys <genexpr>� s c s� � j | j | j d t | � r� � j j t j j � j � � t j d d j
| j � d j
| j � | j
r} d | j
n d | j � n d S( Nt foreign_key_constraints1 Detected added foreign key (%s)(%s) on table %s%ss , s %s.RA ( RF Ry R% R R RG t CreateForeignKeyOpR� RJ RK R� R� R� R� R� ( R� t
compare_to( R Ry R\ ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _add_fk� s c s� � j | j | j d t | � r� � j j t j j | j � � t j d d j
| j � d j
| j � | j
r} d | j
n d | j � n d S( NR s3 Detected removed foreign key (%s)(%s) on table %s%ss , s %s.RA ( RF Ry R% RT R RG R� R� RJ RK R� R� R� R� R� ( R� R ( R R\ ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt
_remove_fk� s ( R" R& R R� t get_foreign_keysR* RQ t correct_for_autogen_foreignkeysRE R. R% Ry ( R R\ R R> Rc R[ R& t metadata_fkst conn_fkst conn_fks_by_sigt metadata_fks_by_sigt metadata_fks_by_namet conn_fks_by_nameR
R t removed_sigR t added_sig( ( R R Rc Ry R\ sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt _compare_foreign_keys� sH
%
%(2 t
sqlalchemyR R: R R� t sqlalchemy.engine.reflectionR R t
operationsR t loggingRA R R R t sqlalchemy.utilR R� t renderR t
contextlibt alembic.ddl.baseR
t getLoggerR� RJ R t
DispatcherRT R# R
t dispatch_forR9 R3 Rj Rl Rz t contextmanagerRX t objectR� R� R� R� R� R� R� R� R R ( ( ( sM /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/compare.pyt <module> s@ l 2 �
Mini Shell