o voi@sUddlmZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddl m Z ddl mZdd lmZe rGdd lmZd ZiZd ed <d6ddZd7ddZd8dd Zd9d#d$Zd:d%d&Zd;d*d+Zed, - dd?d2d3Zed4d>d?d5d4ZdS)@) annotationsN)Any)Callable) TYPE_CHECKING)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 )fnrN/home/ubuntu/.local/lib/python3.10/site-packages/alembic/script/write_hooks.pydecorate'szregister..decorater)rrrrrregisters r revision_pathstr | os.PathLike[str]optionsr rc 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 r KeyErrorr CommandError)rrrhookkerrr_invoke.s    rpathhookslist[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)rrrstatusr)rr rrtype_rrrr _run_hooksFs*  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).0optionrrr is z*_parse_cmdline_options..)r.shlexsplitris_posix)r)rcmdline_options_listrr1r_parse_cmdline_options[s  r7c CsJz||WSty$}ztd|dd|d|d|d}~ww)NzKey r#.z! is required for post write hook )rrr)rrrrrr_get_required_optionps r9 ignore_outputboolcommandcCs^|dd}|dd}t||}i}|rtj|d<|d<tjg||fd|i|dS)Ncwdrstdoutstderr)getr7 subprocessDEVNULLrun)rrr:r<r=r)r6kwrrr _run_hookzs   "rFconsole_scriptsFverify_versiontuple[int, ...] | Nonec Cst|d}tdD] }|j|kr|}n q td||r>d|jd|jd|d|d|jd |jd |jd }nd|jd |jd |jd }tj d |g}t ||||dS)N entrypointrGz*Could not find entrypoint console_scripts.zimport z; assert tuple(int(x) for x in z.__version__.split('.')) >= z, 'need exactly version z of z'; r8z()z; z-c) r9rimportlib_metadata_getrrrmoduleattrsys executablerF) rrr:rHentrypoint_nameentryimplpyscriptr<rrrrGs6    execcCs t|d}t||||gddS)NrO)r<)r9rF)rrr:rOrrrexec_s rUrLcCsHt|d}tj|durtd|tjd|g}t||||dS)NrLzCould not find module z-m)r9 importlibr find_specrrNrOrF)rrr: module_namer<rrrrLs  )rrrr)rrrrrr rr)rrr r!rr")r)rrrrr*)rr rrrr) rrrr r:r;r<r*rr")FN) rrrr r:r;rHrIrr")F)rrrr r:r;rr") __future__rimportlib.utilrVosr3rBrNtypingrrrr>rr util.pyfilesr configr r.r __annotations__rrr(r7r9rFrGrUrLrrrrs<