Edit File: write_hooks.cpython-311.pyc
� v�d8 � � � 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< d'd�Zd(d�Zd)d �Zd*d!�Z ed"� � d+d,d&�� � ZdS )-� )� annotationsN)�Any)�Callable)�Dict)�List)�Mapping)�Optional)�Union� )�util)�compat�REVISION_SCRIPT_FILENAME�dict� _registry�name�str�returnr c � � � � fd�}|S )z�A function decorator that will register that function as a write hook. See the documentation linked below for an example. .. versionadded:: 1.2.0 .. seealso:: :ref:`post_write_hooks_custom` c � �� | t �< | S )N)r )�fnr s ��Q/opt/cloudlinux/venv/lib64/python3.11/site-packages/alembic/script/write_hooks.py�decoratezregister.<locals>.decorate% s �� � � �$��� � � )r r s ` r �registerr s# �� � � � � � � �Or �revision�options� Mapping[str, Union[str, int]]r c � � t | } |||� � S # t $ r }t j d| z � � |�d}~ww xY w)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` z&No formatter with name '%s' registeredN)r �KeyErrorr � CommandError)r r r �hook�kes r �_invoker$ , se � �'����� �t�H�g�&�&�&�� � � � �� � 4�t� ;� � �� ��������s � � A�=�A�path� hook_config�Mapping[str, str]�Nonec � ��� ddl m} |� �� dd� � � � }|D ]���s���fd��D � � }�|d< |d }t j dd � � � 5 t || |� � d d d � � n # 1 swxY w Y �b# t $ r }t j d ��d ���� � |�d }~ww xY wd S )z&Invoke hooks for a generated revision.� )�_split_on_space_comma�hooks� c � �� i | ];}|� �d z � � �|t �� � dz d� �| ��<S )�.r* N)� startswith�len)�.0�keyr& r s ��r � <dictcomp>z _run_hooks.<locals>.<dictcomp>K sY �� � � � ���~�~�d�S�j�)�)� � ��D� � �A� ��� �+�c�"2� � � r � _hook_name�typez Running post write hook {name!r}T)�newlineN�Key z&.type is required for post write hook ) �baser+ �split�getr �statusr$ r r! )r% r& r+ �names�opts�type_r# r s ` @r � _run_hooksr@ A sn ��� � ,�+�+�+�+�+� !� '� '� �����(D�(D� E� E�E�� +� +��� � � � � � � �"� � � �� "��\�� +���L�E� �� ?��N�N�N� +� +���t�T�*�*�*� +� +� +� +� +� +� +� +� +� +� +���� +� +� +� +��� � � � ��#�#�DH�D�D�$�$�O�� �� ����� ����+� +s* �B�,B � B �B � C � B;�;C �cmdline_options_str� List[str]c � �� t | vr t dz | z } t j | t j �� � }�fd�|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 �F �� g | ] }|� t �� � �� S r )�replace�REVISION_SCRIPT_TOKEN)r2 �optionr% s �r � <listcomp>z*_parse_cmdline_options.<locals>.<listcomp>j s8 �� � � � � � ���,�d�3�3� � � r )rH �shlexr: r �is_posix)rA r% �cmdline_options_lists ` r �_parse_cmdline_optionsrN \ sq �� � �$7�7�7� 3�c� 9�<O� O�� �;��6�?� � � �� � � � �*� � � �� � r �console_scriptsF� ignore_output�boolc �( � |d }n9# t $ r,}t j d|d �d|d ��� � |�d }~ww xY wt j d� � D ]}|j |k r|} n�t j d|� �� � �|� dd � � }|� dd � � }t || � � } i } |rt j x| d < | d < t j t j d d |j �d|j �d|j �d�g| z fd|i| �� d S )N� entrypointr8 r5 z,.entrypoint is required for post write hook rO z*Could not find entrypoint console_scripts.�cwdr r- �stdout�stderrz-czimport z; r/ z())r r r! r �importlib_metadata_getr r; rN � subprocess�DEVNULL�run�sys� executable�module�attr) r% r rP �entrypoint_namer# �entry�implrT rA rM �kws r rO rO q s� � � �!�,�/���� � � � �� � ��|�$�$�$�g�l�&;�&;� =� � � � ��������� �.�/@�A�A� � �� �:� � (� (� �D� �E� )� � � J� � J� J� � � � !���U�D�1�1�C�!�+�+�i��4�4��1�2E�t�L�L���B�� 9�&0�&8�8��8� �r�(�|��N� �N� �D�$(�K�K�K�����d�i�i�i� H� � � � � � � � � � � � � s � � A�'<�A)r r r r )r r r r r r r r )r% r r& r' r r( )rA r r% r r rB )F)r% r r r rP rQ r r( )� __future__r rK rX r[ �typingr r r r r r r r- r r rH r �__annotations__r r$ r@ rN rO r r r �<module>rf s� �� "� "� "� "� "� "� "� � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � � 3� �� � � � � �� � � �*'� '� '� '�*+� +� +� +�6 � � � �* �� � � �49�$� $� $� $� � �$� $� $r