a g g>@sddlmZddlmZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddl mZddl mZddlmZddlmZddlmZddlmZddlmZerBddlmZddl m!Z!ddl m"Z"ddl#m$Z$zddl%m&Z&Wne'yjdZ&Yn0ee(ee(fZ)e*dZ+e*dZ,e*dZ-e*dZ.e*d Z/d!Z0e*d"Z1e*d#Z2Gd$d%d%Z3Gd&d'd'ej$Z4dS)() annotations)contextmanagerN) ModuleType)Any)cast)Dict)Iterator)List)Optional)Sequence)Set)Tuple) TYPE_CHECKING)Unionrevision) write_hooks)util) migration)not_none)Config) RevisionStep) StampStep)Revision)tzz (?!\.\#|__init__)(.*\.py)(c|o)?$z(?!\.\#|__init__)(.*\.py)$z([a-f0-9]+)\.py$z(upgrade|downgrade)_([a-z0-9]+)z\w+z%(rev)s_%(slug)sz , *|(?: +)z , *|(?: +)|\:c @seZdZdZedddddddfddddd dd d d d d ddZeddddZej ddZ ddddZ e dddddZ eddd d d d d ddddZdedddd"d#d$Zd%d&d'd(d)Zd*d+d'd,d-Zdd.d'd/d0Zd d1d'd2d3Zd4d4d5dd6d7d8Zd dd9d:Zd;ddd?Zd;dd@dAZdddBdCdDdEZdd dBdCdFdGZd%d%dHdIdJdKZd ddLdMZedNdOZddd5d dPdQdRZddd dSdTdUZ dd dVdWdXZ!dYddZd[Z"dfdd d d d\d d d]d5d.d^ d_d`Z#ddd dYddadbdcZ$dS)gScriptDirectoryaHProvides operations upon an Alembic script directory. This object is useful to get information as to current revisions, most notably being able to get at the "head" revision, for schemes that want to test if the current revision in the database is the most recent:: from alembic.script import ScriptDirectory from alembic.config import Config config = Config() config.set_main_option("script_location", "myapp:migrations") script = ScriptDirectory.from_config(config) head_revision = script.get_current_head() (NFutf-8strz Optional[int]zOptional[List[str]]boolz Optional[str]zOptional[Dict[str, str]]None) dir file_templatetruncate_slug_lengthversion_locations sourcelessoutput_encodingtimezone hook_configrecursive_version_locationsreturnc Csp||_||_||_|pd|_||_||_t|j|_ ||_ ||_ | |_ t |t jsltdt j|dS)NrzVPath doesn't exist: %r. Please use the 'init' command to create a new scripts folder.)r#r$r&r%r'r(rZ RevisionMap_load_revisions revision_mapr)r*r+osaccessF_OKr CommandErrorpathabspath) selfr#r$r%r&r'r(r)r*r+r6V/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/alembic/script/base.py__init__Is   zScriptDirectory.__init__r,cCs*|j}t|dkrtdn|dSdS)Nrz"Multiple version_locations presentr)_version_locationslenrr2)r5locr6r6r7versionsgs  zScriptDirectory.versionscCs4|jrdd|jDStjtj|jdfSdS)NcSsg|]}tjt|qSr6)r/r3r4rcoerce_resource_to_filename).0locationr6r6r7 rsz6ScriptDirectory._version_locations..r=)r&r/r3r4joinr#r5r6r6r7r:os z"ScriptDirectory._version_locationszIterator[Script]c cs|jrdd|jD}n|jg}t}|D]z}t||D]h}tj|}||vrbt d|q:| |tj |}tj |}t|||}|durq:|Vq:q*dS)NcSsg|]}tj|r|qSr6)r/r3exists)r?versr6r6r7rA{s z3ScriptDirectory._load_revisions..zJFile %s loaded twice! ignoring. Please ensure version_locations is unique.)r&r:r=setScript _list_py_dirr/r3realpathrwarnaddbasenamedirname_from_filename) r5pathsZdupesrE file_path real_pathfilenamedir_namescriptr6r6r7r-ys.    zScriptDirectory._load_revisionsr)configr,c CsP|d}|durtd|d}|dur8t|}nd}|d}|r|d}ddtjdd d }z ||}Wn2ty} ztd || WYd} ~ qd} ~ 00|durt |} qd d | |D} nd} |d} | rt t | t j dd<|ddk} tt||dt||ddk|dd| |d|di| d S)zProduce a new :class:`.ScriptDirectory` given a :class:`.Config` instance. The :class:`.Config` need only have the ``script_location`` key present. script_locationNz0No 'script_location' key found in configuration.r%r&version_path_separator :;)Nspacer/rYrZzV'%s' is not a valid value for version_path_separator; expected 'space', 'os', ':', ';'cSsg|] }|r|qSr6r6)r?xr6r6r7rAsz/ScriptDirectory.from_config..prepend_sys_pathrr+truer$r'r(rr)post_write_hooks)r$r%r'r(r&r)r*r+)Zget_main_optionrr2intr/pathsepKeyError ValueError_split_on_space_commasplitlist_split_on_space_comma_colonsysr3rr>_default_file_templateZ get_section) clsrUrVZtslr%Zversion_locations_strrWZ split_on_pathZ split_charZker&r]Zrvlr6r6r7 from_configsv        zScriptDirectory.from_configzIterator[None])ancestormultiple_headsstartend resolutionr,c csRz dVWn@tjy|}zV|dur2tt|j}|durFtt|j}|sNd}|||d}t||WYd}~nd}~0tjy}z<|sd}||p|j t |j d}t||WYd}~nd}~0tj y}z*|durd|j }t||WYd}~nBd}~0tj yL} zt| jd| WYd} ~ n d} ~ 00dS)NzgRequested range %(start)s:%(end)s does not refer to ancestor/descendant revisions along the same branchrnrozMultiple head revisions are present for given argument '%(head_arg)s'; please specify a specific target revision, '@%(head_arg)s' to narrow to a specific head, or 'heads' for all heads)Zhead_argheadsz(Can't locate revision identified by '%s'r)rZRangeNotAncestorErrorrrlowerupperrr2Z MultipleHeadsargumentformat_as_commarrResolutionError RevisionErrorargs) r5rlrmrnrorpZrnaZmhreerrr6r6r7_catch_revision_errorss:        z&ScriptDirectory._catch_revision_errorsbaserr)r}headr,ccsX|j||d6|jj||dddD]}tt|Vq$Wdn1sJ0YdS)a-Iterate through all revisions. :param base: the base revision, or "base" to start from the empty revision. :param head: the head revision; defaults to "heads" to indicate all head revisions. May also be "head" to indicate a single head revision. rqTF)Z inclusiveZassert_relative_lengthN)r|r.iterate_revisionsrrG)r5r}r~revr6r6r7walk_revisionss  zScriptDirectory.walk_revisions _RevIdTypezTuple[Optional[Script], ...])id_r,cCsJ|.ttttdf|j|WdS1s<0YdS)zReturn the :class:`.Script` instance with the given rev identifier, symbolic name, or sequence of identifiers. .N)r|rr r rGr. get_revisionsr5rr6r6r7r's   zScriptDirectory.get_revisionszTuple[str, ...]zSet[Optional[Script]]cCsF|*tttt|j|WdS1s80YdSN)r|rr r rGr.Z_get_all_currentrr6r6r7get_all_current2s zScriptDirectory.get_all_currentOptional[Script]cCsB|&ttt|j|WdS1s40YdS)zReturn the :class:`.Script` instance with the given rev id. .. seealso:: :meth:`.ScriptDirectory.get_revisions` N)r|rr rGr. get_revisionrr6r6r7r8s zScriptDirectory.get_revisionz%Optional[Union[str, Tuple[str, ...]]]cCsX| |j|\}}Wdn1s.0Y|s@dS|dkrL|S|dSdS)z[Convert a symbolic revision, i.e. 'head' or 'base', into an actual revision number.Nrrr)r|r.Z_resolve_revision_number)r5rr branch_namer6r6r7as_revision_numberDs .z"ScriptDirectory.as_revision_numberz!Union[str, Tuple[str, ...], None]r)rtrskwr,cKs ttt|jj||fi|S)aIterate through script revisions, starting at the given upper revision identifier and ending at the lower. The traversal uses strictly the `down_revision` marker inside each migration script, so it is a requirement that upper >= lower, else you'll get nothing back. The iterator yields :class:`.Script` objects. .. seealso:: :meth:`.RevisionMap.iterate_revisions` )rrrGr.r)r5rtrsrr6r6r7rUsz!ScriptDirectory.iterate_revisionscCs:|jdd|jWdS1s,0YdS)aGReturn the current head revision. If the script directory has multiple heads due to branching, an error is raised; :meth:`.ScriptDirectory.get_heads` should be preferred. :return: a string revision number. .. seealso:: :meth:`.ScriptDirectory.get_heads` z}The script directory has multiple heads (due to branching).Please use get_heads(), or merge the branches using alembic merge.rmN)r|r.get_current_headrCr6r6r7rosz ScriptDirectory.get_current_head List[str]cCs t|jjS)aUReturn all "versioned head" revisions as strings. This is normally a list of length one, unless branches are present. The :meth:`.ScriptDirectory.get_current_head()` method can be used normally when a script directory has only one head. :return: a tuple of string revision numbers. )rfr.rrrCr6r6r7 get_headss zScriptDirectory.get_headscCs4|}t|dkr tdn|r,|dSdSdS)a#Return the "base" revision as a string. This is the revision number of the script that has a ``down_revision`` of None. If the script directory has multiple bases, an error is raised; :meth:`.ScriptDirectory.get_bases` should be preferred. rz@The script directory has multiple bases. Please use get_bases().rN) get_basesr;rr2)r5basesr6r6r7get_bases  zScriptDirectory.get_basecCs t|jjS)zreturn all "base" revisions as strings. This is the revision number of all scripts that have a ``down_revision`` of None. )rfr.rrCr6r6r7rszScriptDirectory.get_baseszList[RevisionStep]) destination current_revr,cs\jd|d:j||dd}fddtt|DWdS1sN0YdS)NzFDestination %(end)s is not a valid upgrade target from current head(s)rlroT)Z implicit_basecsg|]}tjj|qSr6)r MigrationStepZupgrade_from_scriptr.r?rTrCr6r7rAsz1ScriptDirectory._upgrade_revs..)r|rreversedrfr5rrrevsr6rCr7 _upgrade_revss  zScriptDirectory._upgrade_revscsTjd|d2j||dd}fdd|DWdS1sF0YdS)NzHDestination %(end)s is not a valid downgrade target from current head(s)rT)Zselect_for_downgradecsg|]}tjj|qSr6)rrZdowngrade_from_scriptr.rrCr6r7rAsz3ScriptDirectory._downgrade_revs..)r|rrr6rCr7_downgrade_revss zScriptDirectory._downgrade_revszList[StampStep])rrrr,c sjdd|}g}|s&d}g}t|D]*}|r4|jjttt ||ddq4t |}|pxdg}|D]}|dur|fdd|Dq~n ||vrq~t j |g} t j |g} | |r"| |rJdd|D} t| |jd d j} || q~q~| |r`d d|D} t| |jdd j} || q~q~td |jddj} || q~q~|WdS1s0YdS) NzCMultiple heads are present; please specify a single target revisionrr}T)Zinclude_dependenciesc s"g|]}t|jdddjqS)NFT)rrrr.r?r~rCr6r7rAsz/ScriptDirectory._stamp_revs..cSsg|] }|jqSr6rrr6r6r7rAFcSsg|] }|jqSr6rrr6r6r7rA"rr6)r|rrto_tupleextendr.Zfilter_for_lineagerr rGZ unique_listrFZ_get_descendant_nodesZ_get_ancestor_nodes intersectionrrrappend) r5rrrZ heads_revsZstepsZfiltered_headsrZdestsdestZ descendantsZ ancestorsZ todo_headsstepr6rCr7 _stamp_revss|             zScriptDirectory._stamp_revscCst|jddS)zRun the script environment. This basically runs the ``env.py`` script present in the migration environment. It is called exclusively by the command functions in :mod:`alembic.command`. env.pyN)rload_python_filer#rCr6r6r7run_env7s zScriptDirectory.run_envcCstjtj|jdS)Nr)r/r3r4rBr#rCr6r6r7env_py_locationBszScriptDirectory.env_py_location)srcrrr,cKs.tjdtj|tj|||jfi|dSNz Generating %s)rstatusr/r3r4Ztemplate_to_filer()r5rrrr6r6r7_generate_templateFsz"ScriptDirectory._generate_template)rrr,cCs"tdtj|tj||dSr)rrr/r3r4shutilcopy)r5rrr6r6r7 _copy_filePszScriptDirectory._copy_file)r3r,cCs0tj|}tj|s,td|tj|dS)NzCreating directory %s)r/r3r4rDrrmakedirs)r5r3r6r6r7_ensure_directoryUs  z!ScriptDirectory._ensure_directoryzdatetime.datetimecCs|jdurvtdurtdt|j}|dur@t|j}|durXtd|jtjjt d |}n tj }|S)Nz>The library 'python-dateutil' is required for timezone supportzCan't locate timezone: %s)tzinfo) r)rrr2ZgettzrtdatetimeutcnowreplaceZtzutc astimezonenow)r5r create_dater6r6r7_generate_create_dateZs(    z%ScriptDirectory._generate_create_dateOptional[bool]zOptional[_RevIdType]) revidmessager~refreshsplice branch_labels version_path depends_onrr,c  s>|dur d}zt|Wn8tjyR} zt| jd| WYd} ~ n d} ~ 00jddDtt t ddfj |} | D]} | dksJqWdn1s0Yt t| t | krtd } |dur.cs g|]}tj||fqSr6)rr.r)r?rrCr6r7rAszscript.py.makocss |]}|dur|jndVqdSrr)r?hr6r6r7 rz4ScriptDirectory.generate_revision..z empty message)Z up_revision down_revisionrrrcommarzVersion %s specified branch_labels %s, however the migration file %s does not have them; have you upgraded your script.py.mako to include the 'branch_labels' section?)+rGZ verify_rev_idrrxrr2ryr|rr r r.rr;rFrr: isinstancer/r3rMr=normpathr4r&r _rev_pathis_headZto_listrrBr#r Ztuple_rev_as_scalartuplerrvr*rZ _run_hooks _from_pathrZ add_revision)r5rrr~rrrrrrr{rrrrZhead_Z norm_pathZ vers_pathr3Zresolved_depends_onr_rTr6rCr7generate_revisionss( ,       (   (   z!ScriptDirectory.generate_revision)r3rev_idrrr,c Cst|}dt|pd}t||jkrP|d|jdddd}d|j ||||j |j |j |j |j|jd }tj||S)N_rrz%s.py) rslugepochyearmonthdayhourminutesecond)r` timestamprB_slug_refindallrsr;r%rsplitr$rrrrrrr/r3)r5r3rrrrrrRr6r6r7rs& zScriptDirectory._rev_path)NNNNN)r}rr)NFFNNN)%__name__ __module__ __qualname____doc__rir8propertyr=rZmemoized_propertyr:r- classmethodrkrr|rrrrrrrrrrrrrrrrrrrrrr6r6r6r7r4sl" O.    `   "rcs eZdZUdZddddfdd Zded<ded<d Zd ed <edd d dZedd ddZ edd ddZ ddZ d+dddddddddZ d,dddddddddZ dd ddZed dd!d"d#d$Zed dd%d"d&d'Zed ddd!d(d)d*ZZS)-rGzRepresent a single revision file in a ``versions/`` directory. The :class:`.Script` instance is returned by methods such as :meth:`.ScriptDirectory.iterate_revisions`. rr )modulerr3c sJ||_||_tj||jtjt|ddddtjt|ddddddS)Nrr6)defaultr)r dependencies)rr3superr8rrrgetattr)r5rrr3 __class__r6r7r8+s  zScript.__init__rr3Nr_db_current_indicatorr9cCstd|jdS))Return the docstring given in the script.z r)rzrelongdocrCr6r6r7docCsz Script.doccCs6|jj}|r.t|jdr&||jj}|SdSdS)r_alembic_source_encodingrN)rrhasattrdecoderstrip)r5rr6r6r7rIs zScript.longdoccCsd|j|jrdnd|jrdnd|jr(dnd|jr4dndf}|jrV|d|f7}n|d|f7}|jr|d t|j7}|jr|d t|j 7}|j r|d t|j f7}|d |j f7}|d d dd|j D7}|S)NzRev: %s%s%s%s%s  (head)r (branchpoint) (mergepoint) (current)z Merges: %s z Parent: %s zAlso depends on: %s zBranches into: %s zBranch names: %s z Path: %s z %s  css|]}d|VqdS)z %sNr6)r?parar6r6r7rwrz#Script.log_entry..)rris_branch_pointis_merge_pointr_format_down_revisionrrrvZnextrevrr3rBr splitlines)r5entryr6r6r7 log_entryWs6       zScript.log_entrycCs:d||j|jrdnd|jr"dnd|jr.dnd|jfS)Nz%s -> %s%s%s%s, %srrrr)rrrrrrrCr6r6r7__str__{s   zScript.__str__FTr!)include_branches include_docinclude_parentstree_indicatorshead_indicatorsr,cCs|j}|r<|jr,d|t|j|f}nd||f}|dusHJ|rf|jrf|dt|j7}|sn|r|d|jr|dnd|jr|jsdnd|jrdndf7}|r|d |j rd nd|j rd ndf7}|r|d |j 7}|S) Nz %s (%s) -> %sz%s -> %sz (%s)z%s%s%srrz (effective head)rz%s%srrz, %s) rrrrrvrZ _is_real_headrrrrr)r5rrrrrtextr6r6r7 _head_onlys<       zScript._head_only)verboserrrrr,cCs|r |jS|||||SdSr)rr)r5rrrrrr6r6r7 cmd_formats zScript.cmd_formatcCs|js dSt|jSdS)Nz)rrrvZ_versioned_down_revisionsrCr6r6r7rszScript._format_down_revisionrr) scriptdirr3r,cCstj|\}}||||Sr)r/r3rerN)rjr r3dir_rRr6r6r7rszScript._from_pathrcsg}tj|ddD]\}}}|dr(qt|D]}|tj||q0|jrtj|dtjrdd|D| fddt D|j sq| q|S)NT)topdown __pycache__cSsh|]}|ddqS).r)re)r?rRr6r6r7 rz&Script._list_py_dir..c3s.|]&}|ddvrtj|VqdS)rrN)rer/r3rB)r?ZpycnamesZ py_cache_pathr6r7rsz&Script._list_py_dir..) r/walkendswithsortedrr3rBr'rDrlistdirr+sort)rjr r3rOrootdirsfilesrRr6rr7rHs"    zScript._list_py_dir)r r rRr,c Cs|jrt|}n t|}|s$dS|d}|jrR|ddk}|ddk}nd}}|sb|rtjtj||}tjtj||d} |s|r| rdSt ||} t | dst |} | st d|q| d} n| j} t| | tj||S)NrrcoFrzCould not determine revision id from filename %s. Be sure the 'revision' variable is declared inside the script (please see 'Upgrading from Alembic 0.1 to 0.2' in the documentation).)r'_sourceless_rev_filematch_only_source_rev_filegroupr/r3rDrBrrr _legacy_revr2rrG) rjr r rRZpy_matchZ py_filenameZis_cZis_oZ py_existsZ pyc_existsrmrr6r6r7rNs6        zScript._from_filename)FFFTT)FFFT)rrrrr8__annotations__rrrrrrrr rrrrHrN __classcell__r6r6rr7rG"s<   # )$rG)5 __future__r contextlibrrr/rzrrhtypesrtypingrrrrr r r r r rrrrrrZruntimerrrUrZruntime.migrationrrZscript.revisionrZdateutilr ImportErrorr rcompilerrr Z _mod_def_rerrirdrgrrGr6r6r6r7s`                               s