�
�M!Vc
@s
dd
lZddl
mZ
ddlmZ
ddlmZ
ddlmZ
d�Z	dd	�
Z
d
eed
ed
d
d
d
d�	Z
d
d
d
d�Zed
d
�Zed
d�Zd�Zd
ed�Zeed�Zed�
Zeed�Zed
d�Zd�Zd
S(i����Ni
(
tScriptDirectory(
tEnvironmentContext(
tutil(
tautogeneratec
Cs�|jd
�

xit
j|j��
D]R}
tt
jj|j�|
d��
�}t|�
}WdQX|jd|
|�
q#W|jd�

|jd�

dS(sList available templatessAvailable templates:
tREADMENs%s - %ss1
Templates are used via the 'init' command, e.g.:s,
  alembic init --template generic ./scripts(tprint_stdouttostlistdirtget_template_directorytopentpathtjointnext(tconfigttempnametreadmetsynopsis((s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytlist_templates	s



	





tgenericc
Cs�
tj
|
tj�r+tjd
|
�
�
ntjj|j�|�}tj
|tj�sqtjd|�
�
ntjdtjj	|
�
tj
|
�
tjj|
d�}tjdtjj	|�
tj
|�
t|
�
}x�tj|�
D]�}tjj||�}|dkrl
tjj	|j
�
}tj
|tj�rS
tjd|�

q�
|j||d|
�

q�tjj|�
r�tjj|
|�}	|j||	�
q�q�Wtjd|�

