Current File : //opt/alt/python37/lib/python3.7/site-packages/alembic/__pycache__/command.cpython-37.pyc
B
�M!V - � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZ dd� Z d$d
d �Z
d%dd�Z d&dd�Z d'dd�Z
d(dd�Zdd� Zd)dd�Zd*dd�Zd+d d �Zd,d d �Zd-d d!�Zd"d#� ZdS ).� N� )�ScriptDirectory)�EnvironmentContext)�util)� autogeneratec
C st | � d� xPt�| �� �D ]>}ttj�| �� |d���}t|�}W dQ R X | � d||� qW | � d� | � d� dS )zList available templateszAvailable templates:
ZREADMENz%s - %sz1
Templates are used via the 'init' command, e.g.:z,
alembic init --template generic ./scripts)� print_stdout�os�listdir�get_template_directory�open�path�join�next)�configZtempnameZreadmeZsynopsis� r � /command.py�list_templates s
r �genericc
C s8 t �|t j�r t�d| ��t j�| �� |�}t �|t j�sJt�d| ��t�dt j� |� t j
|� t j�|d�}t�dt j� |� t j
|� t |�}x�t � |�D ]~}t j�||�}|dkr�t j� | j
�}t �|t j�r�t�d| � n|j|||d� q�t j�|�r�t j�||�} |�|| � q�W t�d| � d S )
z#Initialize a new scripts directory.zDirectory %s already existszNo such template %rzCreating directory %s�versionszalembic.ini.makoz File %s already exists, skipping)Zscript_locationzNPlease edit configuration/connection/logging settings in %r before proceeding.N)r �access�F_OKr � CommandErrorr r
r
Zstatus�abspath�makedirsr r Zconfig_file_name�msgZ_generate_template�isfileZ
_copy_file)
r Z directory�templateZ template_dirr �scriptZfile_Z file_pathZ config_fileZ output_filer r r �init s:
r F�headc
s� t �| �}
t||||||||| d� } t�| |
| �� t�| �d��} |rfd} |rXt�d��� fdd�}
n | rx� fdd�}
n|r�t�d��| r�t | |
|
|� j
� d �� |
� � W d
Q R X d d � � � � D �}t
|�d
kr�|d S |S d
S )zCreate a new revision file.) �messager �sqlr �splice� branch_label� version_path�rev_id�
depends_onZrevision_environmentTz7Using --sql with --autogenerate does not make any sensec s � � | |� g S )N)Zrun_autogenerate)�rev�context)�revision_contextr r �retrieve_migrations` s z%revision.<locals>.retrieve_migrationsc s � � | |� g S )N)Zrun_no_autogenerate)r' r( )r) r r r* d s ziUsing --sql with the revision command when revision_environment is not configured does not make any sense)�fn�as_sql�
template_argsr) Nc S s g | ]}|�qS r r )�.0r r r r �
<listcomp>x s z revision.<locals>.<listcomp>r r )r � from_config�dict�autogenZRevisionContextr ZasboolZget_main_optionr r r- �run_envZgenerate_scripts�len)r r r r! r r" r# r$ r% r& Zscript_directoryZ command_argsZ environmentr* Zscriptsr )r) r �revisionD sF
r5 c C s6 t �| �}d| i}|j|p t�� |fd||d�|��S )z�Merge two revisions together. Creates a new migration file.
.. versionadded:: 0.7.0
.. seealso::
:ref:`branches`
r T)Zrefreshr Z
branch_labels)r r0 Zgenerate_revisionr r% )r Z revisionsr r# r% r r- r r r �merge� s
r6 c sp t �| ��d}d� kr4|s$t�d��� �dd�\}� � �fdd�}t| �|||� |d�� ��� W dQ R X dS )zUpgrade to a later version.N�:zRange revision not allowed� c s �� � | �S )N)Z
_upgrade_revs)r' r( )r5 r r r �upgrade� s zupgrade.<locals>.upgrade)r+ r, � starting_rev�destination_rev�tag)r r0 r r �splitr r3 )r r5 r! r< r: r9 r )r5 r r r9 � s
r9 c s� t �| ��d}d� kr6|s$t�d��� �dd�\}� n|rDt�d��� �fdd�}t| �|||� |d�� ��� W dQ R X dS ) z Revert to a previous version.Nr7 zRange revision not allowedr8 z/downgrade with --sql requires <fromrev>:<torev>c s �� � | �S )N)Z_downgrade_revs)r' r( )r5 r r r � downgrade� s z downgrade.<locals>.downgrade)r+ r, r: r; r< )r r0 r r r= r r3 )r r5 r! r< r: r> r )r5 r r r> � s&
r> c sh t �� ��|dkrD� �fdd�}t� �|d�� ��� W dQ R X n x ��|�D ]}� �|j� qPW dS )z1Show the revision(s) denoted by the given symbol.�currentc s$ x �� | �D ]}� �|j� q W g S )N)�
get_revisionsr � log_entry)r' r( �sc)r r r r � show_current� s zshow.<locals>.show_current)r+ N)r r0 r r3 r@ r rA )r r' rC rB r )r r r �show� s
rD c s� t �| �}|dk r8d|kr$t�d��|�� �d�\}}nd }}�fdd�� d � fdd� }|dkrr|| ||d � n&|dkr�|| ||d
� n� | |||� dS ) z.List changeset scripts in chronological order.Nr7 z9History range requires [start]:[end], [start]:, or :[end]c s< x6|j |p d|pdd�D ] }| �|j� dddd�� qW d S )N�base�heads)rE r T)�verbose�include_branches� include_docZinclude_parents)�walk_revisionsr �
cmd_format)r r rE r rB )rG r r �_display_history� s z!history.<locals>._display_historyc s: �� ���fdd�}t ��|d�� ��� W d Q R X d S )Nc s2 �d kr� ���| � n�d kr.� ��| �� g S )Nr )r' r( )rL rE r r r r r �_display_current_history� s
zMhistory.<locals>._display_history_w_current.<locals>._display_current_history)r+ )r r3 )r r rE r rM )rL )rE r r r r �_display_history_w_current� s
z+history.<locals>._display_history_w_currentr? )r )rE )NN)r r0 r r �stripr= )r Z rev_rangerG r rE r rN r )rL rG r �history� s
rP c C sP t �| �}|r|�d�}n|�|�� �}x"|D ]}| �|j|ddd�� q.W dS )z4Show current available heads in the script directoryrF TF)rH Ztree_indicatorsN)r r0 r@ Z get_headsr rK )r rG Zresolve_dependenciesr rF r' r r r rF s
rF c sd t �| ��xT��� D ]H� � jr| �d� j�dd�d�� �fdd��fdd�� jD �D ��� qW dS ) zShow current branch pointsz%s
%s
T)rH �
c 3 s4 | ],}d dt t� j�� |jdd�d�f V qdS )z%s -> %s� FT)rH rI N)r4 �strr5 rK )r. Zrev_obj)rB rG r r � <genexpr>' s zbranches.<locals>.<genexpr>c 3 s | ]}� � |�V qd S )N)Z get_revision)r. r' )r r r rT , s N)r r0 rJ Zis_branch_pointr rK r
Znextrev)r rG r )rB r rG r �branches s
rU c sN t �� ��|rt�d� � ��fdd�}t� �|d�� ��� W dQ R X dS )z,Display the current revision for a database.z--head-only is deprecatedc sD �r � � dt�|jjj�� x"��| �D ]} � � | ���� q(W g S )NzCurrent revision(s) for %s:)r r Zobfuscate_url_pwZ
connectionZengineZurlr@ rK )r' r( )r r rG r r �display_version9 s z current.<locals>.display_version)r+ N)r r0 r �warnr r3 )r rG Z head_onlyrV r )r r rG r r? 1 s
r? c sp t �| ��d}d� kr4|s$t�d��� �dd�\}� � �fdd�}t| �||� ||d�� ��� W dQ R X dS )zQ'stamp' the revision table with the given revision; don't
run any migrations.Nr7 zRange revision not allowedr8 c s �� � | �S )N)Z _stamp_revs)r' r( )r5 r r r �do_stampW s zstamp.<locals>.do_stamp)r+ r, r; r: r< )r r0 r r r= r r3 )r r5 r! r< r: rX r )r5 r r �stampK s
rY c s| t �| �� |dkrB� fdd�}t| � |d�� � �� W dQ R X n6� �|�}|s^t�d| ��x|D ]}t�|j� qdW dS )z%Edit revision script(s) using $EDITORr? c s2 | st �d��x � �| �D ]}t �|j� qW g S )NzNo current revisions)r r r@ �editr )r' r( rB )r r r � edit_currentl s
zedit.<locals>.edit_current)r+ Nz*No revision files indicated by symbol '%s') r r0 r r3 r@ r r rZ r )r r' r[ ZrevsrB r )r r rZ f s
rZ )r ) NFFr FNNNN)NNN)FN)FN)NF)FF)F)FF)FN)r r r Zruntime.environmentr � r r r2 r r r5 r6 r9 r> rD rP rF rU r? rY rZ r r r r �<module> s&
,
:
-
Mini Shell