Oj^UddlmZddlZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z ddl m Z ddl mZdd lmZe rdd lmZd ZiZd ed <d/dZd0dZd1dZd2d Zd3d!Zd4d%Zed& d5d6d*Zed+d7d8d,Zed-d7d8d.ZdS)9) annotationsN)Any)Callable) TYPE_CHECKING)util)compat)_preserving_path_as_str)PostWriteHookConfigREVISION_SCRIPT_FILENAMEdict _registrynamestrreturnrcfd}|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` c|t<|S)N)r)fnrs n/lsinfo/ai/hellotax_ai/training_center/backend/venv/lib/python3.11/site-packages/alembic/script/write_hooks.pydecoratezregister..decorate's $ )rrs` rregisterrs# Or revision_pathstr | os.PathLike[str]optionsr rct|} t|}|||S#t$r}tjd|d|d}~wwxYw)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)rrrhookkes r_invoker".sz,M::M, tM7+++  9t 9 9 9   s * AA  Apathhookslist[PostWriteHookConfig]Nonec|D]}|d} |d}tjd|d5t|||dddn #1swxYwYV#t$r }tjd|d||d}~wwxYwdS) z&Invoke hooks for a generated revision. _hook_nametypezRunning post write hook T)newlineNzKey 'z;.type' (or 'type' in toml) is required for post write hook )rstatusr"rr)r#r$r rtype_r!s r _run_hooksr-Fs  + +L! +LE3433T + +tT*** + + + + + + + + + + + + + + +    #000'+00   + +s(AA  A A  B#A>>Bcmdline_options_str list[str]ct|vr tdz|z}tj|tj}fd|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.  )posixcFg|]}|tSr)replaceREVISION_SCRIPT_TOKEN).0optionr#s r z*_parse_cmdline_options..is8  ,d33r)r5shlexsplitr is_posix)r.r#cmdline_options_lists ` r_parse_cmdline_optionsr=[sq$7773c9A ignore_outputboolcommandc|dd}|dd}t||}i}|rtjx|d<|d<tjg||fd|i|dS)Ncwdrstdoutstderr)getr= subprocessDEVNULLrun)r#rrArCrEr.r<kws r _run_hookrNzskk%..C!++i4412EtLLB9&0&888 r(|N4W434DD#DDDDDDrconsole_scriptsFverify_versiontuple[int, ...] | Nonect|d}tjdD]}|j|kr|}nt jd||r2d|jd|jd|d|d|jd |jd |jd }nd|jd |jd |jd }tj d |g}t||||dS)N entrypointrOz*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'; r?z()z; z-c) r@r importlib_metadata_getrrrmoduleattrsys executablerN) r#rrArPentrypoint_nameentryimplpyscriptrCs rrOrOsA+7LAAO./@AA   : ( (D E ) J J J   H +dk + +,0K + +Uc + +%3 + +9= + +{ + +"Y + + + HT[GGDKGG$)GGG~tX.G dG]G44444rexeccNt|d}t||||gdS)NrX)rC)r@rN)r#rrArXs rexec_r_s0%g|<rks#""""""   222222-,,,,,,2 &,,,,0++++*    * E E E E   -1 55555< &BBBBB  (5555555r