funcnodes-react-flow 0.3.12__py3-none-any.whl → 0.3.14__py3-none-any.whl

funcnodes_react_flow/__init__.py

@@ -1,22 +1,10 @@
-from typing import Dict, TypedDict, List
+from typing import TypedDict, List
 
 try:
     from .run import run_server  # noqa: F401
 except ImportError:
     pass
-
-
-class ReactPlugin(TypedDict):
-    """
-    A typed dictionary for a React plugin.
-
-    Attributes:
-      js (list[str]): A list of JavaScript files.
-    """
-
-    js: list[str]
-    css: list[str]
-    module: str
+from .plugin_setup import ReactPlugin, setup, FUNCNODES_REACT_PLUGIN
 
 
 class ExpandedReactPlugin(TypedDict):
@@ -32,20 +20,6 @@ class ExpandedReactPlugin(TypedDict):
     css: List[bytes]
 
 
-FUNCNODES_REACT_PLUGIN: Dict[str, ReactPlugin] = {}
-
-
-def add_react_plugin(name: str, plugin: ReactPlugin):
-    """
-    Add a React plugin to the FUNCNODES_REACT_PLUGIN dictionary.
-
-    Args:
-      name (str): The name of the plugin.
-      plugin (ReactPlugin): The plugin to add.
-    """
-    FUNCNODES_REACT_PLUGIN[str(name)] = plugin
-
-
 def get_react_plugin_content(key: str) -> ExpandedReactPlugin:
     """
     Get the content of a React plugin.
@@ -58,6 +32,9 @@ def get_react_plugin_content(key: str) -> ExpandedReactPlugin:
     """
     key = str(key)
 
+    if key not in FUNCNODES_REACT_PLUGIN:
+        raise ValueError(f"React plugin {key} not found")
+
     with open(FUNCNODES_REACT_PLUGIN[key]["module"], "rb") as f:
         module = f.read()
     resp: ExpandedReactPlugin = {"js": [], "module": module, "css": []}
@@ -71,3 +48,15 @@ def get_react_plugin_content(key: str) -> ExpandedReactPlugin:
             with open(css, "rb") as f:
                 resp["css"].append(f.read())
     return resp
+
+
+setup()
+
+
+__all__ = [
+    "run_server",
+    "ReactPlugin",
+    "FUNCNODES_REACT_PLUGIN",
+    "get_react_plugin_content",
+    "ReactPlugin",
+]

funcnodes_react_flow/plugin_setup.py

@@ -0,0 +1,43 @@
+from typing import Dict
+from funcnodes_core import plugins
+
+
+class ReactPlugin(plugins.BasePlugin):
+    """
+    A typed dictionary for a React plugin.
+
+    Attributes:
+      js (list[str]): A list of JavaScript files.
+    """
+
+    js: list[str]
+    css: list[str]
+
+
+FUNCNODES_REACT_PLUGIN: Dict[str, ReactPlugin] = {}
+
+
+def add_react_plugin(name: str, plugin: ReactPlugin):
+    """
+    Add a React plugin to the FUNCNODES_REACT_PLUGIN dictionary.
+
+    Args:
+      name (str): The name of the plugin.
+      plugin (ReactPlugin): The plugin to add.
+    """
+    FUNCNODES_REACT_PLUGIN[str(name)] = plugin
+
+
+def plugin_function(installed_module: plugins.InstalledModule):
+    entry_points = installed_module.entry_points
+    mod = installed_module.module
+
+    if "react_plugin" in entry_points:
+        add_react_plugin(mod, entry_points["react_plugin"])
+    elif hasattr(mod, "REACT_PLUGIN"):
+        add_react_plugin(mod, mod.REACT_PLUGIN)
+        entry_points["react_plugin"] = mod.REACT_PLUGIN
+
+
+def setup():
+    plugins.register_setup_plugin(plugin_function)