o 8c@slddlZddlZddlZddlmZddlmZdZiZddZdd Z d d Z d d Z edddZ dS)N)util)compatREVISION_SCRIPT_FILENAMEcsfdd}|S)zA function decorator that will register that function as a write hook. See the documentation linked below for an example. .. versionadded:: 1.2.0 .. seealso:: :ref:`post_write_hooks_custom` cs |t<dS)N) _registry)fnnameP/var/www/html/gps/gps/lib/python3.10/site-packages/alembic/script/write_hooks.pydecorates zregister..decorater )r r r rr registers r c CsVzt|}Wnty%}ztjtd||dWYd}~dSd}~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` z&No formatter with name '%s' registeredfrom_N)rKeyErrorrraise_r CommandError)r revisionoptionshookker r r _invoke"s   rc sddlm}|dd}|D]GsqfddD}|d<z|d}Wn tyJ}ztjtd f|d WYd }~qd }~wwtj d t |||d dqd S)z&Invoke hooks for a generated revision.)_split_on_space_commahookscs4i|]}|dr|tdd|qS).rN) startswithlen).0key hook_configr r r @s  z_run_hooks.. _hook_nametypez.Key %s.type is required for post write hook %rrNzRunning post write hook "%s"T)newline) basersplitgetrrrrrstatusr)pathr"rnamesoptstype_rr r!r _run_hooks6s>    r/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)roptionr+r r is z*_parse_cmdline_options..)r3shlexr(ris_posix)cmdline_options_strr+cmdline_options_listr r5r _parse_cmdline_options[s  r;console_scriptsc Csddl}z|d}Wn$ty.}ztjtd|d|df|dWYd}~nd}~ww|d|}t|}|dd}|dd }t ||} t j t j d d |jd |jf|jfg| |d dS)Nr entrypointz4Key %s.entrypoint is required for post write hook %rr$rr<cwdrrz-czimport %s; %s()r)r>) pkg_resourcesrrrrriter_entry_pointsnextr)r; subprocessrunsys executable module_namejoinattrs) r+rr?entrypoint_nameriter_implr>r9r:r r r r<ps<      ) r7rBrDrrrr3rr rr/r;r<r r r r s  %