OjKddlmZddlmZddlZddlZddlmZddlm Z ddlm Z ddlm Z dd l m Z dd l mZdd l mZe rdd l mZiZejeZGd dZddZddZedS)) annotations)metadataN) ModuleType)Callable)Pattern) TYPE_CHECKING)util)DispatchPriority)PriorityDispatcher)PriorityDispatchResultcpeZdZdZddZddZ dd ejd ddZe d dZ e d!dZ dS)"PluginaDescribe a series of functions that are pulled in as a plugin. This is initially to provide for portable lists of autogenerate comparison functions, however the setup for a plugin can run any other kinds of global registration as well. .. versionadded:: 1.18.0 namestrc||_td||tvrt d|d|t|<t |_dS)Nzsetup plugin %szA plugin named z is already registered)rloginfo _all_plugins ValueErrorr autogenerate_comparators)selfrs k/lsinfo/ai/hellotax_ai/training_center/backend/venv/lib/python3.11/site-packages/alembic/runtime/plugins.py__init__zPlugin.__init__#sa  "D))) <  KtKKKLL L! T(:(<(<%%%returnNonec t|j=dS)zremove this pluginN)rr)rs rremovez Plugin.remove+s  # # #rNdefault) qualifierpriorityfn%Callable[..., PriorityDispatchResult]compare_targetcompare_element str | Noner!r"r cT|j|||||dS)aURegister an autogenerate comparison function. See the section :ref:`plugins_registering_autogenerate` for detailed examples on how to use this method. :param fn: The comparison function to register. The function receives arguments specific to the type of comparison being performed and should return a :class:`.PriorityDispatchResult` value. :param compare_target: The type of comparison being performed (e.g., ``"table"``, ``"column"``, ``"type"``). :param compare_element: Optional sub-element being compared within the target type. :param qualifier: Database dialect qualifier. Use ``"default"`` for all dialects, or specify a dialect name like ``"postgresql"`` to register a dialect-specific handler. Defaults to ``"default"``. :param priority: Execution priority for this comparison function. Functions are executed in priority order from :attr:`.DispatchPriority.FIRST` to :attr:`.DispatchPriority.LAST`. Defaults to :attr:`.DispatchPriority.MEDIUM`. )subgroupr"r!N)r dispatch_for)rr#r%r&r!r"s radd_autogenerate_comparatorz"Plugin.add_autogenerate_comparator0sLD %22 $ 3       r comparatorsr include_plugins list[str]c t}it}|D]T}|dr+|t|ddBt||<UtD] t fd|Dr fdD}|s0||t d j | j t |}|r*tjdd|dS) z_Populate all current autogenerate comparison functions into a given PriorityDispatcher.~Nc3LK|]}|jVdSNmatchr).0exclplugins r zAPlugin.populate_autogenerate_priority_dispatch..ls1??t4::fk**??????rcTg|]$}|j"|%Sr4)r6inclincluder8s r zBPlugin.populate_autogenerate_priority_dispatch..osBGDM,?,? ,L,Lrz!setting up autogenerate plugin %szDid not locate plugins: z, )set startswithadd_make_rervaluesanyupdaterrr populate_withr differencer CommandErrorjoin) clsr,r-excludematched_expressionsrinclude_matches never_matchedr=r8s @@r'populate_autogenerate_priority_dispatchz.Plugin.populate_autogenerate_priority_dispatchYs&)UU+-(+# / /Ds## / HT!""X..//// ( "))++ G GF????w????? !(O# <#**?;;; HH8&+ F F F  % %f&E F F F FG //0CDD  #E499]+C+CEE   rmodulercJ|t|dS)a<Call the ``setup()`` function of a plugin module, identified by passing the module object itself. E.g.:: from alembic.runtime.plugins import Plugin import myproject.alembic_plugin # Register the plugin manually Plugin.setup_plugin_from_module( myproject.alembic_plugin, "myproject.custom_operations" ) This will generate a new :class:`.Plugin` object with the given name, which will register itself in the global list of plugins. Then the module's ``setup()`` function is invoked, passing that :class:`.Plugin` object. This exact process is invoked automatically at import time for any plugin module that is published via the ``alembic.plugins`` entrypoint. N)setupr)rJrPrs rsetup_plugin_from_modulezPlugin.setup_plugin_from_modules"2  VD\\"""""r)rrrrr3) r#r$r%rr&r'r!rr"r rr)r,r r-r.rr)rPrrrrr) __name__ __module__ __qualname____doc__rrr MEDIUMr+ classmethodrOrSr;rrrrs====$$$$'+ ' #%5%<''''''R$$$[$L###[###rrrrr Pattern[str]c|d}d}|D]<}|dkr|dz }|r |d|zz }+td|tjd|ddd S) N.*z\..+?z\.zInvalid plugin expression ^r $)split isidentifierrrecompile)rtokensregtokens rrBrBs ZZ__F CDD C<< 8OCC    ! ! D 55= CCB$BBCC C :n#abb'nnn % %%rrctjdD]9}|D]"}t||j#:dS)Nzalembic.plugins)group)r entry_pointsloadrrSr) entrypointmods r_setuprosj+2CDDDBB ??$$ B BC  + +C A A A A BBBr)rrrr[rT) __future__r importlibrloggingrdtypesrtypingrrrr^r r r r r getLoggerrUrrrBror;rrrvsZ""""""  ######%%%%%%.------ g!!A#A#A#A#A#A#A#A#H & & & & BBBBr