o voiK@sddlmZddlmZddlZddlZddlmZddlm Z ddlm Z ddlm Z dd l m Z dd l mZdd l mZe rFdd l mZiZeeZGd ddZdddZdddZedS)) annotations)metadataN) ModuleType)Callable)Pattern) TYPE_CHECKING)util)DispatchPriority)PriorityDispatcher)PriorityDispatchResultc@sXeZdZdZd!ddZd"dd Z d#d ejd d$ddZe d%ddZ e d&dd Z d S)'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 namestrcCs>||_td||tvrtd|d|t|<t|_dS)Nzsetup plugin %szA plugin named z is already registered)rloginfo _all_plugins ValueErrorr autogenerate_comparators)selfrrK/home/ubuntu/.local/lib/python3.10/site-packages/alembic/runtime/plugins.py__init__#s   zPlugin.__init__returnNonecCs t|j=dS)zremove this pluginN)rr)rrrrremove+s z Plugin.removeNdefault) qualifierpriorityfn%Callable[..., PriorityDispatchResult]compare_targetcompare_element str | Nonerrr cCs|jj||||d|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`. )subgrouprrN)r dispatch_for)rrr!r"rrrrradd_autogenerate_comparator0s"z"Plugin.add_autogenerate_comparator comparatorsr include_plugins list[str]cst}it}|D]}|dr|t|ddq t||<q tD]-tfdd|Dr6q(fddD}|sCq(||t dj | j q(t |}|rjtd d |dS) z_Populate all current autogenerate comparison functions into a given PriorityDispatcher.~Nc3s|] }|jVqdSNmatchr).0excl)pluginrr lszAPlugin.populate_autogenerate_priority_dispatch..cs g|] }|jr|qSrr-)r/inclincluder1rr os zBPlugin.populate_autogenerate_priority_dispatch..z!setting up autogenerate plugin %szDid not locate plugins: z, )set startswithadd_make_rervaluesanyupdaterrr populate_withr differencer CommandErrorjoin)clsr'r(excludematched_expressionsrinclude_matches never_matchedrr4r'populate_autogenerate_priority_dispatchYs0    z.Plugin.populate_autogenerate_priority_dispatchmodulercCs|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 )rBrHrrrrsetup_plugin_from_moduleszPlugin.setup_plugin_from_module)rrrrr,) rr r!rr"r#rrrr rr)r'r r(r)rr)rHrrrrr) __name__ __module__ __qualname____doc__rrr MEDIUMr& classmethodrGrJrrrrr s  ) &r rrr Pattern[str]cCsf|d}d}|D]}|dkr|d7}q |r|d|7}q td|td|ddd S) N.*z\..+?z\.zInvalid plugin expression ^r$)split isidentifierrrecompile)rtokensregtokenrrrr:s  r:rcCs2tjddD]}|D] }t||jq qdS)Nzalembic.plugins)group)r entry_pointsloadr rJr) entrypointmodrrr_setups  rd)rrrrRrK) __future__r importlibrloggingrZtypesrtypingrrrrTr r r r r getLoggerrLrr r:rdrrrrs(