Current File : //opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyo
�
�M!Vc @ s� d d l m Z m Z 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
Z y d d l m Z d � Z Wn e k
r� d
� Z n Xd � Z d � Z e j � Z Z d � Z d � Z d � Z e j e j � d � � Z e j e j! � d � � Z" e j e j# � d � � Z$ e j e j% � d � � Z& e j e j' � d � � Z( e j e j) � d � � Z* e j e j+ � d � � Z, e j e j- � d � � Z. e j e j/ � d � � Z0 e j e j1 � d � � Z2 e j e j3 � d � � Z4 e j e j5 � d � � Z6 e j e j7 � d � � Z8 d e9 f d! � � YZ: d" � Z; e< d# � Z= d$ � Z> d% � Z? d& � Z@ d' � ZA d( � ZB d) � ZC d* � ZD e< d+ � ZE d, � ZF e j � ZG d- � ZH eG j e jI � d. � � ZJ d/ � ZK d0 � ZL eG j e jM � d1 � � ZN eG j e jO � d2 � � ZP eG j e jQ � d3 � � ZR e jS � Z d S(4 i����( t schemat typest sqli ( t ops( t compatN( t string_types( t util( t
PythonPrinter( t StringIOi� ( t convc C s* t | t � r" t t | � | � S| Sd S( N( t
isinstanceR t _f_namet _alembic_autogenerate_prefix( t autogen_contextt name( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _render_gen_name s c C s | S( N( ( R
R ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR s c C sL t j d t j � j d | � j � } t j d t j � j d | � } | S( Nt ^s s +$t ( t ret compilet Mt subt strip( t text( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _indent s '!c C s� | j } x[ t | j | j � D]D \ } } t t | | � � | | j <t t | | � � | | j <q Wd j t | � � | d <d S( Ns
t imports(
R t zipt upgrade_ops_listt downgrade_ops_listR t _render_cmd_bodyt
upgrade_tokent downgrade_tokent joint sorted( R
t migration_scriptt
template_argsR t upgrade_opst
downgrade_ops( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _render_python_into_templatevars s c C s� t � } t | � } | j d � | j s; | j d � nA x> | j D]3 } t | | � } x | D] } | j | � qa WqE W| j d � | j � S( Ns; ### commands auto generated by Alembic - please adjust! ###t passs ### end Alembic commands ###( R R t writelineR t render_opt getvalue( t op_containerR
t buft printert opt linest line( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR 1 s
c C s+ t j | � } t j | | | � � } | S( N( t rendererst dispatchR t to_list( R
R. t rendererR/ ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR) I s c C s d j t | | � � S( Ns
( R R) ( R
R. ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt render_op_textO s c C s� | j } | j d t � } | j r� g } | r� | j � �_ | j d | j | j f � x- | j D]" } t | | � } | j | � qd W| j d � Wd QXn0 x- | j D]" } t | | � } | j | � q� W| Sd g Sd S( Nt render_as_batchs5 with op.batch_alter_table(%r, schema=%r) as batch_op:R R' (
t optst gett FalseR t
_within_batcht appendt
table_nameR R) t extend( R
R. R7 R6 R/ t t_opt t_lines( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _render_modify_tableS s$
c C sm | j � } g g | j D] } t | | � ^ q D] } | r2 | ^ q2 t g g | j D] } t | | � ^ qW D] } | d k rp | ^ qp � } t | � t k r� d d j | � d } n d j | � } d i t
| j � d 6t | � d 6| d 6} | j
r | d t
| j
� 7} n x> t | j � D]- } | d | j d
d � | j | f 7} q.W| d 7} | S(
Ns *[s ,
t ]s. %(prefix)screate_table(%(tablename)r,
%(args)st tablenamet prefixt argss ,
schema=%rs ,
%s=%rt t _s
)( t to_tablet columnst _render_columnR! t constraintst _render_constraintt Nonet lent MAX_PYTHON_ARGSR t _identR< R R t kwt replace( R
R. t tablet colt const rconsRD R t k( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt
_add_tablep s* )&
+
c C sX d i t | � d 6t | j � d 6} | j rJ | d t | j � 7} n | d 7} | S( Ns %(prefix)sdrop_table(%(tname)rRC t tnames , schema=%rt )( R RO R< R ( R
R. R ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _drop_table� s
c C s | j � } | j } | r$ d } n d } | i t | � d 6t | | j � d 6t | j j � d 6d j t | | � � d 6| j p� t
d 6| j j r� d t | j j � n d
d 6t | j
� r d d j g | j
j � D]% \ } } d | t | | � f ^ q� � n d
d
6} | S( NsL %(prefix)screate_index(%(name)r, [%(columns)s], unique=%(unique)r%(kwargs)s)sa %(prefix)screate_index(%(name)r, %(table)r, [%(columns)s], unique=%(unique)r%(schema)s%(kwargs)s)RC R RR s , RH t uniques , schema=%rR R s %s=%st kwargs( t to_indext
_has_batchR R R RO RR R t _get_index_rendered_expressionsR[ R9 R RM R\ t itemst _render_potential_expr( R
R. t indext has_batcht tmplt keyt valR ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt
_add_index� s
)bc C s | j } | r d } n d } | i t | � d 6t | | j � d 6t | j � d 6| j rp d t | j � n d d 6} | S( Ns %(prefix)sdrop_index(%(name)r)sC %(prefix)sdrop_index(%(name)r, table_name=%(table_name)r%(schema)s)RC R R< s , schema=%rR R ( R^ R R t
index_nameRO R< R ( R
R. Rc Rd R ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _drop_index� s
'c C s t | j � | t � g S( N( t _uq_constraintt
to_constraintt True( R
R. ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _add_unique_constraint� s c C s[ t t | | j � � g } | j sC | j t t | j � � � n | j t t | j � � t g | j D] } t | � ^ qh � t g | j
D] } t | � ^ q� � g � d d d d d d g } | j s� | j d d � n xS | D]K } | | j k r� | j | } | d k r2| j d | | f � q2q� q� Wd
i t | � d 6d j | � d
6S( Nt referent_schemat onupdatet ondeletet initiallyt
deferrablet use_alteri t
source_schemas %s=%rs& %(prefix)screate_foreign_key(%(args)s)RC s , RD ( t reprR t constraint_nameR^ R; RO t source_tableR= t referent_tablet
local_colst remote_colst insertRP RL R R ( R
R. RD RS R\ RV t value( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _add_fk_constraint� s, %,
!
c C s
t � � d S( N( t NotImplementedError( t
constraintR
( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _add_pk_constraint� s c C s
t � � d S( N( R~ ( R R
( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _add_check_constraint s c C s� | j r d } n d } | i t | � d 6t | | j � d 6t | j � d 6| j d 6| j rt d t | j � n d d 6} | S(
Ns3 %(prefix)sdrop_constraint(%(name)r, type_=%(type)r)sO %(prefix)sdrop_constraint(%(name)r, '%(table_name)s'%(schema)s, type_=%(type)r)RC R R< t types
, schema='%s'R R ( R^ R R Rv RO R< t constraint_typeR ( R
R. t templateR ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _drop_constraint s
'c C s� | j | j | j } } } | j r/ d } n# d } | rH | d 7} n | d 7} | i t | � d 6| d 6t | | � d 6| d 6} | S( Ns %(prefix)sadd_column(%(column)s)s* %(prefix)sadd_column(%(tname)r, %(column)ss , schema=%(schema)rRY RC RX t columnR ( R R< R� R^ R RI ( R
R. R RX R� R� R ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _add_column s
c C s� | j | j | j } } } | j r/ d } n# d } | rH | d 7} n | d 7} | i t | � d 6t | � d 6t | � d 6t | � d 6} | S( Ns %(prefix)sdrop_column(%(cname)r)s* %(prefix)sdrop_column(%(tname)r, %(cname)rs , schema=%(schema)rRY RC RX t cnameR ( R R< t column_nameR^ R RO ( R
R. R RX R� R� R ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _drop_column1 s
c C s� | j } | j } | j } | j } | j } | j } | j } | j } | j }
d } | j ri d } n d } | i t
| � d 6| d 6| d 6}
| d k r� |
d | t | | � f 7}
n | t
k r� t | | � } |
d | | f 7}
n | d k r |
d
| t | | � f 7}
n | d k rA|
d | | f 7}
n | d k rp| d k rp|
d | | f 7}
n | t
k r�| r�t | | � } |
d
| | f 7}
n |
r�| j r�|
d | |
f 7}
n |
d 7}
|
S( NRE i s %(prefix)salter_column(%(cname)rs+ %(prefix)salter_column(%(tname)r, %(cname)rRC RX R� s ,
%sexisting_type=%ss ,
%sserver_default=%ss ,
%stype_=%ss ,
%snullable=%rs ,
%sexisting_nullable=%rs ,
%sexisting_server_default=%ss
,
%sschema=%rRY s ( R< R� t modify_server_defaultt modify_typet modify_nullablet
existing_typet existing_nullablet existing_server_defaultR R^ R RL t
_repr_typeR9 t _render_server_default( R
R. RX R� t server_defaultt type_t nullableR� R� R� R t indentR� R t rendered( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt
_alter_columnG sZ
R c B s e Z d � Z d � Z RS( c C s | | _ | | _ d S( N( RC R ( t selfRC R ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt __init__� s c C s d | j t | j � f S( Ns %sf(%r)( RC RO R ( R� ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt __repr__� s ( t __name__t
__module__R� R� ( ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR ~ s c C s� | d k r | St j ry t | t j j � ry t j ri y | j d � SWqv t k
re t j
| � SXq� t j
| � Sn t | t j � r� | Sd S( s� produce a __repr__() object for a string identifier that may
use quoted_name() in SQLAlchemy 0.9 and greater.
The issue worked around here is that quoted_name() doesn't have
very good repr() behavior by itself when unicode is involved.
t asciiN( RL R t sqla_09R
R t elementst quoted_namet py2kt encodet UnicodeErrort text_typeR ( R ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyRO � s
c C s� t | t j � r� t j r4 t d i t d 6� } n i } | rI d } n d } | i t | � d 6t j | j d | j
| � � d 6St | � Sd S( Nt compile_kwargst
literal_bindss %(prefix)stext(%(sql)r)s %(sql)rRC t dialectR ( R
R t
ClauseElementR t sqla_08t dictRl t _sqlalchemy_autogenerate_prefixR� R R� Ru ( R| R
t wrap_in_textt
compile_kwR� ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyRa � s
c C s� t j r\ g | j D]E } t | t j � rI t t t | d d � � � n t
| | � ^ q Sg | j D]$ } t t t | d d � � � ^ qf Sd S( NR ( R R� t expressionsR
t sa_schemat ColumnRu RO t getattrRL Ra RH ( t idxR
t expRS ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR_ � s Pc C s g } | j } | j r7 | j d t | j � f � n | j r_ | j d t | j � f � n | r� | r� | j j r� | j d t | j j � f � n | r� | j r� | j d t | | j � f � n | r�t
t | | j � � g } | s| t
t | j j � � g 7} n | j t
g | j D] } t | j � ^ q(� � | j g | D] \ } } d | | f ^ qT� d i t
| � d 6d j | � d 6Sg | j D] } t
t | j � � ^ q�} | j g | D] \ } } d | | f ^ q�� d
i t | � d 6d j | � d 6Sd S( NRr Rq R R s %s=%rs, %(prefix)screate_unique_constraint(%(args)s)RC s , RD s$ %(prefix)sUniqueConstraint(%(args)s)( R^ Rr R; t strRq RR R RO R R Ru RH R= R R R� ( R R
t alterR7 Rc RD RS RV t v( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyRj � s6 ""20
+0
c C s, | j d } | d k r$ d | j S| Sd S( Nt user_module_prefixs %s.( R7 RL R� ( R
t targetRC ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _user_autogenerate_prefix� s
c C s | j d p d S( Nt sqlalchemy_module_prefixR ( R7 ( R
( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR� � s c C s" | j r
d S| j d p d Sd S( Ns batch_op.t alembic_module_prefixR ( R^ R7 ( R
( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR � s c C sN d | j k rJ | j d } | rJ | | | | � } | t k rG | SqJ n t S( Nt render_item( R7 R9 ( R� t object_R
t renderR� ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _user_defined_render� s
c C s t d | | � } | t k r" | Sg } | j rb t | j | � } | rb | j d | f � qb n | j s� | j d | j f � n | j d k r� | j d | j f � n d i t | � d 6t | j
� d 6t | j | � d 6d j
g | D] \ } } d
| | f ^ q� � d 6S( NR� R� t
autoincrementR� s, %(prefix)sColumn(%(name)r, %(type)s, %(kw)s)RC R R� s , s %s=%sRP ( R� R9 R� R� R; R� R� RL R� RO R R� R� R ( R� R
R� R7 t kwnameRf ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyRI s$
c C s� t d | | � } | t k r" | St | t j � rh t | j t j � rU | j } qh t | j | � Sn t | t � r� | r� t t
j d d | � � } n | S( NR� s ^'|'$R ( R� R9 R
R� t
DefaultClauset argR R Ra Ru R R ( t defaultR
t repr_R� ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR� # s c C s� t d | | � } | t k r" | St | � j } | j } | j d � r� t j d | � j d � } | d k r� | j
d | � n d | | f S| j d � r� t | � } d | | f St | | � } d | | f Sd S( NR� s sqlalchemy.dialectss sqlalchemy\.dialects\.(\w+)i s" from sqlalchemy.dialects import %ss %s.%rs sqlalchemy.s %s%r(
R� R9 R� R� R t
startswithR t matcht groupRL t addR� R� ( R� R
R� t modR t dnameRC ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyR� 4 s c C s t j | � } | | | � S( N( t _constraint_renderersR2 ( R R
R4 ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyRK K s c C s� t d | | � } | t k r" | S| j s/ d Sg } | j rf | j d t t | | j � � f � n d i t | � d 6d j g | j D] } t | j
� ^ q� g | D] \ } } d | | f ^ q� � d 6S( Nt primary_keyR s( %(prefix)sPrimaryKeyConstraint(%(args)s)RC s , s %s=%sRD ( R� R9 RH RL R R; Ru R R� R Re ( R R
R� R7 t cR� Rf ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _render_primary_keyP s
"c C s� | j � } | j d � } | d \ } } | d k r\ t | � d k r\ d | | f } n d j | d d !� } | j d k r� | j j d k r� | j j j } | | j k r� t | j | j
| j � } q� n d | | f } | S( so Implement a 'safe' version of ForeignKey._get_colspec() that
never tries to resolve the remote table.
t .i����i s %s.%si i����N( t _get_colspect splitRL RM R t parentRR t metadatat tablesRO R� R ( t fkt metadata_schemat colspect tokensRX t colnamet table_fullnamet parent_metadata( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _fk_colspecf s ! c C s� | j r( | j d t | j � f � n | j rP | j d t | j � f � n | j rx | j d t | j � f � n | j r� | j d t | j � f � n | j r� | j d t | j � f � n d S( NRo Rp Rq Rr Rs ( Ro R; Ru Rp Rq Rr Rs ( R R7 ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _populate_render_fk_opts� s c s� t d | | � } | t k r" | Sg } | j rY | j d t t | | j � � f � n t | | � | j j j � d i t
| � d 6d j d � | j D� � d 6d j � f d � | j D� � d 6d j g | D] \ } } d
| | f ^ q� � d 6S( Nt foreign_keyR sC %(prefix)sForeignKeyConstraint([%(cols)s], [%(refcols)s], %(args)s)RC s , c s s% | ] } d t | j j � Vq d S( s %rN( RO R� R ( t .0t f( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pys <genexpr>� s t colsc 3 s$ | ] } t t | � � � Vq d S( N( Ru R� ( R� R� ( t apply_metadata_schema( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pys <genexpr>� s t refcolss %s=%sRD (
R� R9 R R; Ru R R� R� R� R R� R R� ( R R
R� R7 R� Rf ( ( R� sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _render_foreign_key� s$
c C s2 t d | | � } | t k r" | St | | t � S( NR[ ( R� R9 Rj ( R R
R� ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _render_unique_constraint� s c C s� t d | | � } | t k r" | S| j rY t | j d � rY t | j j t j � rY d Sg } | j r� | j
d t t | | j � � f � n d i t
| � d 6| r� d d j d � | D� � n d d 6t | j | d
t �d 6S( Nt checkR� R s. %(prefix)sCheckConstraint(%(sqltext)s%(opts)s)RC s , c s s% | ] \ } } d | | f Vq d S( s %s=%sN( ( R� RV R� ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pys <genexpr>� s R R7 R� t sqltext( R� R9 t _create_rulet hasattrR
R� t sqltypest
TypeEngineRL R R; Ru R R� R Ra R� ( R R
R� R7 ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt _render_check_constraint� s(
*(T t
sqlalchemyR R� R R� R t
operationsR R R R t util.compatR R t
mako.pygenR R RN t sqlalchemy.sql.namingR R t ImportErrorR R& t
Dispatchert default_renderersR1 R R) R5 t dispatch_fort ModifyTableOpsR@ t
CreateTableOpRW t DropTableOpRZ t
CreateIndexOpRg t DropIndexOpRi t CreateUniqueConstraintOpRm t CreateForeignKeyOpR} t CreatePrimaryKeyOpR� t CreateCheckConstraintOpR� t DropConstraintOpR� t AddColumnOpR� t DropColumnOpR� t
AlterColumnOpR� t objectR RO Rl Ra R_ Rj R� R� R R� RI R� R� R� RK t PrimaryKeyConstraintR� R� R� t ForeignKeyConstraintR� t UniqueConstraintR� t CheckConstraintR� t branch( ( ( sL /opt/alt/python27/lib/python2.7/site-packages/alembic/autogenerate/render.pyt <module> sj
!(7
%
"
Mini Shell