Current File : //opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyo
�
�M!Vc @ s� d d l m Z m Z m Z m Z m Z m Z m Z d d l m Z d d l m
Z d d l m
Z
d d l m Z d d l m Z m Z d e f d � � YZ d
e f d � � YZ d S(
i����( t Tablet MetaDatat Indext selectt Columnt ForeignKeyConstraintt cast( t types( t schema( t OrderedDicti ( t util( t _columns_for_constraintt _is_type_boundt BatchOperationsImplc B s� e Z d � Z e d � � Z e d � � Z d � Z d � Z d � Z d � Z d � Z
d � Z d � Z d
� Z
d � Z d � Z d
� Z d � Z RS( c C s� t j s t d � � n | | _ | | _ | | _ | d k rN t d � � n | | _ | | _ | | _ t
| � | _ | | _ | | _
|
| _ g | _ d S( Ns. batch mode requires SQLAlchemy 0.8 or greater.t autot alwayst nevers4 recreate may be one of 'auto', 'always', or 'never'.( R s alwaysR ( R
t sqla_08t NotImplementedErrort
operationst
table_nameR t
ValueErrort recreatet copy_fromt
table_argst dictt table_kwargst reflect_argst reflect_kwargst naming_conventiont batch( t selfR R R R R R R R R R ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt __init__ s" c C s
| j j j S( N( R t implt dialect( R ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR" s c C s
| j j S( N( R R! ( R ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR! $ s c C s= | j d k r" | j j j | � S| j d k r5 t St Sd S( NR R ( R R R! t requires_recreate_in_batcht Truet False( R ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt _should_recreate( s
c C s7 | j � } | sT x | j D]1 \ } } } t | j j | � } | | | � q Wn� | j rr t d | j � } n t � } | j d k r� | j } n< t | j
| d | j d t d | j j
� | j | j �} t | | j | j � } x6 | j D]+ \ } } } t | | � } | | | � q� W| j | j � d S( NR R t autoloadt
autoload_with( R& R t getattrR R! R R R t NoneR R R R$ t get_bindR R t ApplyBatchImplR R t _create( R t should_recreatet opnamet argt kwt fnt m1t existing_tablet
batch_impl( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt flush0 s, c O s | j j d | | f � d S( Nt alter_column( R t append( R R0 R1 ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR7 O s c O s | j j d | | f � d S( Nt
add_column( R R8 ( R R0 R1 ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR9 R s c O s | j j d | | f � d S( Nt drop_column( R R8 ( R R0 R1 ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR: U s c C s | j j d | f i f � d S( Nt add_constraint( R R8 ( R t const( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR; X s c C s | j j d | f i f � d S( Nt drop_constraint( R R8 ( R R< ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR= [ s c O s | j j d | | f � d S( Nt rename_table( R R8 ( R R0 R1 ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR> ^ s c C s | j j d | f i f � d S( Nt create_index( R R8 ( R t idx( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR? a s c C s | j j d | f i f � d S( Nt
drop_index( R R8 ( R R@ ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRA d s c C s t d � � d S( Ns Can't create table in batch mode( R ( R t table( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt create_tableg s c C s t d � � d S( Ns Can't drop table in batch mode( R ( R RB ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt
drop_tablej s ( t __name__t
__module__R t propertyR" R! R& R6 R7 R9 R: R; R= R> R? RA RC RD ( ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR
s R, c B s� e Z d � Z d � Z d � Z d � Z d � Z d � Z d e d d d d � Z
d � Z d � Z d � Z
d
� Z d � Z d � Z d
� Z RS( c C sQ | | _ | | _ | | _ d | _ t d � | j j D� � | _ | j � d S( Nc s s% | ] } | j i | d 6f Vq d S( t exprN( t name( t .0t c( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pys <genexpr>u s ( RB R R R* t new_tableR RK t column_transferst _grab_table_elements( R RB R R ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR o s c C s9 | j j } t � | _ xF | j j D]8 } | j d | � } t | _ | _ | | j | j <q% Wi | _
g | _ i | _ i | _
xR | j j D]D } t | � r� q� n | j r� | | j
| j <q� | j j | � q� Wx$ | j j D] } | | j | j <q� Wx1 | j j D]# } | j j | | j j | � qWd S( NR ( RB R R t columnsRK t copyR% t uniquet indexRI t named_constraintst unnamed_constraintst indexest new_indexest constraintsR R8 t kwargsR t
setdefault( R R RK t c_copyR< R@ t k( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRN y s( c C s t � } | j j } t d | d | t | j j � � t | j � | j �| _ } x� t | j
j � � | j D]� } t g t
| � D] } | j ^ q� � } | j | j � s� qs n | j d | d | � } t | t � r� | j | | � n | j | � qs Wd S( Nt _alembic_batch_tempR t target_table( R RB R R t listRO t valuesR R RL RS RT t setR t keyt issubsetRM RP t
isinstanceR t _setup_referentt append_constraint( R t mR RL R< RK t
const_columnst
const_copy( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt _transfer_elements_to_new_table� s "%c
C s� g } | j | j j � � xe | j j � D]T } | j t | j d | j g | j j � D] } | j
j | ^ qZ | j �� q, W| S( NRQ (
t extendRU R_ RV R8 R RI RQ RO t keysRL RK RX ( R R@ RR t col( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt _gather_indexes_from_both_tables� s )c
C sN | j d j � } | j d � } | d } t | � d k rK | d } n d } | d k rJt j | | � } | | j k r� | j | } x� | j D]M } | j � j d � d }
| j j |
� s� | j
t |
t j
� � q� q� WqJt | | d | g g | j D] } | j � j d � d ^ qD] } t | t j
� ^ q(�n d S( Ni t .i����i R\ i����R ( t elementst _get_colspect splitt lenR* t
sql_schemat _get_table_keyt tablesRK t contains_columnt
append_columnR t sqltypest NULLTYPER ( R t metadatat
constraintt spect partst tnamet referent_schemaRa t tt elemt colnamet n( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRd � s$
c C s> | j � | j | j � | j | j � y� | j | j j d t � j t d � | j
j � D� � t g | j
j
� D] } d | k rw | d ^ qw � � � | j | j � Wn | j | j � � nm X| j d | j j d | j j �| j j | j _ z( x! | j � D] } | j | � qWWd d | j _ Xd S( Nt inlinec s s' | ] \ } } d | k r | Vq d S( RH N( ( RJ R[ t transfer( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pys <genexpr>� s RH R\ R ( Ri t prep_table_for_batchRB RC RL t _exect insertR$ t from_selectR^ RM t itemsR R_ RD R> RI R Rm R? ( R t op_implR� R@ ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR- � s2
#
c K s� | j | } | j | }
| d k rH | | k rH | | _ | |
d <n | d k r� t j | � } | | _ t |
d | � |
d <n | d k r� | | _ n | t k r� | | _
n | d k r� t | � | _ n d S( NRI RH (
RO RM R* RI Rx t to_instancet typeR t nullableR% t server_defaultt boolt
autoincrement( R R t column_nameR� R� RI t type_R� R1 t existingt existing_transfer( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR7 � s
c K s6 | j d | j j � | j | j <i | j | j <d S( NR ( RP RB R RO RI RM ( R R t columnR1 ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR9 s "c K s | j | j =| j | j =d S( N( RO RI RM ( R R R� R1 ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR: s
c C so | j s t d � � n t | t j � r[ | j j | j k r[ | j j | j j � q[ n | | j | j <d S( Ns Constraint must have a name(
RI R Rc Rs t PrimaryKeyConstraintRB t primary_keyRT t removeRS ( R R< ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR; s c C sT | j s t d � � n y | j | j =Wn$ t k
rO t d | j � � n Xd S( Ns Constraint must have a names No such constraint: '%s'( RI R RS t KeyError( R R< ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR= # s
c C s | | j | j <d S( N( RV RI ( R R@ ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR? + s c C s< y | j | j =Wn$ t k
r7 t d | j � � n Xd S( Ns No such index: '%s'( RU RI R� R ( R R@ ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRA . s
c O s t d � � d S( Nt TODO( R ( R R0 R1 ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR> 4 s N( RE RF R RN Ri Rm Rd R- R* R% R7 R9 R: R; R= R? RA R> ( ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR, n s$
$ N( t
sqlalchemyR R R R R R R R Rx R Rs t sqlalchemy.utilR t R
t util.sqla_compatR R t objectR
R, ( ( ( sI /opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt <module> s 4d
Mini Shell