mirror of
https://github.com/simonw/sqlite-utils.git
synced 2026-07-23 17:34:32 +02:00
f448b61f5e* fix: defer plugin entrypoint loading until runtime Fixes simonw/sqlite-utils#713 * Lazy-load plugins when listing them Keep plugin entrypoints from loading at module import time, but preserve the existing behavior of get_plugins() by loading entrypoints the first time plugins are listed outside tests. --------- Co-authored-by: Simon Willison <swillison@gmail.com>
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from typing import Dict, List, Union
|
|
|
|
import pluggy
|
|
import sys
|
|
from . import hookspecs
|
|
|
|
pm: pluggy.PluginManager = pluggy.PluginManager("sqlite_utils")
|
|
pm.add_hookspecs(hookspecs)
|
|
_plugins_loaded = False
|
|
|
|
|
|
def ensure_plugins_loaded() -> None:
|
|
global _plugins_loaded
|
|
if _plugins_loaded or getattr(sys, "_called_from_test", False):
|
|
return
|
|
pm.load_setuptools_entrypoints("sqlite_utils")
|
|
_plugins_loaded = True
|
|
|
|
|
|
def get_plugins() -> List[Dict[str, Union[str, List[str]]]]:
|
|
ensure_plugins_loaded()
|
|
plugins: List[Dict[str, Union[str, List[str]]]] = []
|
|
plugin_to_distinfo = dict(pm.list_plugin_distinfo())
|
|
for plugin in pm.get_plugins():
|
|
hookcallers = pm.get_hookcallers(plugin) or []
|
|
plugin_info: Dict[str, Union[str, List[str]]] = {
|
|
"name": plugin.__name__,
|
|
"hooks": [h.name for h in hookcallers],
|
|
}
|
|
distinfo = plugin_to_distinfo.get(plugin)
|
|
if distinfo:
|
|
plugin_info["version"] = distinfo.version
|
|
plugin_info["name"] = distinfo.project_name
|
|
plugins.append(plugin_info)
|
|
return plugins
|