← Back
Editing: write_hooks.cpython-312.pyc
� ���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
Save File
Cancel