�
f��g� � �( � 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< dd�Z dd�Zdd�Zdd�Z ed� d d d�� Z ed� d d d�� Zy) � )� annotationsN)�Any)�Callable)�Dict)�List)�Mapping)�Optional)�Union� )�util)�compat�REVISION_SCRIPT_FILENAME�dict� _registryc � � � � fd�}|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 � �� | t �< | S )N)r )�fn�names ��I/opt/hc_python/lib/python3.12/site-packages/alembic/script/write_hooks.py�decoratezregister.<locals>.decorate& s �� � � �$��� � � )r r s ` r �registerr s �� �� �Or c �~ � t | } |||� S # t $ r }t j d| � d�� |�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`
zNo formatter with name 'z ' registeredN)r �KeyErrorr � CommandError)r �revision�options�hook�kes r �_invoker! - sS � �'����� �H�g�&�&�� � �� � �&�t�f�L� 9�
�� ���s � � <�7�<c �� � ddl m} |j |j dd� � }|D ]v }|s�|D �ci c]+ }|j |dz � r|t |� dz d || ��- }}||d< |d }t
j d |��d
� � 5 t || |� ddd� �x yc c}w # 1 sw Y ��xY w# t $ r!}t
j d |� d
|��� |�d}~ww xY w)z&Invoke hooks for a generated revision.� )�_split_on_space_comma�hooks� �.N�
_hook_name�typezRunning post write hook T)�newline�Key z&.type is required for post write hook ) �baser$ �split�get�
startswith�lenr �statusr! r r ) �path� hook_configr$ �namesr �key�opts�type_r s r �
_run_hooksr8 B s � � ,� !� '� '� �����(D� E�E���� � #�
�"���~�~�d�S�j�)�
��D� �A�
�� �+�c�"2� 2�"�
�
�
"��\��
+���L�E� ���*�4�(�3�T�� ��t�T�*�� �! ��
�� �� � ��#�#��t�f� B�4�(�K���
�� �s) �0B$�,B5� B)�)B2 �5 C �> C�C c �� � t | vr t dz | z } t j | t j �� }|D �cg c] }|j t |� �� }}|S c c}w )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.
� )�posix)�REVISION_SCRIPT_TOKEN�shlexr- r
�is_posix�replace)�cmdline_options_strr2 �cmdline_options_list�options r �_parse_cmdline_optionsrC _ sq � � �$7�7� 3�c� 9�<O� O�� �;�;��6�?�?� ��
+� �*�F� ���,�d�3�*� � � � �� s � A �console_scriptsc �@ � |d }t j d� D ] }|j
|k( s�|} n t j d|� �� �|j
dd � }|j
dd � }t || � } i }
|rt j x|
d
< |
d < t j t j d d
|j � d|j � d|j � d�g| z fd|i|
�� y # t $ r'}t j d|d � d|d ��� |�d }~ww xY w)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
�importlib_metadata_getr r. rC �
subprocess�DEVNULL�run�sys�
executable�module�attr) r2 r �
ignore_output�entrypoint_namer �entry�implrG r@ rA �kws r rD rD t sX � ��!�,�/�� �.�.�/@�A�� �:�:� � (� �D� � B�
� � �8� �8I� J�
�
� !���U�D�1�C�!�+�+�i��4��1�2E�t�L���B��&0�&8�&8�8��8� �r�(�|��N�N� �N�N� ��d�k�k�]�"�T�[�[�M��4�9�9�+�R� @�
�
� � �
� �
� ��+ � �� � ��7�<�(�)� *�!�,�/�2�
4�
� � ���s �C- �- D �6"D�D �execc �D � |d }|j dd � }|j dd� }t || � }i }|rt
j x|d< |d < t j |g|�fd|i|�� y # t $ r'}t j d|d � d|d ��� |�d }~ww xY w)
NrO r+ r( z,.executable is required for post write hook rG r r& rH rI )r r r r. rC rK rL rM ) r2 r rR rO r rG r@ rA rV s r �exec_rY � s� � �� �\�*�
� !���U�D�1�C�!�+�+�i��4��1�2E�t�L���B��&0�&8�&8�8��8� �r�(�|��N�N� �
�
!�
��
� �
�
�� � �� � ��7�<�(�)� *�!�,�/�2�
4�
� � ���s �A/ �/ B �8"B�B )r �str�returnr )r rZ r rZ r z Mapping[str, Union[str, int]]r[ r )r2 rZ r3 zMapping[str, str]r[ �None)r@ rZ r2 rZ r[ z List[str])F)r2 rZ r r rR �boolr[ r\ )�
__future__r r= rK rN �typingr r r r r r r
r& r r
r< r �__annotations__r r! r8 rC rD rY r r r �<module>ra s� �� #� � �
� � � � � � � � � � 3� �� �4� ��&'�
�'� �'�'D�'��'�*+�: �*
�
� �49�#�
�#� �#�-1�#� �#� �#�L
�&��� �r