Update the old updater if present when running on the windows standalone.
This commit is contained in:
parent
36f7face37
commit
e61755ead0
7
main.py
7
main.py
|
@ -193,6 +193,13 @@ if __name__ == "__main__":
|
||||||
folder_paths.set_temp_directory(temp_dir)
|
folder_paths.set_temp_directory(temp_dir)
|
||||||
cleanup_temp()
|
cleanup_temp()
|
||||||
|
|
||||||
|
if args.windows_standalone_build:
|
||||||
|
try:
|
||||||
|
import new_updater
|
||||||
|
new_updater.update_windows_updater()
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
loop = asyncio.new_event_loop()
|
loop = asyncio.new_event_loop()
|
||||||
asyncio.set_event_loop(loop)
|
asyncio.set_event_loop(loop)
|
||||||
server = server.PromptServer(loop)
|
server = server.PromptServer(loop)
|
||||||
|
|
|
@ -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