Edit File: 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)�config�tempname�readmeZsynopsis� 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 � directory�templateZ template_dirr �scriptZfile_� file_path� config_file� 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/ )r0 r r r1 d s ziUsing --sql with the revision command when revision_environment is not configured does not make any sense)�fn�as_sql� template_argsr0 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 r4 �run_envZgenerate_scripts�len)r r' r r( r& r) r* r+ r, r- Zscript_directoryZ command_args� environmentr1 �scriptsr )r0 r �revisionD sF r> 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 r7 Zgenerate_revisionr r, )r Z revisionsr' r* r, r! r4 r r r �merge� s r? 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/ )r> r! r r �upgrade� s zupgrade.<locals>.upgrade)r2 r3 � starting_rev�destination_rev�tag)r r7 r r �splitr r: )r r> r( rE rC rB r )r> r! r rB � s rB 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.Nr@ zRange revision not allowedrA z/downgrade with --sql requires <fromrev>:<torev>c s �� � | �S )N)Z_downgrade_revs)r. r/ )r> r! r r � downgrade� s z downgrade.<locals>.downgrade)r2 r3 rC rD rE )r r7 r r rF r r: )r r> r( rE rC rG r )r> r! r rG � s& rG 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)r2 N)r r7 r r: rI r rJ )r r. rL rK r )r r! r �show� s rM 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.Nr@ z9History range requires [start]:[end], [start]:, or :[end]c s< x6|j |p d|pdd�D ] }| �|j� dddd�� qW d S )N�base�heads)rN r& T)�verbose�include_branches� include_docZinclude_parents)�walk_revisionsr � cmd_format)r r! rN r& rK )rP 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/ )rU rN r r& r! r r �_display_current_history� s zMhistory.<locals>._display_history_w_current.<locals>._display_current_history)r2 )r r: )r r! rN r& rV )rU )rN r r& r! r �_display_history_w_current� s z+history.<locals>._display_history_w_currentrH )r& )rN )NN)r r7 r r �striprF )r Z rev_rangerP r! rN r& rW r )rU rP r �history� s rY c C sP t �| �}|r|�d�}n|�|�� �}x"|D ]}| �|j|ddd�� q.W dS )z4Show current available heads in the script directoryrO TF)rQ Ztree_indicatorsN)r r7 rI Z get_headsr rT )r rP Zresolve_dependenciesr! rO r. r r r rO s rO c sd t �| ��xT��� D ]H� � jr| �d� j�dd�d�� �fdd��fdd�� jD �D ��� qW dS ) zShow current branch pointsz%s %s T)rQ � c 3 s4 | ],}d dt t� j�� |jdd�d�f V qdS )z%s -> %s� FT)rQ rR N)r; �strr> rT )r5 Zrev_obj)rK rP r r � <genexpr>' s zbranches.<locals>.<genexpr>c 3 s | ]}� � |�V qd S )N)Z get_revision)r5 r. )r! r r r] , s N)r r7 rS Zis_branch_pointr rT r Znextrev)r rP r )rK r! rP r �branches s r^ 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 connectionZengine�urlrI rT )r. r/ )r r! rP r r �display_version9 s z current.<locals>.display_version)r2 N)r r7 r �warnr r: )r rP Z head_onlyr` r )r r! rP r rH 1 s rH 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.Nr@ zRange revision not allowedrA c s �� � | �S )N)Z _stamp_revs)r. r/ )r> r! r r �do_stampW s zstamp.<locals>.do_stamp)r2 r3 rD rC rE )r r7 r r rF r r: )r r> r( rE rC rb r )r> r! r �stampK s rc 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 $EDITORrH c s2 | st �d��x � �| �D ]}t �|j� qW g S )NzNo current revisions)r r rI �editr )r. r/ rK )r! r r � edit_currentl s zedit.<locals>.edit_current)r2 Nz*No revision files indicated by symbol '%s') r r7 r r: rI r r rd r )r r. re ZrevsrK r )r! r rd f s rd )r ) NFFr& FNNNN)NNN)FN)FN)NF)FF)F)FF)FN)r r! r Zruntime.environmentr � r r r9 r r% r> r? rB rG rM rY rO r^ rH rc rd r r r r �<module> s& , : -