Validate plugins in lookup functions

edutap-eu · Dec 12, 2024 · a82da93 · a82da93
1 parent 31737a9
commit a82da93
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/edutap/wallet_google/plugins.py b/src/edutap/wallet_google/plugins.py
@@ -6,9 +6,17 @@
 
 def get_image_providers() -> Iterable[ImageProvider]:
     eps = entry_points(group="edutap.wallet_google.plugins")
-    return [ep.load() for ep in eps if ep.name == "ImageProvider"]
+    plugins = [ep.load() for ep in eps if ep.name == "ImageProvider"]
+    for plugin in plugins:
+        if not isinstance(plugin, ImageProvider):
+            raise ValueError(f"{plugin} not implements ImageProvider")
+    return [plugin() for plugin in plugins]
 
 
 def get_callback_handlers() -> Iterable[CallbackHandler]:
     eps = entry_points(group="edutap.wallet_google.plugins")
-    return [ep.load() for ep in eps if ep.name == "CallbackHandler"]
+    plugins = [ep.load() for ep in eps if ep.name == "CallbackHandler"]
+    for plugin in plugins:
+        if not isinstance(plugin, CallbackHandler):
+            raise ValueError(f"{plugin} not implements CallbackHandler")
+    return [plugin() for plugin in plugins]