Q&' L/ x d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm
Z
d d
lm Z d d lm Z d d lm
Z
d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlm Z erBd dl m Z d dl m Z d dl!m"Z" d d lm#Z# d d lm$Z$ dd lm%Z% dd l&m'Z' dd l(m)Z) dd!l*m+Z+ dd"l,m-Z- dd#l.m/Z/ e ee0ee0d$f f Z1eeee0e0f ed% gd&f Z2ee0ed'gee0ed( f f Z3ed) Z4e
ed* e e0 f Z5ee e0 e4e5ge6f Z7ed+e e0 e4e6e d+ ge6f Z8eed,ee e e0ef gd&f Z9eed-d-e e0 e e e e0 ge e6 f Z: G d. d/ej; Z | j j S )a Return True if the context is configured to expect a
transactional DDL capable backend.
This defaults to the type of database in use, and
can be overridden by the ``transactional_ddl`` argument
to :meth:`.configure`
This function requires that a :class:`.MigrationContext`
has first been made available via :meth:`.configure`.
) get_contextimpltransactional_ddlr? s r: is_transactional_ddlz'EnvironmentContext.is_transactional_ddl s !!&88r< c , | S rC )rK r? s r: requires_connectionz&EnvironmentContext.requires_connection s ''))))r<
_RevNumberc 6 | j d S )a Return the hex identifier of the 'head' script revision.
If the script directory has multiple heads, this
method raises a :class:`.CommandError`;
:meth:`.EnvironmentContext.get_head_revisions` should be preferred.
This function does not require that the :class:`.MigrationContext`
has been configured.
.. seealso:: :meth:`.EnvironmentContext.get_head_revisions`
headr3 as_revision_numberr? s r: get_head_revisionz$EnvironmentContext.get_head_revision s {--f555r< c 6 | j d S )a Return the hex identifier of the 'heads' script revision(s).
This returns a tuple containing the version number of all
heads in the script directory.
This function does not require that the :class:`.MigrationContext`
has been configured.
headsrV r? s r: get_head_revisionsz%EnvironmentContext.get_head_revisions s {--g666r< c | j 1| j | j S d| j v r%| j | j d S t
j d )aT Return the 'starting revision' argument,
if the revision was passed using ``start:end``.
This is only meaningful in "offline" mode.
Returns ``None`` if no value is available
or was configured.
This function does not require that the :class:`.MigrationContext`
has been configured.
N starting_revz+No starting revision argument is available.)r1 r3 rW rM _start_from_revr8 r CommandErrorr? s r: get_starting_revision_argumentz1EnvironmentContext.get_starting_revision_argument s " .;11 ""2
t 0
0
0;11!.1
#=
r< c L | j | j d S )a Get the 'destination' revision argument.
This is typically the argument passed to the
``upgrade`` or ``downgrade`` command.
If it was specified as ``head``, the actual
version number is returned; if specified
as ``base``, ``None`` is returned.
This function does not require that the :class:`.MigrationContext`
has been configured.
destination_rev)r3 rW r8 r? s r: get_revision_argumentz(EnvironmentContext.get_revision_argument% s* {-- / 0
r<
Optional[str]c 8 | j dd S )ag Return the value passed for the ``--tag`` argument, if any.
The ``--tag`` argument is not used directly by Alembic,
but is available for custom ``env.py`` configurations that
wish to use it; particularly for offline generation scripts
that wish to generate tagged filenames.
This function does not require that the :class:`.MigrationContext`
has been configured.
.. seealso::
:meth:`.EnvironmentContext.get_x_argument` - a newer and more
open ended system of extending ``env.py`` scripts via the command
line.
tagNrI r? s r: get_tag_argumentz#EnvironmentContext.get_tag_argument7 s $ $$UD111r<
as_dictionaryLiteral[False] List[str]c d S rC r9 rh s r: get_x_argumentz!EnvironmentContext.get_x_argumentK r<
Literal[True]Dict[str, str]c d S rC rl rm s r: rn z!EnvironmentContext.get_x_argumentO ro r< . Union[List[str], Dict[str, str]]c d S rC rl rm s r: rn z!EnvironmentContext.get_x_argumentS s r< Fc | j j | j j j pg }ng }|rt d |D }|S )a Return the value(s) passed for the ``-x`` argument, if any.
The ``-x`` argument is an open ended flag that allows any user-defined
value or values to be passed on the command line, then available
here for consumption by a custom ``env.py`` script.
The return value is a list, returned directly from the ``argparse``
structure. If ``as_dictionary=True`` is passed, the ``x`` arguments
are parsed using ``key=value`` format into a dictionary that is
then returned.
For example, to support passing a database URL on the command line,
the standard ``env.py`` script can be modified like this::
cmd_line_url = context.get_x_argument(
as_dictionary=True).get('dbname')
if cmd_line_url:
engine = create_engine(cmd_line_url)
else:
engine = engine_from_config(
config.get_section(config.config_ini_section),
prefix='sqlalchemy.',
poolclass=pool.NullPool)
This then takes effect by running the ``alembic`` script as::
alembic -x dbname=postgresql://user:pass@host/dbname upgrade head
This function does not require that the :class:`.MigrationContext`
has been configured.
.. seealso::
:meth:`.EnvironmentContext.get_tag_argument`
:attr:`.Config.cmd_opts`
Nc 3 B K | ]}| d d V dS )=r N)split).0rA s r: z4EnvironmentContext.get_x_argument.. s0 <r s " " " " " " ! ! ! ! ! ! ! ! ! ! ! ! ( ( ( ( ( ( . . . . . . % % % % % % ( ( ( ( ( ( ' ' ' ' ' ' # # # # # #
.%%%%%%111111555555......000000((((((111111 !!!!!!111111------
eCsCx 01
2
%uS#X-?(@A4G # %WU^(;"<< ' FG SM
c]N$9:D@
3c9JK
TN
h' h' h' h' h', h' h' h' h' h'r<