o Ph@sLUddlmZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddlm Z ddlm Z ddlmZdd lmZd d lmZd d lmZd d lmZer_d dlmZdZiZded<d8ddZd9ddZd:d#d$Zd;d'd(Zdd?d2d0Z ed3d>d?d4d5Z!ed6d>d?d7d6Z"dS)@) 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)fnr_/var/www/html/stock_analysis/be/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_invoke2s    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_hooksJs*  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 ms z*_parse_cmdline_options..)r2shlexsplitr is_posix)r-r#cmdline_options_listrr5r_parse_cmdline_options_s  r;c CsJz||WSty$}ztd|dd|d|d|d}~ww)NzKey r'.z! is required for post write hook )rr r)rrr!rrr_get_required_optionts r= ignore_outputboolcommandcCs^|dd}|dd}t||}i}|rtj|d<|d<tjg||fd|i|dS)Ncwdrstdoutstderr)getr; subprocessDEVNULLrun)r#rr>r@rAr-r:kwrrr _run_hook~s   "rJconsole_scriptsFc Cstt|d}tdD] }|j|kr|}n q td|tjdd|jd|jd|j dg}t ||||dS) N entrypointrKz*Could not find entrypoint console_scripts.z-czimport z; r<z()) r=r importlib_metadata_getrr rsys executablemoduleattrrJ)r#rr>entrypoint_nameentryimplr@rrrrKs  execcCs t|d}t||||gddS)NrO)r@)r=rJ)r#rr>rOrrrexec_s rVrPcCsHt|d}tj|durtd|tjd|g}t||||dS)NrPzCould not find module z-m)r= importlibr find_specrrNrOrJ)r#rr> module_namer@rrrrPs  )rrrr)rrrrrrrr)r#rr$r%rr&)r-rr#rrr.)rrrrrr) r#rrrr>r?r@r.rr&)F)r#rrrr>r?rr&)# __future__rimportlib.utilrWosr7rFrNtypingrrrrrrr rBr r util.pyfilesr configrr2r__annotations__rr"r,r;r=rJrKrVrPrrrrsB