Files
whmc76-ComfyUI-Openpose-Edi…/__init__.py
T
2023-11-10 14:34:05 +08:00

41 lines
1.4 KiB
Python

import os
import filecmp
import shutil
import __main__
from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
def update_javascript():
extensions_folder = os.path.join(os.path.dirname(os.path.realpath(__main__.__file__)),
"web" + os.sep + "extensions" + os.sep + "ComfyUI-OpenPose-Editor-Plus")
javascript_folder = os.path.join(os.path.dirname(os.path.realpath(__file__)), "js")
if not os.path.exists(extensions_folder):
print("Creating frontend extension folder: " + extensions_folder)
os.mkdir(extensions_folder)
result = filecmp.dircmp(javascript_folder, extensions_folder)
if result.left_only or result.diff_files:
print('Update to javascripts files detected')
file_list = list(result.left_only)
file_list.extend(x for x in result.diff_files if x not in file_list)
for file in file_list:
print(f'Copying {file} to extensions folder')
src_file = os.path.join(javascript_folder, file)
dst_file = os.path.join(extensions_folder, file)
if os.path.exists(dst_file):
os.remove(dst_file)
shutil.copy(src_file, dst_file)
update_javascript()
print('\033[34mOpenpose Editor Plus: \033[92mLoaded\033[0m')