36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
|
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.")
|