d	S(
s#Initialize a new scripts directory.sDirectory %s already existssNo such template %rsCreating directory %stversionssalembic.ini.makos File %s already exists, skippingtscript_locationsNPlease edit configuration/connection/logging settings in %r before proceeding.N(RtaccesstF_OKRtCommandErrorR
RRtstatustabspathtmakedirsRRtconfig_file_nametmsgt_generate_templatetisfilet
_copy_file(
R
t	directoryttemplatettemplate_dirRtscripttfile_t	file_pathtconfig_filetoutput_file((s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytinits:

	



















	
theadc
sf
tj
|�
}
td
|
d|d|d|d|d|d|d|d	|	�	}tj||
|��tj|jd
�
�
}|r�t}|r�tj	d�
�
n�f
d�}
n0|r��f
d
�}
n|r�tj	d�
�
n|r%
t
||
d|
d|d�jd���
|
j�
WdQXng�j
�D]}|^q2
}t|�
dkr^
|dS|SdS(sCreate a new revision file.tmessageRtsqlR)tsplicetbranch_labeltversion_pathtrev_idt
depends_ontrevision_environments7Using --sql with --autogenerate does not make any sensecs�j||
�
gS(
N(
trun_autogenerate(trevtcontext(
trevision_context(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytretrieve_migrations`s

cs�j||
�
gS(
N(
trun_no_autogenerate(R3R4(
R5(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyR6ds

siUsing --sql with the revision command when revision_environment is not configured does not make any sensetfntas_sqlt
template_argsR5Ni
i(Rtfrom_configtdicttautogentRevisionContextRtasbooltget_main_optiontTrueRR
R:trun_envtgenerate_scriptstlen(R
R*RR+R)R,R-R.R/R0tscript_directorytcommand_argstenvironmentR6R#tscripts((
R5s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytrevisionDsF

















	



cCsMtj
|�
}i
|d
6}|j|p1tj�|dtd|
d||�S(s�Merge two revisions together.  Creates a new migration file.

    .. versionadded:: 0.7.0

    .. seealso::

        :ref:`branches`

    R
trefreshR)t
branch_labels(RR;tgenerate_revisionRR/RA(R
t	revisionsR*R-R/R#R:((s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytmerge�s




	
c
s�tj
|�
�
d
}d
�krT|s9tjd�
�
n�jd
d�\}�n��
fd�}t|�
d|d|d|d�d	|��
�
j�
Wd
QXd
S(sUpgrade to a later version.t
:sRange revision not allowedics�
j�|�S(
N(
t
_upgrade_revs(R3R4(RIR#(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytupgrade�s
R8R9tstarting_revtdestination_revttagN(RR;tNoneRRtsplitR
RB(R
RIR+RTRRRQ((RIR#s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyRQ�s 











c
s�tj
|�
�
d}d
�krT|s9tjd�
�
n�jd
d�\}�n|rltjd�
�
n��
fd�}t|�
d|d|d|d	�d
|��
�
j�
WdQXdS(sRevert to a previous version.ROsRange revision not allowedis/downgrade with --sql requires <fromrev>:<torev>cs�
j�|�S(
N(
t_downgrade_revs(R3R4(RIR#(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyt	downgrade�s
R8R9RRRSRTN(RR;RURRRVR
RB(R
RIR+RTRRRX((RIR#s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyRX�s&















cs�tj
��
�
|
d
krV��
fd�}t��
d|�
�
�
j�
WdQXn*x'�
j|
�
D]}�j|j�

qfWdS(s1Show the revision(s) denoted by the given symbol.tcurrentcs.x'�
j|�
D]}�j
|j�

qWgS(
N(t
get_revisionsRt	log_entry(R3R4tsc(R
R#(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytshow_current�s


R8N(RR;R
RBRZRR[(R
R3R]R\((R
R#s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytshow�s





cs�tj
|�
}|
dk	rWd
|
kr9tjd�
�
n|
j�jd
�
\}}n
d}}�
f
d��dd�f
d�}|dkr�|||d|�

n5|dkr�|||d|�

n�||||�
dS(	s.List changeset scripts in chronological order.ROs9History range requires [start]:[end], [start]:, or :[end]c
s^xW|
jd
|pd
d|p!d�D]1}|j
|jd�dtdtdt��

q%WdS(NtbaseR)theadstverbosetinclude_branchestinclude_doctinclude_parents(twalk_revisionsRt
cmd_formatRA(R
R#R_R)R\(
Ra(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyt_display_history�s




	

csE���
��fd
�}t�
�d|�
�
�j
�
WdQXdS(NcsH�dkr"����
|�
n"�
dkrD���|��
ngS(
N(
RU(R3R4(RgR_R
R)R#(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyt_display_current_history�s





R8(R
RB(R
R#R_R)Rh(
Rg(R_R
R)R#s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyt_display_history_w_current�s




RYR)R_N(RR;RURRtstripRV(R
t	rev_rangeRaR#R_R)Ri((RgRas@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pythistory�s




	


cCsstj
|�
}|r'|jd
�
}n|j|j��
}x0|D](}|j|j|
dtdt�
�

qCWdS(s4Show current available heads in the script directoryR`Rbttree_indicatorsN(RR;RZt	get_headsRRfRAtFalse(R
Ratresolve_dependenciesR#R`R3((s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyR`
s





cs�tj
|�
�
xq�
j�D]c��jr|jd
�j�dt�

dj��fd��
f
d��jD�
D�
�
�
qqWdS(sShow current branch pointss%s
%s
Rbs

c
	3sG|]=}
dd
tt
�j�
�
|
jtdtd�
�
fV
qdS(s%s -> %st
 RbRcN(RDtstrRIRfRoRA(t.0trev_obj(R\Ra(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pys	<genexpr>'
sc
3s|]}
�j|
�
V
qdS(
N(
tget_revision(RsR3(
R#(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pys	<genexpr>,
sN(	RR;Retis_branch_pointRRfRARtnextrev(R
Ra((R\R#Ras@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytbranches
s

	




csdtj
��
�
|r%tjd
�

n��
�fd�}t��
d|�
�
�
j�
WdQXdS(s,Display the current revision for a database.s--head-only is deprecatedcs_�r+�jd
t
j|
jjj�
�
nx-�
j|�
D]}�j|j��
�

q;WgS(NsCurrent revision(s) for %s:(RRtobfuscate_url_pwt
connectiontengineturlRZRf(R3R4(R
R#Ra(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytdisplay_version9
s





R8N(RR;RtwarnR
RB(R
Rat	head_onlyR}((R
R#Ras@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyRY1
s





c
s�tj
|�
�
d
}d
�krT|s9tjd�
�
n�jd
d�\}�n��
fd�}t|�
d|d|d�d|d	|��
�
j�
Wd
QXd
S(sQ'stamp' the revision table with the given revision; don't
    run any migrations.ROsRange revision not allowedics�
j�|�S(
N(
t_stamp_revs(R3R4(RIR#(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytdo_stampW
s
R8R9RSRRRTN(RR;RURRRVR
RB(R
RIR+RTRRR�((RIR#s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytstampK
s 











cs�tj
|�
�|
d
krS�f
d�}t|�d|�
�
�j�
WdQXnL�j|
�
}|s~tjd|
�
�
nx|D]}tj|j�

q�WdS(s%Edit revision script(s) using $EDITORRYcsF|stj
d
�
�
nx'�j|�
D]}tj|j�

q(WgS(NsNo current revisions(RRRZteditR
(R3R4R\(
R#(s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pytedit_currentl
s





R8Ns*No revision files indicated by symbol '%s'(	RR;R
RBRZRRR�R
(R
R3R�trevsR\((
R#s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyR�f
s










(RR#Rtruntime.environmentR
tRRR=RR(RURoRIRNRQRXR^RlR`RxRYR�R�(((s@/opt/alt/python27/lib/python2.7/site-packages/alembic/command.pyt<module>
s&


	,	
	
:	-