o 8h@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)PostWriteHookConfigZREVISION_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)fnrb/var/www/html/figdemos/bartoux_crm/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#rrtype_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..)r1shlexsplitr is_posix)r,r"cmdline_options_listrr4r_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?r@r,r9kwrrr _run_hook~s   "rIconsole_scriptsFc Cstt|d}tdD] }|j|kr|}n q td|tjdd|jd|jd|j dg}t ||||dS) N entrypointrJz*Could not find entrypoint console_scripts.z-czimport z; r;z()) r<r importlib_metadata_getrr rsys executablemoduleattrrI)r"rr=Zentrypoint_nameentryimplr?rrrrJs  execcCs t|d}t||||gddS)NrN)r?)r<rI)r"rr=rNrrrexec_s rTrOcCsHt|d}tj|durtd|tjd|g}t||||dS)NrOzCould not find module z-m)r< importlibr find_specrrMrNrI)r"rr= module_namer?rrrrOs  )rrrr)rrrrrrrr)r"rr#r$rr%)r,rr"rrr-)rrrrrr) r"rrrr=r>r?r-rr%)F)r"rrrr=r>rr%)# __future__rimportlib.utilrUosr6rErMtypingrrrrrrr rAr r util.pyfilesr configrr1r__annotations__rr!r+r:r<rIrJrTrOrrrrsB