feat: support better reboot mechanism based on comfy-cli

2024-04-23 20:46:54 +09:00
parent 4c8635d893
commit 12a364dd6a
2 changed files with 9 additions and 1 deletions
--- a/glob/manager_server.py
+++ b/glob/manager_server.py
@@ -1112,6 +1112,14 @@ def restart(self):
    except Exception as e:
        pass

+    if '__COMFY_CLI_SESSION__' in os.environ:
+        with open(os.path.join(os.environ['__COMFY_CLI_SESSION__'], '.reboot'), 'w') as file:
+            pass
+
+        print(f"\nRestarting...\n\n")
+        exit(0)
+
+    print(f"\nRestarting... [Legacy Mode]\n\n")
    return os.execv(sys.executable, [sys.executable] + sys.argv)