o gV@sFddlmZddlmZddlmZddlmZddlZddlZddl Z ddl m Z ddl m Z ddl m Z dd l mZdd l mZdd l mZdd l mZdd l mZddl mZddlmZddlmZddlmZddlmZddlmZGdddZGdddeddZGdddZ  d(d)d%d&Zed'kredSdS)*) annotations)ArgumentParser) Namespace) ConfigParserN)Any)cast)Dict)Mapping)Optional)overload)Sequence)TextIO)Union) TypedDict) __version__)command)util)compatc@s6eZdZUdZdddejdedfdDddZdZ d e d < dZ de d< dZ de d< ej dEddZdFddZej dGd!d"ZdHd#d$Ze %dIdJd)d*ZedKd,d*ZedLd/d*Z dMdNd1d*ZdOd3d4ZdPd5d6ZdQd8d9Z dMdRd;d<ZedSd=d>Ze dMdTd?d>Z dMdTd@d>Zej dUdBdCZdS)VConfiga` Represent an Alembic configuration. Within an ``env.py`` script, this is available via the :attr:`.EnvironmentContext.config` attribute, which in turn is available at ``alembic.context``:: from alembic import context some_param = context.config.get_main_option("my option") When invoking Alembic programmatically, a new :class:`.Config` can be created by passing the name of an .ini file to the constructor:: from alembic.config import Config alembic_cfg = Config("/path/to/yourapp/alembic.ini") With a :class:`.Config` object, you can then run Alembic commands programmatically using the directives in :mod:`alembic.command`. The :class:`.Config` object can also be constructed without a filename. Values can be set programmatically, and new sections will be created as needed:: from alembic.config import Config alembic_cfg = Config() alembic_cfg.set_main_option("script_location", "myapp:migrations") alembic_cfg.set_main_option("sqlalchemy.url", "postgresql://foo/bar") alembic_cfg.set_section_option("mysection", "foo", "bar") .. warning:: When using programmatic configuration, make sure the ``env.py`` file in use is compatible with the target configuration; including that the call to Python ``logging.fileConfig()`` is omitted if the programmatic configuration doesn't actually include logging directives. For passing non-string values to environments, such as connections and engines, use the :attr:`.Config.attributes` dictionary:: with engine.begin() as connection: alembic_cfg.attributes['connection'] = connection command.upgrade(alembic_cfg, "head") :param file\_: name of the .ini file to open. :param ini_section: name of the main Alembic section within the .ini file :param output_buffer: optional file-like input buffer which will be passed to the :class:`.MigrationContext` - used to redirect the output of "offline generation" when using Alembic programmatically. :param stdout: buffer where the "print" output of commands will be sent. Defaults to ``sys.stdout``. :param config_args: A dictionary of keys and values that will be used for substitution in the alembic config file. The dictionary as given is **copied** to a new one, stored locally as the attribute ``.config_args``. When the :attr:`.Config.file_config` attribute is first invoked, the replacement variable ``here`` will be added to this dictionary before the dictionary is passed to ``ConfigParser()`` to parse the .ini file. :param attributes: optional dictionary of arbitrary Python keys/values, which will be populated into the :attr:`.Config.attributes` dictionary. .. seealso:: :ref:`connection_sharing` Nalembicfile_"Union[str, os.PathLike[str], None] ini_sectionstr output_bufferOptional[TextIO]stdoutr cmd_optsOptional[Namespace] config_argsMapping[str, Any] attributesOptional[Dict[str, Any]]returnNonecCs@||_||_||_||_||_t||_|r|j|dSdS)z Construct a new :class:`.Config`N) config_file_nameconfig_ini_sectionrrrdictr r"update)selfrrrrrr r"r+J/var/www/html/flaskapp/venv/lib/python3.10/site-packages/alembic/config.py__init__ds  zConfig.__init__r&r'Dict[str, Any]cCsiS)aA Python dictionary for storage of additional state. This is a utility dictionary which can include not just strings but engines, connections, schema objects, or anything else. Use this to pass objects into an env.py script, such as passing a :class:`sqlalchemy.engine.base.Connection` when calling commands from :mod:`alembic.command` programmatically. .. seealso:: :ref:`connection_sharing` :paramref:`.Config.attributes` r+r*r+r+r,r"szConfig.attributestextargrcGs8|r t||}nt|}tj|j|dfi|jdS)aRender a message to standard out. When :meth:`.Config.print_stdout` is called with additional args those arguments will formatted against the provided text, otherwise we simply output the provided text verbatim. This is a no-op when the``quiet`` messaging option is enabled. e.g.:: >>> config.print_stdout('Some text %s', 'arg') Some Text arg  N)rrwrite_outstreamrmessaging_opts)r*r0r1outputr+r+r, print_stdoutszConfig.print_stdoutrcCs`|jrtjtj|j}nd}||jd<t|j}|jr(t||jg|S| |j |S)aReturn the underlying ``ConfigParser`` object. Direct access to the .ini file is available here, though the :meth:`.Config.get_section` and :meth:`.Config.get_main_option` methods provide a possibly simpler interface. here) r&ospathabspathdirnamer rrread_config_parser add_sectionr')r*r8 file_configr+r+r,r?s    zConfig.file_configcCs,ddl}tjtj|j}tj|dS)zReturn the directory where Alembic setup templates are found. This method is used by the alembic ``init`` and ``list_templates`` commands. rN templates)rr9r:r;r<__file__join)r*r package_dirr+r+r,get_template_directoryszConfig.get_template_directory.namedefaultOptional[Dict[str, str]]cCdSNr+r*rErFr+r+r, get_sectionzConfig.get_sectionDict[str, str]cCrHrIr+rJr+r+r,rKrLMapping[str, str](Union[Dict[str, str], Mapping[str, str]]cCrHrIr+rJr+r+r,rKrLOptional[Mapping[str, str]]cCs |j|s|St|j|S)zReturn all the configuration options from a given .ini file section as a dictionary. If the given section does not exist, the value of ``default`` is returned, which is expected to be a dictionary or other mapping. )r? has_sectionr(itemsrJr+r+r,rKs valuecCs||j||dS)a:Set an option programmatically within the 'main' section. This overrides whatever was in the .ini file. :param name: name of the value :param value: the value. Note that this value is passed to ``ConfigParser.set``, which supports variable interpolation using pyformat (e.g. ``%(some_value)s``). A raw percent sign not part of an interpolation symbol must therefore be escaped, e.g. ``%%``. The given value may refer to another value already in the file using the interpolation format. N)set_section_optionr')r*rErSr+r+r,set_main_optionszConfig.set_main_optioncCs|j|j|dSrI)r? remove_optionr')r*rEr+r+r,remove_main_optionszConfig.remove_main_optionsectioncCs,|j|s |j||j|||dS)aSet an option programmatically within the given section. The section is created if it doesn't exist already. The value here will override whatever was in the .ini file. :param section: name of the section :param name: name of the value :param value: the value. Note that this value is passed to ``ConfigParser.set``, which supports variable interpolation using pyformat (e.g. ``%(some_value)s``). A raw percent sign not part of an interpolation symbol must therefore be escaped, e.g. ``%%``. The given value may refer to another value already in the file using the interpolation format. N)r?rQr>set)r*rXrErSr+r+r,rTs  zConfig.set_section_option Optional[str]cCs@|j|std|j|f|j||r|j||S|S)z9Return an option from the given section of the .ini file.z6No config file %r found, or file has no '[%s]' section)r?rQr CommandErrorr& has_optionget)r*rXrErFr+r+r,get_section_option*s zConfig.get_section_optioncCrHrIr+rJr+r+r,get_main_option8szConfig.get_main_optioncCrHrIr+rJr+r+r,r_;rLcCs||j||S)zReturn an option from the 'main' section of the .ini file. This defaults to being a key from the ``[alembic]`` section, unless the ``-n/--name`` flag were used to indicate a different section. )r^r'rJr+r+r,r_@s MessagingOptionsc Cstttdt|jddiS)zThe messaging options.quietF)rr`r immutabledictgetattrrr/r+r+r,r4Ls zConfig.messaging_opts)rrrrrrrr rrr r!r"r#r$r%)r$r.)r0rr1rr$r%)r$r)r$r).)rErrFr%r$rG)rErrFrMr$rM)rErrFrNr$rOrI)rErrFrPr$rP)rErrSrr$r%)rErr$r%)rXrrErrSrr$r%)rXrrErrFrZr$rZ)rErrFrr$r)rErrFrZr$rZ)r$r`)__name__ __module__ __qualname____doc__sysrrrbr-r__annotations__r&r'memoized_propertyr"r6r?rDr rKrUrWrTr^r_r4r+r+r+r,rsZ J                 rc@seZdZUded<dS)r`boolraN)rdrerfrir+r+r+r,r`Ws r`F)totalc@s8eZdZddddZddd ZdddZddddZdS) CommandLineNprogrZr$r%cCs||dSrI)_generate_args)r*rnr+r+r,r-\szCommandLine.__init__c sd1fdd }t|d }|jd d d td|jddttjdddd|jddtddd|jdddd|jdddd|jd d!dd"d|}tj d#d$iid%d&t tDD]t rj d'd(krjd)krt}|d*dur|d'd+t|d* }|d't|d* d}n |d'd+d}g}vrfd,d&|D}j}|rg} |d-D]} | sn| | qng} |jj d.| d/|||j||fd0q]||_dS)2Nfnrparser positionalkwargsr$r%c sidddtdtddfddd ttd d fd d tdddfddttdd fddttdd fddtdddfddtdddfddttd d fd!d"ttd#d fd$d%ttd&d fd'd(d)tdd*dfd+d,tdd-dfd.d/tdd0dfd1d2d3td4d5dfd6d7d8tdd9dfd:d;tddtdd?df}d@dAdBdC}|D]}||vr||}|dDdE|dE}}|j|i|q|D](}|dFks|vr||dFkrjdFdG|dFdHqɈj|||dIqdS)JNtemplatez-tz --templategenericz"Setup template for use with 'init')rFtypehelpmessagez-mz --messagez%Message string to use with 'revision')rvrwsqlz--sql store_truez\Don't emit SQL to database - dump to standard output/file instead. See docs on offline mode.actionrwtagz--tagz@head to base new revision on.splicez--splicez6Allow a non-head revision as the 'head' to splice onto depends_onz --depends-onappendzNSpecify one or more revision identifiers which this revision should depend on.rev_idz--rev-idz9Specify a hardcoded revision id instead of generating one version_pathz--version-pathz2Specify specific path from config for version file branch_labelz--branch-labelz3Specify a branch label to apply to the new revisionverbosez-vz --verbosezUse more verbose outputresolve_dependenciesz--resolve-dependenciesz+Treat dependency versions as down revisions autogeneratez--autogeneratezgPopulate revision script with candidate migration operations, based on comparison of database to model. rev_rangez-rz --rev-rangestorez1Specify a revision range; format is [start]:[end]indicate_currentz-iz--indicate-currentzIndicate the current revisionpurgez--purgez7Unconditionally erase the version table before stampingpackagez --packagezFWrite empty __init__.py files to the environment and version locationszlocation of scripts directoryzrevision identifierz/one or more revisions, or 'heads' for all heads) directoryrevision revisionsrr+)nargsrwrw)r(r add_argumentr]) rprqrrrs kwargs_optspositional_helpr1argskw)positional_translations subparserr+r, add_options`sF "*2:BJR W^gpx z/CommandLine._generate_args..add_optionsrnz --versionversionz %%(prog)s %s)r|rz-cz--configALEMBIC_CONFIGz alembic.inizaAlternate config file; defaults to value of ALEMBIC_CONFIG environment variable, or "alembic.ini")rvrFrwz-nz--namerz6Name of section in .ini file to use for Alembic configz-xrzlAdditional arguments consumed by custom env.py scripts, e.g. -x setting1=somesetting -x setting2=somesettingr{z --raiseerrrzz!Raise a full stack trace on errorz-qz--quietzDo not log to std output.rrcSsg|]}tt|qSr+)rcr).0nr+r+r, 1sz.CommandLine._generate_args..r_zalembic.commandrcsg|] }||qSr+)r])rrE)rprr+r,r@sr2 r)cmd) rprrqrrrrrsrr$r%)rrrrr9environr]add_subparsersrstampdirinspect isfunctionrdrerinspect_getfullargspeclenrgsplitstripr add_parserrB set_defaultsrq) r*rnrrq subparsersspecrrkwarghelp_ help_textliner+)rprrr,ro_s &         zCommandLine._generate_argsconfigroptionsrc sj\}}}z||gfdd|DRifdd|DWdStjyF}zjr/tjt|fi|jWYd}~dSd}~ww)Ncsg|]}t|dqSrIrcrkrr+r,r]sz'CommandLine.run_cmd..csi|] }|t|dqSrIrrrr+r, ^sz'CommandLine.run_cmd..)rrr[raiseerrerrrr4)r*rrrprrrer+rr,run_cmdWs  &zCommandLine.run_cmdargvOptional[Sequence[str]]cCsH|j|}t|ds|jddSt|j|j|d}|||dS)Nrztoo few arguments)rrr)rq parse_argshasattrerrorrrrEr)r*rrcfgr+r+r,mainfs  zCommandLine.mainrI)rnrZr$r%)rrrrr$r%)rrr$r%)rdrerfr-rorrr+r+r+r,rm[s    yrmrrrnrZrsrr$r%cKst|dj|ddS)z(The console runner function for Alembic.r)rN)rmr)rrnrsr+r+r,rusr__main__)NN)rrrnrZrsrr$r%) __future__rargparserr configparserrrr9rhtypingrrrr r r r r rtyping_extensionsrr7rrrrrr`rmrrdr+r+r+r,sD                 >