관리-도구
편집 파일: write_hooks.cpython-38.pyc
U 0?�f� � @ s* U d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d d lm Z d d l mZ d d lmZ d Zi Zded< ddd�dd�Zddddd�dd�Zdddd �d d �Zddd d �d!d"�Zed#�d+ddd%dd&�d'd#��Zed(�d,ddd%dd&�d)d*��ZdS )-� )� annotationsN)�Any)�Callable)�Dict)�List)�Mapping)�Optional)�Union� )�util)�compatZREVISION_SCRIPT_FILENAME�dict� _registry�strr )�name�returnc s � fdd�}|S )z�A function decorator that will register that function as a write hook. See the documentation linked below for an example. .. seealso:: :ref:`post_write_hooks_custom` c s | t � < | S )N)r )�fn�r � �H/opt/hc_python/lib/python3.8/site-packages/alembic/script/write_hooks.py�decorate& s zregister.<locals>.decorater )r r r r r �register s r z Mapping[str, Union[str, int]]r )r �revision�optionsr c C sR z t | }W n6 tk rB } zt�d| � d��|�W 5 d}~X Y n X |||�S dS )a7 Invokes the formatter registered for the given name. :param name: The name of a formatter in the registry :param revision: A :class:`.MigrationRevision` instance :param options: A dict containing kwargs passed to the specified formatter. :raises: :class:`alembic.util.CommandError` zNo formatter with name 'z ' registeredN)r �KeyErrorr � CommandError)r r r �hook�ker r r �_invoke- s ��r zMapping[str, str]�None)�path� hook_configr c s� ddl m} |�� �dd��}|D ]���s,q"� �fdd�� D �}�|d< z |d }W n: tk r� } z t�d �� d ����|�W 5 d }~X Y q"X tjd ���d d�� t|| |� W 5 Q R X q"d S )z&Invoke hooks for a generated revision.� )�_split_on_space_comma�hooks� c s4 i | ],}|� �d �r|t��d d� � | �qS )�.r" N)� startswith�len)�.0�key�r! r r r � <dictcomp>L s � z _run_hooks.<locals>.<dictcomp>� _hook_name�type�Key z&.type is required for post write hook NzRunning post write hook T)�newline) �baser# �split�getr r r �statusr )r r! r# �names�opts�type_r r r+ r � _run_hooksB s, � �� �r8 z List[str])�cmdline_options_strr r c s: t | krt d | } tj| tjd�}� fdd�|D �}|S )z�Parse options from a string into a list. Also substitutes the revision script token with the actual filename of the revision script. If the revision script token doesn't occur in the options string, it is automatically prepended. � )�posixc s g | ]}|� t� ��qS r )�replace�REVISION_SCRIPT_TOKEN)r) �option�r r r � <listcomp>m s �z*_parse_cmdline_options.<locals>.<listcomp>)r= �shlexr2 r Zis_posix)r9 r �cmdline_options_listr r? r �_parse_cmdline_options_ s � �rC �console_scriptsF�bool)r r � ignore_outputr c C s� z |d }W nB t k rN } z$t�d|d � d|d ���|�W 5 d }~X Y nX t�d�D ]}|j|krZ|} q�qZt�d|� ���|�dd �}|�dd �}t|| �} i } |r�tj | d < | d < tj t j d d |j � d|j � d|j� d�g| fd|i| �� d S )N� entrypointr/ r- z,.entrypoint is required for post write hook rD z*Could not find entrypoint console_scripts.�cwdr r% �stdout�stderrz-czimport z; r& z())r r r r Zimportlib_metadata_getr r3 rC � subprocess�DEVNULL�run�sys� executable�module�attr) r r rF Zentrypoint_namer �entry�implrH r9 rB �kwr r r rD t sB �� � ������execc C s� z |d }W nB t k rN } z$t�d|d � d|d ���|�W 5 d }~X Y nX |�dd �}|�dd�}t|| �}i }|r�tj |d< |d < tj|f|�fd|i|�� d S ) NrO r/ r- z,.executable is required for post write hook rH r r% rI rJ )r r r r3 rC rK rL rM ) r r rF rO r rH r9 rB rT r r r �exec_� s0 �� �����rV )F)F)� __future__r rA rK rN �typingr r r r r r r r% r r r= r �__annotations__r r r8 rC rD rV r r r r �<module> s0 �&