Edit File: command.cpython-37.opt-1.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 �@/opt/alt/python37/lib/python3.7/site-packages/alembic/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 �status�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 r6 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 r1 Zgenerate_revisionr r& )r Z revisionsr! r$ r& r r. r r r �merge� s r7 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) )r6 r r r �upgrade� s zupgrade.<locals>.upgrade)r, r- � starting_rev�destination_rev�tag)r r1 r r �splitr r4 )r r6 r" r= r; r: r )r6 r r r: � s r: 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.Nr8 zRange revision not allowedr9 z/downgrade with --sql requires <fromrev>:<torev>c s �� � | �S )N)Z_downgrade_revs)r( r) )r6 r r r � downgrade� s z downgrade.<locals>.downgrade)r, r- r; r< r= )r r1 r r r> r r4 )r r6 r" r= r; r? r )r6 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 r1 r r4 rA r rB )r r( rD rC r )r r r �show� s rE 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.Nr8 z9History range requires [start]:[end], [start]:, or :[end]c s< x6|j |p d|pdd�D ] }| �|j� dddd�� qW d S )N�base�heads)rF r T)�verbose�include_branches� include_docZinclude_parents)�walk_revisionsr � cmd_format)r r rF r rC )rH 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) )rM rF r r r r r �_display_current_history� s zMhistory.<locals>._display_history_w_current.<locals>._display_current_history)r, )r r4 )r r rF r rN )rM )rF r r r r �_display_history_w_current� s z+history.<locals>._display_history_w_currentr@ )r )rF )NN)r r1 r r �stripr> )r Z rev_rangerH r rF r rO r )rM rH r �history� s rQ c C sP t �| �}|r|�d�}n|�|�� �}x"|D ]}| �|j|ddd�� q.W dS )z4Show current available heads in the script directoryrG TF)rI Ztree_indicatorsN)r r1 rA Z get_headsr rL )r rH Zresolve_dependenciesr rG r( r r r rG s rG c sd t �| ��xT��� D ]H� � jr| �d� j�dd�d�� �fdd��fdd�� jD �D ��� qW dS ) zShow current branch pointsz%s %s T)rI � c 3 s4 | ],}d dt t� j�� |jdd�d�f V qdS )z%s -> %s� FT)rI rJ N)r5 �strr6 rL )r/ Zrev_obj)rC rH r r � <genexpr>' s zbranches.<locals>.<genexpr>c 3 s | ]}� � |�V qd S )N)Z get_revision)r/ r( )r r r rU , s N)r r1 rK Zis_branch_pointr rL r Znextrev)r rH r )rC r rH r �branches s rV 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 connectionZengineZurlrA rL )r( r) )r r rH r r �display_version9 s z current.<locals>.display_version)r, N)r r1 r �warnr r4 )r rH Z head_onlyrW r )r r rH 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.Nr8 zRange revision not allowedr9 c s �� � | �S )N)Z _stamp_revs)r( r) )r6 r r r �do_stampW s zstamp.<locals>.do_stamp)r, r- r< r; r= )r r1 r r r> r r4 )r r6 r" r= r; rY r )r6 r r �stampK s rZ 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 rA �editr )r( r) rC )r r r � edit_currentl s zedit.<locals>.edit_current)r, Nz*No revision files indicated by symbol '%s') r r1 r r4 rA r r r[ r )r r( r\ ZrevsrC r )r r r[ f s r[ )r ) NFFr FNNNN)NNN)FN)FN)NF)FF)F)FF)FN)r r r Zruntime.environmentr � r r r3 r r r6 r7 r: r? rE rQ rG rV r@ rZ r[ r r r r �<module> s& , : -