RPOS \ d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ ddlm Z
dd lm Z dd
l m
Z
dd lmZ erd d lmZ d d
lmZ ddl mZ d@dZ dAdBdZ dCdDd)ZdEd+Z dFdGd.Z dHdId1Z dHdId2Zd3 Z dJdKd7Z dLd8Z dMd9Z dMdNd:Z dOdPd<Z!dQd>Z"dMdRd?Z#dS )S ) annotationsN)List)Optional)
TYPE_CHECKING)Union ) autogenerate)util)EnvironmentContext)ScriptDirectory)Config)Script)ProcessRevisionDirectiveFnconfigr
c | d t j | D ]}t t j | |d 5 }t | }ddd n # 1 swxY w Y | d|| | d | d dS )zNList available templates.
:param config: a :class:`.Config` object.
zAvailable templates:
READMENz%s - %sz1
Templates are used via the 'init' command, e.g.:z,
alembic init --template generic ./scripts) print_stdoutoslistdirget_template_directoryopenpathjoinnextrstrip)r tempnamereadmesynopsiss a/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/alembic/command.pylist_templatesr s& 0111Jv < < > >?? ; ;
GLL6688(H M M
-
F||**,,H - - - - - - - - - - - - - - - Ix::::
LMMM
GHHHHHs >"B,,B0 3B0 genericF directorystrtemplatepackageboolreturnNonec t j |t j r+t j | rt j d|z t j | | }t j |t j st j d|z t j |t j sct j dt j
| fi | j 5 t j | ddd n # 1 swxY w Y t j |d }t j dt j
| fi | j 5 t j | ddd n # 1 swxY w Y t | }d}t j | D ]}t j || } |dk r| j J t j
| j }t j |t j r t j d|dfi | j | | || t j | r6t j || }
| | |
|rt j t j
| d
t j t j
| d
fD ]Y} t j di | j 5 t' | d 5 ddd n # 1 swxY w Y ddd n # 1 swxY w Y Z|J t j d
|dfi | j dS )a Initialize a new scripts directory.
:param config: a :class:`.Config` object.
:param directory: string path of the target directory
:param template: string name of the migration environment template to
use.
:param package: when True, write ``__init__.py`` files into the
environment location as well as the versions/ location.
.. versionadded:: 1.2
z,Directory %s already exists and is not emptyzNo such template %rzCreating directory Nversionszalembic.ini.makozFile z already exists, skipping)script_locationz __init__.pyAdding {path!r}wz9Please edit configuration/connection/logging settings in z before proceeding.)r, )r accessF_OKr r
CommandErrorr r r statusabspathmessaging_optsmakedirsr config_file_namemsg_generate_templateisfile
_copy_filer ) r r" r$ r% template_dirr* script config_filefile_ file_path output_filer s r initr@ ' s .
yBG$$
I)>)>
:Y F
7<< = = ? ?JJL
9\27 + + B 5 @AAA
9Y ( ( #
[ @"'//)"<"< @ @
#
# #
K " " " # # # # # # # # # # # # # # # w||Iz22H
;bgooh 7 7;;
H Y
'
'F "KL)) 6 6GLL u55 & & &*666 '//&*ABBKybg..
DKDDD +
)) {I * W^^I
&
& 6 ',,y%88K i 5 5 5 GLL 33] C C GLL 22M B B
D HH&2GHH
$__
" " "H :" : : :
sZ D**D.1D. F--F14F10N&N N&N N&N N&&N* -N* headmessage
Optional[str]r sqlsplice branch_label version_pathrev_id
depends_onprocess_revision_directives$Optional[ProcessRevisionDirectiveFn]/Union[Optional[Script], List[Optional[Script]]]c \ t j | } t ||||||||| } t j | | | |
t j | d }
|r d}
|rt j d fd}n |
rfd}n|rt j d |
rFt | | ||j
5 | d
d
d
n # 1 swxY w Y d D }t | d k r|d
S |S )a Create a new revision file.
:param config: a :class:`.Config` object.
:param message: string message to apply to the revision; this is the
``-m`` option to ``alembic revision``.
:param autogenerate: whether or not to autogenerate the script from
the database; this is the ``--autogenerate`` option to
``alembic revision``.
:param sql: whether to dump the script out as a SQL string; when specified,
the script is dumped to stdout. This is the ``--sql`` option to
``alembic revision``.
:param head: head revision to build the new revision upon as a parent;
this is the ``--head`` option to ``alembic revision``.
:param splice: whether or not the new revision should be made into a
new head of its own; is required when the given ``head`` is not itself
a head. This is the ``--splice`` option to ``alembic revision``.
:param branch_label: string label to apply to the branch; this is the
``--branch-label`` option to ``alembic revision``.
:param version_path: string symbol identifying a specific version path
from the configuration; this is the ``--version-path`` option to
``alembic revision``.
:param rev_id: optional revision identifier to use instead of having
one generated; this is the ``--rev-id`` option to ``alembic revision``.
:param depends_on: optional list of "depends on" identifiers; this is the
``--depends-on`` option to ``alembic revision``.
:param process_revision_directives: this is a callable that takes the
same form as the callable described at
:paramref:`.EnvironmentContext.configure.process_revision_directives`;
will be applied to the structure generated by the revision process
where it can be altered programmatically. Note that unlike all
the other parameters, this option is only available via programmatic
use of :func:`.command.revision`
rB r rD rA rE rF rG rH rI )rJ revision_environmentTz7Using --sql with --autogenerate does not make any sensec 4 | | g S Nrun_autogeneraterevcontextrevision_contexts r retrieve_migrationsz%revision..retrieve_migrations s - -c7 ; ; ;I c 4 | | g S rQ )run_no_autogeneraterT s r rX z%revision..retrieve_migrations s 0 0g > > >IrY ziUsing --sql with the revision command when revision_environment is not configured does not make any sensefnas_sql
template_argsrW Nc g | ]}|S ra ).0r; s r
z revision.. s HHH&vHHHrY r r )r from_configdictautogenRevisionContextr
asboolget_main_optionr0 r r_ run_envgenerate_scriptslen)r rB r rD rA rE rF rG rH rI rJ script_directory command_args environmentrX scriptsrW s @r revisionrq { s t '26::!
!!
L .$? +f 4 45K L LMMK
#I
M
'
"*8 -
' '
$ $ & & & ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' IH$4$E$E$G$GHHHG
7||qqzs C((C,/C,'Config'c t j | }t ddddddddd }t j | || fd}t | ||dj 5 | ddd n # 1 swxY w Y j d }|j
}|rt j d | |
d
dS ) zCheck if revision command with autogenerate has pending upgrade ops.
:param config: a :class:`.Config` object.
.. versionadded:: 1.9.0
NTFrA rN c 4 | | g S rQ rR rT s r rX z"check..retrieve_migrations s ))#w777 rY r\ z!New upgrade operations detected: z#No new upgrade operations detected.)r rd re rf rg r r_ rj generated_revisions upgrade_opsas_diffsr
AutogenerateDiffsDetectedr )r rm rn rX migration_scriptdiffsrW s @r checkr| s{ '26::
L .
&4)
# # """# # # # # # # # # # # # # # # (;B? ( 1 1 3 3E C, 7 7 7
ABBBBBs #BB B revisionsOptional[Script]c ~ t j | }d| i} |j |pt j |fd||d|S )a{ Merge two revisions together. Creates a new migration file.
:param config: a :class:`.Config` instance
:param message: string message to apply to the revision
:param branch_label: string label name to apply to the new revision
:param rev_id: hardcoded revision identifier instead of generating a new
one.
.. seealso::
:ref:`branches`
r T)refreshrA
branch_labels)r rd generate_revisionr
rH )r r} rB rF rH r; r_ s r merger 2 si 0
(
0
0F&M $6 # $+--
"
rY rq tagc . t j | d}dv r/|st j d dd \ }fd}t | |||| 5 ddd dS # 1 swxY w Y dS )ad Upgrade to a later version.
:param config: a :class:`.Config` instance.
:param revision: string revision target or range for --sql mode
:param sql: if True, use ``--sql`` mode
:param tag: an arbitrary "tag" that can be intercepted by custom
``env.py`` scripts via the :meth:`.EnvironmentContext.get_tag_argument`
method.
N:Range revision not allowed c 0 | S rQ )
_upgrade_revsrU rV rq r; s r upgradezupgrade..upgradeu s ##Hc222rY r] r^ starting_revdestination_revr r rd r
r0 splitr rj )r rq rD r r r r; s ` @r r r Y s (
(
0
0FL
h B#$@AA A!) Q!7!7 h3 3 3 3 3 3
!
s (B
BBc \ t j | d}dv r0|st j d dd \ }n|rt j d fd}t | |||| 5 ddd dS # 1 swxY w Y dS )af Revert to a previous version.
:param config: a :class:`.Config` instance.
:param revision: string revision target or range for --sql mode
:param sql: if True, use ``--sql`` mode
:param tag: an arbitrary "tag" that can be intercepted by custom
``env.py`` scripts via the :meth:`.EnvironmentContext.get_tag_argument`
method.
Nr r r z/downgrade with --sql requires :c 0 | S rQ )_downgrade_revsr s r downgradez downgrade..downgrade s %%h444rY r r )r rq rD r r r r; s ` @r r r s9 (
(
0
0FL
h B#$@AA A!) Q!7!7 hh
=
5 5 5 5 5 5
!
s ?B!!B%(B%c 0 t j |dk rG fd}t | 5 ddd dS # 1 swxY w Y dS | D ] } |j dS )zShow the revision(s) denoted by the given symbol.
:param config: a :class:`.Config` instance.
:param revision: string revision target
currentc l | D ] } |j g S rQ )
get_revisionsr log_entry)rU rV scr r; s r show_currentzshow..show_current s> **3//
2
2##BL1111IrY r] N)r rd r rj r r r )r rU r r r; s ` @r showr s
(
0
0F
i
<
@
@
@ NN &&s++ . .B - - - - . .s AA A rev_rangeverboseindicate_currentc t j | }|Cd|vrt j d | d \ }}ndx}}t j | d p}d fd fd}|dk s|dk s|r || ||| dS | ||| dS )
zList changeset scripts in chronological order.
:param config: a :class:`.Config` instance.
:param rev_range: string revision range
:param verbose: output in verbose mode.
:param indicate_current: indicate current revision.
Nr z9History range requires [start]:[end], [start]:, or :[end]rO ra c | |pd|pd D ]>}r|j |v |_ | | ddd ?d S )Nbaseheads)r rA T)r include_branches include_docinclude_parents)walk_revisionsrq _db_current_indicatorr
cmd_format)r r; r rA currentsr r r s r _display_historyz!history.._display_history s '' dog (
B
C+-;(+B(
#%) $$(
rY c fd}t | 5 d d d d S # 1 swxY w Y d S )Nc ~ dk r | | n%dk r | | n | g S )Nr ra )rU rV r r r rA r; s r _display_current_historyzMhistory.._display_history_w_current.._display_current_history so y sC@@@@"" dC@@@@ tSAAAIrY r )r rj )r r; r rA r r s ```` r _display_history_w_currentz+history.._display_history_w_current s 3K
L
L
L NN s AA
Ar )ra )r rd r
r0 stripr rh ri )
r r r r r; r rA ro r r s
`` @r historyr s, &
(
0
0F i #N
__&&,,S11
ddt
F**+ABBCC
" yDI - - -""664>>>>>t44444rY c t j | }|r| d }n'| | }|D ]-}| | |dd .dS )zShow current available heads in the script directory.
:param config: a :class:`.Config` instance.
:param verbose: output in verbose mode.
:param resolve_dependencies: treat dependency version as down revisions.
r TF)r tree_indicatorsN)r rd r get_headsr r )r r resolve_dependenciesr; r rU s r r r s
(
0
0F 9$$W--$$V%5%5%7%788
NN$
rY c
* t j | D ]fj r]| d d d fdfdj D D gdS )zShow current branch points.
:param config: a :class:`.Config` instance.
:param verbose: output in verbose mode.
z%s
%s
T)r
c 3 K | ]C}d t t j z d| dd V DdS ) z -> FT)r r N)rl r# rq r )rb rev_objr r s r zbranches..8 s c#bk"2"2 3 3333 ** !Dg + rY c 3 B K | ]} | V d S rQ ) get_revision)rb rU r; s r r zbranches..@ sB $ $58++C00$ $ $ $ $ $rY N)r rd r is_branch_pointr r r nextrev)r r r r; s `@@r branchesr * s
(
0
0F##%%
g
== $ $ $ $<>J$ $ $
rY c t j fd}t |d 5 ddd dS # 1 swxY w Y dS )zDisplay the current revision for a database.
:param config: a :class:`.Config` instance.
:param verbose: output in verbose mode.
c r7 dt j |j j j | D ]*} | +g S )NzCurrent revision(s) for %s:)r r
obfuscate_url_pw
connectionengineurlget_all_currentr )rU rV r r; r s r display_versionz current..display_versionR s -%g&8&?&CDD
))#.. 9 9C w 7 7 8 8 8 8 rY T)r] dont_mutateNr rd r rj )r r r r; s `` @r r r G s
(
0
0F
?
s AAApurgec
t j |
|rmg d}t j | D ]S}d|v r8| dd \ }}||k r||}nt j d | Tnt j |
fd}t |
|||r|ndt j || 5
ddd dS # 1 swxY w Y dS )a 'stamp' the revision table with the given revision; don't
run any migrations.
:param config: a :class:`.Config` instance.
:param revision: target revision or list of revisions. May be a list
to indicate stamping of multiple branch heads.
.. note:: this parameter is called "revisions" in the command line
interface.
.. versionchanged:: 1.2 The revision may be a single revision or
list of revisions when stamping multiple branch heads.
:param sql: use ``--sql`` mode
:param tag: an arbitrary "tag" that can be intercepted by custom
``env.py`` scripts via the :class:`.EnvironmentContext.get_tag_argument`
method.
:param purge: delete all entries in the version table before stamping.
.. versionadded:: 1.2
Nr r zMStamp operation with --sql only supports a single starting revision at a timec T t j | S rQ ) _stamp_revsr
to_tuple)rU rV destination_revsr; s r do_stampzstamp..do_stamp s$ !!$-0@"A"A3GGGrY )r] r^ r r r r )
r rd r
to_listr r0 appendr r rj ) r rq rD r r r _revisionsrevr r r; s @@r stampr c s B
(
0
0F
2 h// / /Ii "+//#q"9"9i 4''#+'+ " / A
# #I . . . . / <11H H H H H H
%(2\\d
&677
s
C//C36C3rU c h t j | |dk rFfd}t | | 5 ddd dS # 1 swxY w Y dS | }|st j d|z |D ] }|sJ t j |j dS )zEdit revision script(s) using $EDITOR.
:param config: a :class:`.Config` instance.
:param rev: target revision.
r c | st j d | D ]}t j |j g S )NzNo current revisions)r
r0 r open_in_editorr )rU rV r r; s r edit_currentzedit..edit_current sV
@'(>??? **3//
-
-#BG,,,,IrY r Nz*No revision files indicated by symbol '%s') r rd r rj r r
r0 r r )r rU r revsr r; s @r editr s8
(
0
0F
i <
@
@
@ NN ##C(( #.do_ensure_version s %%''' rY )r] r^ Nr )r rD r; r s r ensure_versionr s
(
0
0F
s A
AA)r r
)r! F)
r r
r" r# r$ r# r% r& r' r( )
NFFrA FNNNNN)r r
rB rC r r& rD r& rA r# rE r& rF rC rG rC rH rC rI rC rJ rK r' rL )r rr r' r( )NNN) r r
r} r# rB rC rF rC rH rC r' r~ )FN)
r r
rq r# rD r& r rC r' r( )NFF)
r r
r rC r r& r r& r' r( )FF)F)r r
r r& r' r( )FNF) r r
rq r# rD r& r rC r r& r' r( )r r
rU r# r' r( )r r
rD r& r' r( )$
__future__r r typingr r r r r rf r
runtime.environmentr r; r alembic.configr
alembic.script.baser r r r@ rq r| r r r r r r r r r r r ra rY r r s " " " " " " % % % % % % 3 3 3 3 3 3 # # # # # # @%%%%%%******??????I I I I, Q Q Q Q Ql " "&"& $HL| | | | |~5C 5C 5C 5Cv ""& $ $ $ $ $T ( ( ( ( (\ + + + + +\. . .6 $ " C5 C5 C5 C5 C5L
2 : > C C C C CL ) ) ) )B rY