Create __init__.py

2024-04-02 17:44:28 -07:00
parent 217f98c862
commit 1998f79be8
1 changed files with 28 additions and 0 deletions
@@ -0,0 +1,28 @@
+# Comfy mappings
+from .node import CustomScheduler
+
+NODE_CLASS_MAPPINGS = {"CustomScheduler": CustomScheduler}
+NODE_DISPLAY_NAME_MAPPINGS = {"CustomScheduler" : "CustomScheduler"}
+__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
+
+# Copy extension to comfy
+import folder_paths
+import shutil
+import os
+
+cwd_path = os.path.dirname(os.path.realpath(__file__))
+comfy_path = folder_paths.base_path
+web_extension_path = os.path.join(comfy_path, "web", "extensions", "CustomScheduler")
+widgets_js_file = os.path.join(cwd_path, "extension.js")
+
+if not os.path.exists(web_extension_path):
+    os.makedirs(web_extension_path)
+else:
+    try:
+        shutil.rmtree(web_extension_path)
+        os.makedirs(web_extension_path)
+    except:
+        print("Unable to remove old web extension.")
+        pass
+        
+shutil.copy(widgets_js_file, web_extension_path)