51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
# Title: ComfyUI Install Customs Nodes and javascript files
|
|
# Author: Hina Chen
|
|
# Version: 2023.11.30
|
|
|
|
import os
|
|
import filecmp
|
|
import shutil
|
|
|
|
import __main__
|
|
|
|
from distutils.dir_util import copy_tree
|
|
from .poseEditor3D_node import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
|
|
|
# ComfyUI folders web
|
|
folder_web = os.path.join(os.path.dirname(os.path.realpath(__main__.__file__)), "web")
|
|
extensions_folder = os.path.join(folder_web, 'extensions', 'ComfyUI_3dPoseEditor')
|
|
web_folder = os.path.join(extensions_folder, 'web')
|
|
assets_folder = os.path.join(extensions_folder, 'assets')
|
|
webSource_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "web")
|
|
assetsSource_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "assets")
|
|
|
|
def update_web():
|
|
shutil.rmtree(extensions_folder)
|
|
|
|
if not os.path.exists(extensions_folder):
|
|
print("[3D OpenPose Editor] Creating frontend extension folder: " + extensions_folder)
|
|
os.mkdir(extensions_folder)
|
|
|
|
print('[3D OpenPose Editor] Copy web folder')
|
|
if os.path.exists(web_folder):
|
|
shutil.rmtree(web_folder)
|
|
copy_tree(webSource_folder, web_folder)
|
|
|
|
print('[3D OpenPose Editor] Copy assets folder')
|
|
if os.path.exists(assets_folder):
|
|
shutil.rmtree(assets_folder)
|
|
copy_tree(assetsSource_folder, assets_folder)
|
|
|
|
print('[3D OpenPose Editor] Copy js file')
|
|
if os.path.exists(extensions_folder + os.sep + 'openposeeditor.js'):
|
|
os.remove(extensions_folder + os.sep + 'openposeeditor.js')
|
|
shutil.copy(
|
|
os.path.dirname(os.path.realpath(__file__)) + os.sep + 'openposeeditor.js',
|
|
extensions_folder + os.sep + 'openposeeditor.js'
|
|
)
|
|
|
|
update_web()
|
|
|
|
print('\033[34m3D OpenPose Editor: \033[92mLoaded\033[0m') |