jUddlmZddlmZddlZddlZddlmZddlmZddl m Z ddl m Z dd l m Z dd l mZdd l mZe rdd l mZiZd ed<ejeZGddZddiZddZddZddZedS)) annotations)metadataN)Pattern) ModuleType)Callable) TYPE_CHECKING)util)DispatchPriority)PriorityDispatcher)PriorityDispatchResultzdict[str, Plugin] _all_pluginscpeZdZdZddZddZ 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)rloginfor ValueErrorr autogenerate_comparators)selfrs z/home/lighthouse/badge-speaker-benchmark/.benchmark-venv-community/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]n}|dr8|tt |ddOtt ||<ot D] 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..ns1??t4::fk**??????rcTg|]$}|j"|%Sr4)r6inclincluder8s r zBPlugin.populate_autogenerate_priority_dispatch..qsBGDM,?,? ,L,Lrz!setting up autogenerate plugin %szDid not locate plugins: z, )set startswithadd_make_re_translate_legacy_namervaluesanyupdaterrr populate_withr differencer CommandErrorjoin) clsr,r-excludematched_expressionsrinclude_matches never_matchedr=r8s @@r'populate_autogenerate_priority_dispatchz.Plugin.populate_autogenerate_priority_dispatchYs&)UU+-(+# G GDs## G H%;DH%E%EFFGGGG ()?)E)E F F "))++ 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)rKrQrs 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)rQrrrrr) __name__ __module__ __qualname____doc__rrr MEDIUMr+ classmethodrPrTr;rrrrs====$$$$'+ ' #%5%<''''''R&&&[&P###[###rrz+alembic.autogenerate.checkconstraint_bynamez"alembic.ext.checkconstraint_bynamerrrc8t||S)a@Resolve a plugin name that has been renamed to its current name. This allows an ``env.py`` written against an earlier Alembic release to keep working, whether the name is used to include or to exclude the plugin. Wildcard expressions are never translated, as only fully stated names are eligible. )_legacy_plugin_namesget)rs rrCrCs # #D$ / //r 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_pointsloadrrTr) entrypointmods r_setuprssj+2CDDDBB ??$$ B BC  + +C A A A A BBBr)rrrr)rrrr_rU) __future__r importlibrloggingrhrtypesrtypingrrrbr r r r r__annotations__ getLoggerrVrrr]rCrBrsr;rrr{s"""""""  ######%%%%%%.------"$ $$$$g!!C#C#C#C#C#C#C#C#T2,  0 0 0 0 & & & & BBBBr