comfyanonymous
9 months ago
2 changed files with 42 additions and 0 deletions
@ -0,0 +1,35 @@
|
||||
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.") |
Loading…
Reference in new issue