fix: Plugins are still loaded when running tests (#719)

* 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>

sqlite_utils/cli.py

@@ -16,7 +16,7 @@ from sqlite_utils.db import (
     NoTable,
     quote_identifier,
 )
-from sqlite_utils.plugins import pm, get_plugins
+from sqlite_utils.plugins import ensure_plugins_loaded, pm, get_plugins
 from sqlite_utils.utils import maximize_csv_field_size_limit
 from sqlite_utils import recipes
 import textwrap
@@ -3412,6 +3412,7 @@ def plugins_list():
     click.echo(json.dumps(get_plugins(), indent=2))
 
 
+ensure_plugins_loaded()
 pm.hook.register_commands(cli=cli)
 cli.add_command(migrate)
 

sqlite_utils/db.py

@@ -41,7 +41,7 @@ from typing import (
     Tuple,
 )
 import uuid
-from sqlite_utils.plugins import pm
+from sqlite_utils.plugins import ensure_plugins_loaded, pm
 
 try:
     iterdump = importlib.import_module("sqlite_dump").iterdump
@@ -401,6 +401,7 @@ class Database:
         self._registered_functions: set = set()
         self.use_counts_table = use_counts_table
         if execute_plugins:
+            ensure_plugins_loaded()
             pm.hook.prepare_connection(conn=self.conn)
         self.strict = strict
 

sqlite_utils/plugins.py

@@ -6,13 +6,19 @@ from . import hookspecs
 
 pm: pluggy.PluginManager = pluggy.PluginManager("sqlite_utils")
 pm.add_hookspecs(hookspecs)
+_plugins_loaded = False
 
-if not getattr(sys, "_called_from_test", False):
-    # Only load plugins if not running tests
+
+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():