o ODi@sUddlmZddlZddlZddlZddlZddlmZddlmZddlm Z ddlm Z ddlm Z ddlm Z dd lm Z d d lmZd d lmZd d lmZe r[d dlmZdZiZded<d1ddZd2ddZd3d#d$Zd4d'd(Zed) *d5d6d-d)Zed.d5d6d/d0ZdS)7) annotationsN)Any)Callable)Dict)List)Optional) TYPE_CHECKING)Union)util)compat)_preserving_path_as_str)PostWriteHookConfigREVISION_SCRIPT_FILENAMEdict _registrynamestrreturnrcsfdd}|S)zA 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` cs |t<|S)N)r)fnrS/var/www/Datamplify/venv/lib/python3.10/site-packages/alembic/script/write_hooks.pydecorate*szregister..decorater)rrrrrregisters r revision_pathUnion[str, os.PathLike[str]]optionsrrc CsLt|}zt|}Wnty }z td|d|d}~ww|||S)a1Invokes the formatter registered for the given name. :param name: The name of a formatter in the registry :param revision: string path to the revision file :param options: A dict containing kwargs passed to the specified formatter. :raises: :class:`alembic.util.CommandError` zNo formatter with name 'z ' registeredN)r rKeyErrorr CommandError)rrrhookkerrr_invoke1s    r"pathhookslist[PostWriteHookConfig]Nonec Cs|D]E}|d}z|d}Wnty&}z td|d||d}~wwtjd|ddt|||Wdn1sBwYqdS) z&Invoke hooks for a generated revision. _hook_nametypezKey 'z;.type' (or 'type' in toml) is required for post write hook NzRunning post write hook T)newline)rr rstatusr")r#r$r rtype_r!rrr _run_hooksIs*  r,cmdline_options_str List[str]cs:t|vr td|}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).0optionr#rr ls z*_parse_cmdline_options..)r2shlexsplitr is_posix)r-r#cmdline_options_listrr5r_parse_cmdline_options^s  r;console_scriptsF ignore_outputboolc Csz|d}Wnty"}ztd|dd|d|d}~wwtdD] }|j|kr3|}n q(td||dd}|dd }t||} i} |rZtj | d <| d <tj t j d d |j d|j d|jdg| fd|i| dS)N entrypointKey r'z,.entrypoint is required for post write hook r<z*Could not find entrypoint console_scripts.cwdrstdoutstderrz-czimport z; .z())rr rr importlib_metadata_getrgetr; subprocessDEVNULLrunsys executablemoduleattr) r#rr=entrypoint_namer!entryimplrAr-r:kwrrrr<ssL       execc Csz|d}Wnty"}ztd|dd|d|d}~ww|dd}|dd}t||}i}|rAtj|d<|d <tj|g|fd|i|dS) NrLr@r'z,.executable is required for post write hook rArrBrCrD)rr rrGr;rHrIrJ) r#rr=rLr!rAr-r:rRrrrexec_s8      rT)rrrr)rrrrrrrr)r#rr$r%rr&)r-rr#rrr.)F)r#rrrr=r>rr&) __future__rosr7rHrKtypingrrrrrrr rBr r util.pyfilesr configrr2r__annotations__rr"r,r;r<rTrrrrs8                &