| import os
|
| import shutil
|
|
|
| base_path = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
|
| def update_windows_updater():
|
| top_path = os.path.dirname(base_path)
|
| updater_path = os.path.join(base_path, ".ci/update_windows/update.py")
|
| bat_path = os.path.join(base_path, ".ci/update_windows/update_comfyui.bat")
|
|
|
| dest_updater_path = os.path.join(top_path, "update/update.py")
|
| dest_bat_path = os.path.join(top_path, "update/update_comfyui.bat")
|
| dest_bat_deps_path = os.path.join(top_path, "update/update_comfyui_and_python_dependencies.bat")
|
|
|
| try:
|
| with open(dest_bat_path, 'rb') as f:
|
| contents = f.read()
|
| except:
|
| return
|
|
|
| if not contents.startswith(b"..\\python_embeded\\python.exe .\\update.py"):
|
| return
|
|
|
| shutil.copy(updater_path, dest_updater_path)
|
| try:
|
| with open(dest_bat_deps_path, 'rb') as f:
|
| contents = f.read()
|
| contents = contents.replace(b'..\\python_embeded\\python.exe .\\update.py ..\\ComfyUI\\', b'call update_comfyui.bat nopause')
|
| with open(dest_bat_deps_path, 'wb') as f:
|
| f.write(contents)
|
| except:
|
| pass
|
| shutil.copy(bat_path, dest_bat_path)
|
| print("Updated the windows standalone package updater.")
|
|
|
