sync (#658)

build_chb.py

@@ -0,0 +1,64 @@
+import os
+import shutil
+import stat
+import fnmatch
+
+from modules.launch_util import run
+
+
+def onerror(func, path, execinfo):
+    os.chmod(path, stat.S_IWUSR)
+    func(path)
+
+
+def get_empty_folder(path):
+    if os.path.isdir(path) or os.path.exists(path):
+        shutil.rmtree(path, onerror=onerror)
+    os.makedirs(path, exist_ok=True)
+    return path
+
+
+def git_clone(url, dir, hash=None):
+    run(f'git clone {url} {dir}')
+
+
+def findReplace(directory, find, replace, filePattern):
+    for path, dirs, files in os.walk(os.path.abspath(directory)):
+        for filename in fnmatch.filter(files, filePattern):
+            filepath = os.path.join(path, filename)
+            with open(filepath, encoding='utf-8') as f:
+                s = f.read()
+            s = s.replace(find, replace)
+            with open(filepath, "w", encoding='utf-8') as f:
+                f.write(s)
+
+
+repo = "https://github.com/comfyanonymous/ComfyUI"
+commit_hash = None
+
+temp_path = get_empty_folder(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'backend', 'temp'))
+core_path = get_empty_folder(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'backend', 'headless'))
+
+git_clone(repo, temp_path, commit_hash)
+
+
+def get_item(name, rename=None):
+    if rename is None:
+        rename = name
+    shutil.move(os.path.join(temp_path, name), os.path.join(core_path, rename))
+
+
+get_item('comfy', 'fcbh')
+get_item('comfy_extras', 'fcbh_extras')
+get_item('latent_preview.py')
+get_item('folder_paths.py')
+get_item('nodes.py')
+get_item('LICENSE')
+
+shutil.rmtree(temp_path, onerror=onerror)
+
+findReplace("./backend", "comfy", "fcbh", "*.py")
+findReplace("./backend", "Comfy", "FCBH", "*.py")
+findReplace("./backend", "FCBHUI", "fcbh_backend", "*.py")
+
+print('Backend is built.')