a g gt@sUddlmZddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z d d l m Z d d l mZd ZiZd ed<dddddZdddddddZddddddZddddd d!Zed"d'dd d$dd%d&d"ZdS)() annotationsN)Any)Callable)Dict)List)Optional)Union)util)compatZREVISION_SCRIPT_FILENAMEdict _registrystrr)namereturncsfdd}|S)zA 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` cs |t<|S)N)r )fnr]/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/alembic/script/write_hooks.pydecorate$szregister..decorater)rrrrrregisters rzDict[str, Union[str, int]]r)rrevisionoptionsrc CsPz t|}Wn4ty@}ztd||WYd}~nd}~00|||SdS)a7Invokes 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)rrrhookkerrr_invoke+s  rzDict[str, str]None)path hook_configrc sddlm}|dd}|D]s,q"fddD}|d<z |d}Wn8ty}z td f|WYd }~q"d }~00tjd t|||d d q"d S)z&Invoke hooks for a generated revision.)_split_on_space_commahookscs4i|],}|dr|tdd|qS).r!N) startswithlen).0keyr rrr Jsz_run_hooks.. _hook_nametypez.Key %s.type is required for post write hook %rNzRunning post write hook "%s"T)newline) baser"splitgetrr rstatusr)rr r"namesoptstype_rrr*r _run_hooks@s2    r6z List[str])cmdline_options_strrrcs:t|vrtd|}tj|tjd}fdd|D}|S)zParse 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.  )posixcsg|]}|tqSr)replaceREVISION_SCRIPT_TOKEN)r(optionrrr osz*_parse_cmdline_options..)r;shlexr0r Zis_posix)r7rcmdline_options_listrr=r_parse_cmdline_optionsas   rAconsole_scriptsFbool)rr ignore_outputrc Csz |d}Wn@tyL}z(td|d|df|WYd}~n d}~00tdD]}|j|krX|}qqXtd||dd}|dd}t||} i} |rtj | d <| d <tj t j d d |j |j |jfg| fd|i| dS) NZ entrypointz4Key %s.entrypoint is required for post write hook %rr,rBz*Could not find entrypoint console_scripts.cwdrr$stdoutstderrz-czimport %s; %s.%s())rr rr Zimportlib_metadata_getrr1rA subprocessDEVNULLrunsys executablemoduleattr) rrrDZentrypoint_namerentryimplrEr7r@kwrrrrBvsF     )F) __future__rr?rHrKtypingrrrrrrr$r r r;r __annotations__rrr6rArBrrrrs(         !