관리-도구
편집 파일: config.cpython-37.pyc
B �M!V�= � @ s� d dl mZ ddlmZ d dlZd dlZd dlZddlmZ ddlm Z ddlm Z ddl m Z G d d � d e �Z G d d � d e �Zdd d�Zedkr�e� dS )� )�ArgumentParser� )�SafeConfigParserN)�command)�util)� package_dir)�compatc @ s� e Zd ZdZdddejde�� dfdd�ZdZ dZ dZ ej dd� �Z dd � Zej d d � �Zd d � Zdd� Zdd� Zdd� Zdd� Zddd�Zddd�ZdS ) �Configak 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 programatically, 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("url", "postgresql://foo/bar") alembic_cfg.set_section_option("mysection", "foo", "bar") 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``. .. versionadded:: 0.4 :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 ``SafeConfigParser()`` to parse the .ini file. .. versionadded:: 0.7.0 :param attributes: optional dictionary of arbitrary Python keys/values, which will be populated into the :attr:`.Config.attributes` dictionary. .. versionadded:: 0.7.5 .. seealso:: :ref:`connection_sharing` N�alembicc C s< || _ || _|| _|| _|| _t|�| _|r8| j�|� dS )z*Construct a new :class:`.Config` N) �config_file_name�config_ini_section� output_buffer�stdout�cmd_opts�dict� config_args� attributes�update)�self�file_� ini_sectionr r r r r � r �?/opt/alt/python37/lib/python3.7/site-packages/alembic/config.py�__init__U s zConfig.__init__c C s i S )a A 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. .. versionadded:: 0.7.5 .. seealso:: :ref:`connection_sharing` :paramref:`.Config.attributes` r )r r r r r } s zConfig.attributesc G s t �| jt�|�| d� dS )z!Render a message to standard out.� N)r Zwrite_outstreamr r Z text_type)r �text�argr r r � print_stdout� s zConfig.print_stdoutc C s\ | j r tj�tj�| j ��}nd}|| jd< t| j�}| j rL|�| j g� n |�| j � |S )a Return 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 �os�path�abspath�dirnamer r �read� add_sectionr )r r � file_configr r r r& � s zConfig.file_configc C s t j�td�S )z�Return the directory where Alembic setup templates are found. This method is used by the alembic ``init`` and ``list_templates`` commands. Z templates)r r! �joinr )r r r r �get_template_directory� s z Config.get_template_directoryc C s t | j�|��S )zfReturn all the configuration options from a given .ini file section as a dictionary. )r r& �items)r �namer r r � get_section� s zConfig.get_sectionc C s | j �| j||� dS )zzSet an option programmatically within the 'main' section. This overrides whatever was in the .ini file. N)r& �setr )r r* �valuer r r �set_main_option� s zConfig.set_main_optionc C s | j �| j|� d S )N)r& � remove_optionr )r r* r r r �remove_main_option� s zConfig.remove_main_optionc C s, | j �|�s| j �|� | j �|||� dS )z�Set 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. N)r& � has_sectionr% r, )r �sectionr* r- r r r �set_section_option� s zConfig.set_section_optionc C sD | j �|�s t�d| j|f ��| j �||�r<| j �||�S |S dS )zCReturn an option from the given section of the .ini file. z6No config file %r found, or file has no '[%s]' sectionN)r& r1 r � CommandErrorr � has_option�get)r r2 r* �defaultr r r �get_section_option� s zConfig.get_section_optionc C s | � | j||�S )z�Return 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. )r8 r )r r* r7 r r r �get_main_option� s zConfig.get_main_option)N)N)�__name__� __module__� __qualname__�__doc__�sysr r Z immutabledictr r r r Zmemoized_propertyr r r& r( r+ r. r0 r3 r8 r9 r r r r r s" F r c @ s0 e Zd Zd dd�Zdd� Zdd� Zd dd �ZdS ) � CommandLineNc C s | � |� d S )N)�_generate_args)r �progr r r r � s zCommandLine.__init__c s: � fdd�}t |d�}|jddtddd� |jd d td d d� |jd ddd� |jdddd� |�� }x�dd� tt�D �D ]�}t�|�rz|jd dkrz|j dkrzt� |�}|d r�|d dt |d � � }|d t |d � d � }n|d dd � }g }|j |j|j d�� |� ||� � j|||fd � qzW || _d S ) Nc sx ddt dtdd�fddt tdd �fd t d d d �fdt tdd �fdt tdd �fdt d dd �fdt ddd �fdt tdd �fdt tdd �fdt td d �fd d t d d d �fd t d d!d �fd"t d d#d �fd$t d d%d �fd&d't d(d)d �fd*�}d+d,d-d.�}x>|D ]6}||kr�|| }|d/d0� |d0 }}| j||� q�W xD|D ]<}|d1k�rZ� j|d2|�|�d3� n� j||�|�d4� �q4W d S )5Nz-tz --templateZgenericz"Setup template for use with 'init')r7 �type�helpz-mz --messagez%Message string to use with 'revision')rB rC z--sql� store_truezADon't emit SQL to database - dump to standard output/file instead)�actionrC z--tagz<Arbitrary 'tag' name - can be used by custom env.py scripts.z--headzCSpecify head revision or <branchname>@head to base new revision on.z--splicez6Allow a non-head revision as the 'head' to splice ontoz --depends-on�appendzNSpecify one or more revision identifiers which this revision should depend on.z--rev-idz9Specify a hardcoded revision id instead of generating onez--version-pathz2Specify specific path from config for version filez--branch-labelz3Specify a branch label to apply to the new revisionz-vz --verbosezUse more verbose outputz--resolve-dependenciesz+Treat dependency versions as down revisionsz--autogeneratezgPopulate revision script with candidate migration operations, based on comparison of database to model.z --head-onlyz0Deprecated. Use --verbose for additional outputz-rz --rev-range�storez1Specify a revision range; format is [start]:[end])�template�messageZsql�tag�headZspliceZ depends_onZrev_idZ version_pathZ branch_label�verboseZresolve_dependenciesZ autogenerateZ head_onlyZ rev_rangez location of scripts directoryzrevision identifierz/one or more revisions, or 'heads' for all heads)� directory�revision� revisionsr ���rO �+)�nargsrC )rC )r �str� add_argumentr6 )�parser� positional�kwargsZ kwargs_optsZpositional_helpr �args�kw)� subparserr r � add_options� s� z/CommandLine._generate_args.<locals>.add_options)rA z-cz--configz alembic.inizAlternate config file)rB r7 rC z-nz--namer z6Name of section in .ini file to use for Alembic configz-xrF zlAdditional arguments consumed by custom env.py scripts, e.g. -x setting1=somesetting -x setting2=somesetting)rE rC z --raiseerrrD z!Raise a full stack trace on errorc S s g | ]}t t|��qS r )�getattrr )�.0�nr r r � <listcomp>� s z.CommandLine._generate_args.<locals>.<listcomp>r �_zalembic.command� r )rC )�cmd)r rT rS �add_subparsers�dirr �inspect� isfunctionr: r; � getargspec�len� add_parserr= � set_defaultsrU ) r rA r[ rU Z subparsers�fn�specrV �kwargr )rZ r r@ � s@ zCommandLine._generate_argsc s� � j \}}}y4||f� fdd�|D ��t� fdd�|D ��� W n< tjk r| } z � jr^� nt�t|�� W d d }~X Y nX d S )Nc s g | ]}t � |��qS r )r\ )r] �k)�optionsr r r_ � s z'CommandLine.run_cmd.<locals>.<listcomp>c 3 s | ]}|t � |�fV qd S )N)r\ )r] rn )ro r r � <genexpr>� s z&CommandLine.run_cmd.<locals>.<genexpr>)rb r r r4 Zraiseerr�errrS )r �configro rk rV rm �er )ro r �run_cmd� s zCommandLine.run_cmdc C sF | j �|�}t|d�s$| j �d� n t|j|j|d�}| �||� d S )Nrb ztoo few arguments)r r r )rU � parse_args�hasattr�errorr rr r* rt )r �argvro Zcfgr r r �main� s zCommandLine.main)N)N)r: r; r< r r@ rt ry r r r r r? � s /r? c K s t |d�j| d� dS )z(The console runner function for Alembic.)rA )rx N)r? ry )rx rA rW r r r ry � s ry �__main__)NN)�argparser Z util.compatr re r r> r r r r r �objectr r? ry r: r r r r �<module> s f N