Skip to content

Commit

Permalink
Update the old updater if present when running on the windows standal…
Browse files Browse the repository at this point in the history
…one.
  • Loading branch information
comfyanonymous committed Feb 26, 2024
1 parent 36f7fac commit e61755e
Show file tree
Hide file tree
Showing 2 changed files with 42 additions and 0 deletions.
7 changes: 7 additions & 0 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -193,6 +193,13 @@ def load_extra_path_config(yaml_path):
folder_paths.set_temp_directory(temp_dir)
cleanup_temp()

if args.windows_standalone_build:
try:
import new_updater
new_updater.update_windows_updater()
except:
pass

loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
server = server.PromptServer(loop)
Expand Down
35 changes: 35 additions & 0 deletions new_updater.py
Original file line number Diff line number Diff line change
@@ -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.")

0 comments on commit e61755e

Please sign in to comment.