Experimental support for DATASETTE_LOAD_PLUGINS, refs #2164

datasette/plugins.py

@@ -1,4 +1,5 @@
 import importlib
+import os
 import pluggy
 import pkg_resources
 import sys
@@ -22,10 +23,26 @@
 pm = pluggy.PluginManager("datasette")
 pm.add_hookspecs(hookspecs)
 
-if not hasattr(sys, "_called_from_test"):
+DATASETTE_LOAD_PLUGINS = os.environ.get("DATASETTE_LOAD_PLUGINS") or None
+
+if not hasattr(sys, "_called_from_test") and not DATASETTE_LOAD_PLUGINS:
     # Only load plugins if not running tests
     pm.load_setuptools_entrypoints("datasette")
 
+# Load any plugins specified in DATASETTE_LOAD_PLUGINS
+if DATASETTE_LOAD_PLUGINS:
+    for package_name in DATASETTE_LOAD_PLUGINS.split(","):
+        try:
+            distribution = pkg_resources.get_distribution(package_name)
+            entry_map = distribution.get_entry_map()
+            if "datasette" in entry_map:
+                for plugin_name, entry_point in entry_map["datasette"].items():
+                    mod = entry_point.load()
+                    pm.register(mod, plugin_name)
+        except pkg_resources.DistributionNotFound:
+            sys.stderr.write("Plugin {} could not be found\n".format(package_name))
+
+
 # Load default plugins
 for plugin in DEFAULT_PLUGINS:
     mod = importlib.import_module(plugin)