Restart: only do restart if running via the wrapper script

This commit is contained in:
Aarni Koskela
2023-06-05 20:04:28 +03:00
parent 9781f31f74
commit 46a5bd64ed
5 changed files with 33 additions and 13 deletions

23
modules/restart.py Normal file
View File

@@ -0,0 +1,23 @@
import os
from pathlib import Path
from modules.paths_internal import script_path
def is_restartable() -> bool:
"""
Return True if the webui is restartable (i.e. there is something watching to restart it with)
"""
return bool(os.environ.get('SD_WEBUI_RESTART'))
def restart_program() -> None:
"""creates file tmp/restart and immediately stops the process, which webui.bat/webui.sh interpret as a command to start webui again"""
(Path(script_path) / "tmp" / "restart").touch()
stop_program()
def stop_program() -> None:
os._exit(0)