funcnodes-react-flow 0.3.13__py3-none-any.whl → 0.3.15__py3-none-any.whl

funcnodes_react_flow/__init__.py

@@ -35,9 +35,11 @@ def get_react_plugin_content(key: str) -> ExpandedReactPlugin:
     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": []}
+    resp: ExpandedReactPlugin = {
+        "js": [],
+        "module": FUNCNODES_REACT_PLUGIN[key]["module"],
+        "css": [],
+    }
     if "js" in FUNCNODES_REACT_PLUGIN[key]:
         for js in FUNCNODES_REACT_PLUGIN[key]["js"]:
             with open(js, "rb") as f:

funcnodes_react_flow/plugin_setup.py

@@ -1,3 +1,4 @@
+from types import ModuleType
 from typing import Dict
 from funcnodes_core import plugins
 
@@ -17,7 +18,7 @@ class ReactPlugin(plugins.BasePlugin):
 FUNCNODES_REACT_PLUGIN: Dict[str, ReactPlugin] = {}
 
 
-def add_react_plugin(name: str, plugin: ReactPlugin):
+def add_react_plugin(module: ModuleType, plugin: ReactPlugin):
     """
     Add a React plugin to the FUNCNODES_REACT_PLUGIN dictionary.
 
@@ -25,7 +26,8 @@ def add_react_plugin(name: str, plugin: ReactPlugin):
       name (str): The name of the plugin.
       plugin (ReactPlugin): The plugin to add.
     """
-    FUNCNODES_REACT_PLUGIN[str(name)] = plugin
+    plugin["module"] = module.__file__
+    FUNCNODES_REACT_PLUGIN[str(module)] = plugin
 
 
 def plugin_function(installed_module: plugins.InstalledModule):