o g@sUddlmZddlZddlZddlZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z d d l mZd d lmZd ZiZded<d.ddZd/ddZd0d d!Zd1d$d%Zed& 'd2d3d*d&Zed+d2d3d,d-ZdS)4) annotationsN)Any)Callable)Dict)List)Mapping)Optional)Union)util)compatREVISION_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)fnrV/var/www/html/flaskapp/venv/lib/python3.10/site-packages/alembic/script/write_hooks.pydecorate&szregister..decorater)rrrrrregisters rrevisionoptionsMapping[str, Union[str, int]]rc CsDzt|}Wnty}z td|d|d}~ww|||S)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` zNo formatter with name 'z ' registeredN)rKeyErrorr CommandError)rrrhookkerrr_invoke-s   r path hook_configMapping[str, str]Nonec sddlm}|dd}|D]RsqfddD}|d<z|d}WntyB}z td d |d }~wwtjd d dt|||Wd n1s^wYqd S)z&Invoke hooks for a generated revision.)_split_on_space_commahookscs4i|]}|dr|tdd|qS).r%N) startswithlen).0keyr"rrr Ls  z_run_hooks.. _hook_nametypeKey z&.type is required for post write hook NzRunning post write hook T)newline) baser&splitgetrr rstatusr )r!r"r&namesoptstype_rrr.r _run_hooksBs4    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)r,optionr!rr ms z*_parse_cmdline_options..)rAshlexr5r is_posix)r<r!cmdline_options_listrrCr_parse_cmdline_options_s  rHconsole_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 entrypointr2r0z,.entrypoint is required for post write hook rIz*Could not find entrypoint console_scripts.cwdrr(stdoutstderrz-czimport z; r)z())rr rr importlib_metadata_getrr6rH subprocessDEVNULLrunsys executablemoduleattr) r!rrJentrypoint_namerentryimplrMr<rGkwrrrrItsL       execc Csz|d}Wnty"}ztd|dd|d|d}~ww|dd}|dd}t||}i}|rAtj|d<|d <tj|g|fd|i|dS) NrUr2r0z,.executable is required for post write hook rMrr(rNrO)rr rr6rHrQrRrS) r!rrJrUrrMr<rGr[rrrexec_s8      r])rrrr)rrrrrrrr)r!rr"r#rr$)r<rr!rrr=)F)r!rrrrJrKrr$) __future__rrErQrTtypingrrrrrrr r(r r rAr__annotations__rr r;rHrIr]rrrrs0              &