�
�M!Vc@s�dd
lm
Z
mZmZmZmZmZmZ
ddlmZ	
ddlm
Z
ddlm
Z

ddlmZ
ddlmZmZ
def
d	��YZd
ef
d��YZdS(
i����(tTabletMetaDatatIndextselecttColumntForeignKeyConstrainttcast(
ttypes(
tschema(
tOrderedDicti(
tutil(t_columns_for_constraintt_is_type_boundtBatchOperationsImplcBs�eZ
d�Zed
��
Zed��
Zd�Zd�Zd�Zd�Z	d�Z
d�Zd	�Zd
�Z
d�Zd�Zd
�Zd�ZRS(cCs�tj
std
�
�
n|
|_||_||_|dkrNtd�
�
n||_||_||_	t
|�
|_||_|	|_
|
|_g|_dS(Ns.batch mode requires SQLAlchemy 0.8 or greater.tautotalwaystnevers4recreate may be one of 'auto', 'always', or 'never'.(RsalwaysR(R
tsqla_08tNotImplementedErrort
operationst
table_nameRt
ValueErrortrecreatet	copy_fromt
table_argstdictttable_kwargstreflect_argstreflect_kwargstnaming_conventiontbatch(tselfRRRRRRRRRR((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt__init__s"	


	
	
	



	
	
	

	
	
	
c


Cs
|jj
jS(
N(Rtimpltdialect(
R((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR" sc


Cs
|jj
S(
N(RR!(
R((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR!$sc

Cs=|jd
kr"|j
jj|�
S|jdkr5tStSdS(NRR(RRR!trequires_recreate_in_batchtTruetFalse(
R((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt_should_recreate(s




c
	Cs7
|j�}
|
sTx
|j
D]1\}}}t|jj|�}|||�
qWn�|jrrtd
|j�
}n	t�}|jdk	r�|j}n<t	|j
|d|jdtd|jj
�|j|j�}t||j|j�}x6|j
D]+\}}}t||�}|||�
q�W|j|j�

dS(NRRtautoloadt
autoload_with(R&RtgetattrRR!RR
RtNoneRRRR$tget_bindRRtApplyBatchImplRRt_create(	Rtshould_recreatetopnametargtkwtfntm1texisting_tablet
batch_impl((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pytflush0s,



	
	


	






c
Os|jj
d
|
|f�

dS(Ntalter_column(Rtappend(RR0R1((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR7Os
c
Os|jj
d
|
|f�

dS(Nt
add_column(RR8(RR0R1((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR9Rs
c
Os|jj
d
|
|f�

dS(Ntdrop_column(RR8(RR0R1((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR:Us
cCs |jj
d
|
f
if�

dS(Ntadd_constraint(RR8(Rtconst((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR;Xs
cCs |jj
d
|
f
if�

dS(Ntdrop_constraint(RR8(RR<((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR=[s
c
Os|jj
d
|
|f�

dS(Ntrename_table(RR8(RR0R1((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR>^s
cCs |jj
d
|
f
if�

dS(Ntcreate_index(RR8(Rtidx((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR?as
cCs |jj
d
|
f
if�

dS(Nt
drop_index(RR8(RR@((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRAds
cCstd
�
�
dS(Ns Can't create table in batch mode(
R(Rttable((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pytcreate_tablegs
cCstd
�
�
dS(NsCan't drop table in batch mode(
R(RRB((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt
drop_tablejs
(t__name__t
__module__R tpropertyR"R!R&R6R7R9R:R;R=R>R?RARCRD(((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR

s
												R,cBs�eZ
d�Zd
�Zd�Zd�Zd�Zd�Zde	dddd�Z
d�Zd�Zd	�Z
d
�Zd�Zd�Zd
�ZRS(cCsQ|
|_||_
||_d|_td
�|jjD�
�
|_|j�
dS(Nc
ss%|]}
|
ji
|
d6fV
qd
S(texprN(
tname(t.0t
c((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pys	<genexpr>us(	RBRRR*t	new_tableR	RKtcolumn_transferst_grab_table_elements(RRBRR((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR os
	
	
	
	

c
Cs9
|jj
}
t�|_xF|jjD]8}|jd
|
�
}t|_|_||j|j	<q%Wi|_
g|_i|_i|_
xR|jjD]D}t|�
r�q�n|j	r�||j
|j	<q�|jj|�

q�Wx$|jjD]}||j|j	<q�Wx1|jjD]#}|jj||jj|�
q
WdS(NR(RBRR	tcolumnsRKtcopyR%tuniquetindexRItnamed_constraintstunnamed_constraintstindexestnew_indexestconstraintsRR8tkwargsRt
setdefault(RRRKtc_copyR<R@t
k((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRNys(






	
	
	
	



	


c
Cs#
|jdkstd
�
�
t�}
|jj}td|
d|t|jj	��
t|j
�
|j�
|_}x�t|jj	��
|j
D]�}tgt|�
D]}|j^q��
}|j|j�
s�q�n|jd|d|�}t|t�r
|j|
|�
n|j|�

q�WdS(NsCan only create new table oncet_alembic_batch_tempRttarget_table(RLR*tAssertionErrorR
RBRRtlistROtvaluesRRRSRTtsetRtkeytissubsetRMRPt
isinstanceRt_setup_referenttappend_constraint(Rt
mRRLR<RKt
const_columnst
const_copy((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt_transfer_elements_to_new_table�s"
	

	"


%




c

Cs�g}
|
j|j
j��

xe|jj�D]T}|
jt|jd
|jg|jj	�D]}|j
j|^qZ|j�

�

q,W|
S(NRQ(
textendRUR`RVR8RRIRQROtkeysRLRKRX(RR@RRtcol((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt _gather_indexes_from_both_tables�s





	

)
c
CsN
|jd
j
�}|jd�
}|d}t|�
dkrK|d
}nd}|dkrJ
tj||�}||
jkr�|
j|}x�|jD]M}	|	j
�jd�
d}
|jj	|
�
s�|j
t|
tj
��

q�q�WqJ
t||
d|gg|jD]}	|	j
�jd�
d^q
D]}t|tj
�^q(
�

ndS(Nit
.i����iR\i����R(telementst_get_colspectsplittlenR*t
sql_schemat_get_table_keyttablesRKtcontains_columnt
append_columnRtsqltypestNULLTYPER(Rtmetadatat
constrainttspectpartsttnametreferent_schemaRbt
ttelemtcolnamet
n((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRe�s$

















	cCs>
|j�
|
j
|j�

|
j|j�

y�|
j|jjd
t�
jt	d�|j
j�D�
�
tg|j
j
�D]}d|krw|d^qw�
��

|
j|j�

Wn


|
j|j�

�nmX|
jd|jjd|jj�

|jj|j_z(x!|j�D]}|
j|�

q
WWdd|j_XdS(Ntinlinec
ss'|]\}
}d|kr|
V
qd
S(RHN((RJR[ttransfer((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pys	<genexpr>�s	
RHR\R(Rjtprep_table_for_batchRBRCRLt_exectinsertR$tfrom_selectR_RMtitemsRR`RDR>RIRRnR?(Rtop_implR�R@((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR-�s2





	




#









cKs�|j|}	|j
|}
|dk	rH||krH||	_||
d
<n|dk	r�tj|�
}||	_t|
d|�|
d<n|dk	r�||	_n|t	k	r�||	_
n|dk	r�t|�
|	_ndS(NRIRH(
RORMR*RIRytto_instancettypeRtnullableR%tserver_defaulttboolt
autoincrement(RRtcolumn_nameR�R�RIttype_R�R1texistingtexisting_transfer((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR7�s



	



	






cKs6|jd
|j
j�
|j|j<i|j|j<dS(NR(RPRBRRORIRM(RRtcolumnR1((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR9
s"
cKs|j|j
=|j|j
=dS(
N(RORIRM(RRR�R1((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR:
s


cCso|
jst
d
�
�
nt|
tj�r[|jj|jkr[|jj|jj�

q[n|
|j	|
j<dS(NsConstraint must have a name(
RIRRdRttPrimaryKeyConstraintRBtprimary_keyRTtremoveRS(RR<((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR;
s
	



cCsT|
jst
d
�
�
ny|j|
j=Wn$tk
rO


t
d|
j�
�
n
XdS(NsConstraint must have a namesNo such constraint: '%s'(RIRRStKeyError(RR<((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR=#
s
	





cCs|
|j|
j
<dS(
N(RVRI(RR@((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR?+
s
cCs<y|j|
j
=Wn$tk
r7


td
|
j
�
�
n
XdS(NsNo such index: '%s'(RURIR�R(RR@((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyRA.
s




c
Ostd
�
�
dS(NtTODO(
R(RR0R1((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR>4
s
N(RERFR RNRjRnReR-R*R%R7R9R:R;R=R?RAR>(((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyR,ns$
	
			
		$



							N(t
sqlalchemyRR
RRRRRRRyRRttsqlalchemy.utilR	tR
tutil.sqla_compatRRtobjectR
R,(((sI/opt/alt/python27/lib/python2.7/site-packages/alembic/operations/batch.pyt<module>
s4



d