tests for manager

2023-11-19 08:42:14 +01:00
parent a61712b5ec
commit c72970a917
2 changed files with 19 additions and 47 deletions
@@ -1,57 +1,30 @@
 # code based on pysssss repo
-import importlib.util
 import glob
 import os
-import sys
 from .nnodes import init, get_ext_dir,check_and_install,downloader
 import folder_paths

-import traceback


-NODE_CLASS_MAPPINGS = {}
-NODE_DISPLAY_NAME_MAPPINGS = {}
+py = get_ext_dir("py")
+files = glob.glob("*.py", root_dir=py, recursive=False)
+check_and_install('py-cpuinfo',"cpuinfo")
+check_and_install('gitpython','git')
+check_and_install('moviepy')
+check_and_install("opencv-python","cv2") 
+check_and_install('zipfile')
+check_and_install('scikit-build',"skbuild")
+#LLAMA DEPENTENCIES
+check_and_install('typing')
+check_and_install('diskcache')
+check_and_install('llama_cpp')

-
-  
-
-if init():
-    py = get_ext_dir("py")
-    files = glob.glob("*.py", root_dir=py, recursive=False)
-    check_and_install('py-cpuinfo',"cpuinfo")
-    check_and_install('gitpython','git')
-    check_and_install('moviepy')
-    check_and_install("opencv-python","cv2") 
-    check_and_install('zipfile')
-    check_and_install('scikit-build',"skbuild")
-    #LLAMA DEPENTENCIES
-    check_and_install('typing')
-    check_and_install('diskcache')
-    check_and_install('llama_cpp')
-    
-
-    #git clone https://github.com/hzwer/Practical-RIFE.git
-    from git import Repo
-    if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle")):
-        Repo.clone_from("https://github.com/hzwer/Practical-RIFE.git", os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle"))
-    #if train_log folder not exists
-    if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle","train_log")):
-        downloader("https://github.com/Nuked88/DreamingAI/raw/main/RIFE_trained_model_v4.7.zip")
+#git clone https://github.com/hzwer/Practical-RIFE.git
+from git import Repo
+if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle")):
+    Repo.clone_from("https://github.com/hzwer/Practical-RIFE.git", os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle"))
+#if train_log folder not exists
+if not os.path.exists(os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0],"ComfyUI-N-Nodes","libs","rifle","train_log")):
+    downloader("https://github.com/Nuked88/DreamingAI/raw/main/RIFE_trained_model_v4.7.zip")
                   

-    for file in files:
-        try:
-            name = os.path.splitext(file)[0]
-            spec = importlib.util.spec_from_file_location(name, os.path.join(py, file))
-            module = importlib.util.module_from_spec(spec)
-            sys.modules[name] = module
-            spec.loader.exec_module(module)
-            if hasattr(module, "NODE_CLASS_MAPPINGS") and getattr(module, "NODE_CLASS_MAPPINGS") is not None:
-                NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
-                if hasattr(module, "NODE_DISPLAY_NAME_MAPPINGS") and getattr(module, "NODE_DISPLAY_NAME_MAPPINGS") is not None:
-                    NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)
-        except Exception as e:
-            traceback.print_exc()
-
-
-__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
@@ -2,7 +2,6 @@ py-cpuinfo
 gitpython
 moviepy
 opencv-python
-zipfile
 scikit-build
 typing
 diskcache