iKddlmZddlmZddlZddlZddlmZddlm Z ddlm Z ddlm Z dd l m Z dd l mZdd l mZe rdd l mZiZej$eZGd dZddZddZey)) annotations)metadataN) ModuleType)Callable)Pattern) TYPE_CHECKING)util)DispatchPriority)PriorityDispatcher)PriorityDispatchResultceZdZdZd dZd dZ d dejd d dZe ddZ e dd Z y)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 c||_tjd||tvrt d|d|t|<t |_y)Nzsetup plugin %szA plugin named z is already registered)nameloginfo _all_plugins ValueErrorr autogenerate_comparators)selfrs l/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/alembic/runtime/plugins.py__init__zPlugin.__init__#sJ  "D) < tf4JKL L! T(:(<%c&t|j=y)zremove this pluginN)rr)rs rremovez Plugin.remove+s  #rNdefault) qualifierprioritycN|jj|||||y)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)rfncompare_targetcompare_elementrrs radd_autogenerate_comparatorz"Plugin.add_autogenerate_comparator0s6D %%22 $ 3  rc t}i}t}|D]?}|jdr|jt|dd2t|||<Atj D] t fd|Dr|Dcgc]#}||j js"|%}}|sI|j|tjd j|j jt|j|} | r'tj ddj#| ycc}w)z_Populate all current autogenerate comparison functions into a given PriorityDispatcher.~Nc3TK|]}|jj!ywN)matchr).0exclplugins r zAPlugin.populate_autogenerate_priority_dispatch..ls?wt4::fkk*ws%(z!setting up autogenerate plugin %szDid not locate plugins: z, )set startswithadd_make_rervaluesanyr,rupdaterr populate_withr differencer CommandErrorjoin) cls comparatorsinclude_pluginsexcludeincludematched_expressionsrinclinclude_matches never_matchedr/s @r'populate_autogenerate_priority_dispatchz.Plugin.populate_autogenerate_priority_dispatchYs4&)U+-(+#Ds# HT!"X./ ( $ #))+F?w??")!(GDM,?,? ,L ##**?; HH8&++ F  % %f&E&E F,G //0CD ##*499]+C*DE  s #E 0E c8|jt|y)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)r<modulers rsetup_plugin_from_modulezPlugin.setup_plugin_from_modules2  VD\"r)rstrreturnNoner+) r#z%Callable[..., PriorityDispatchResult]r$rJr%z str | NonerrJrr rLrM)r=r r>z list[str]rLrM)rHrrrJrLrM) __name__ __module__ __qualname____doc__rrr MEDIUMr& classmethodrErIrrrrs=$'+ ' #%5%<%<' 1''$ '  '#' 'R$,$?H$ $$L##rrc|jd}d}|D]3}|dk(r|dz }|jr |d|zz }'td|tjd|ddd S) N.*z\..+?z\.zInvalid plugin expression ^r $)split isidentifierrrecompile)rtokensregtokens rr4r4sz ZZ_F C C< 8OC    ! 55= C9$BC C  ::#ab'!n %%rctjdD]7}|jD]"}tj ||j $9y)Nzalembic.plugins)group)r entry_pointsloadrrIr) entrypointmods r_setuprhs>++2CD ??$C  + +C A%Er)rrJrLz Pattern[str]rK) __future__r importlibrloggingr]typesrtypingrrrrWr r r r r getLoggerrNrrr4rhrTrrrosd"  #%- g!A#A#H & Br