You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.2 KiB
35 lines
1.2 KiB
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.")
|
|
|