Przeglądaj źródła

move _stop route to api

w-e-w 2 lat temu
rodzic
commit
b9664ab615
2 zmienionych plików z 9 dodań i 9 usunięć
  1. 9 2
      modules/api/api.py
  2. 0 7
      webui.py

+ 9 - 2
modules/api/api.py

@@ -208,8 +208,11 @@ class Api:
         self.add_api_route("/sdapi/v1/reload-checkpoint", self.reloadapi, methods=["POST"])
         self.add_api_route("/sdapi/v1/scripts", self.get_scripts_list, methods=["GET"], response_model=models.ScriptsList)
         self.add_api_route("/sdapi/v1/script-info", self.get_script_info, methods=["GET"], response_model=List[models.ScriptInfo])
-        self.add_api_route("/sdapi/v1/quit-webui", self.quit_webui, methods=["POST"])
-        self.add_api_route("/sdapi/v1/restart-webui", self.restart_webui, methods=["POST"])
+
+        if shared.cmd_opts.add_stop_route:
+            self.add_api_route("/sdapi/v1/quit-webui", self.quit_webui, methods=["POST"])
+            self.add_api_route("/sdapi/v1/restart-webui", self.restart_webui, methods=["POST"])
+            self.add_api_route("/_stop", self.stop_route, methods=["POST"])
 
         self.default_script_arg_txt2img = []
         self.default_script_arg_img2img = []
@@ -724,3 +727,7 @@ class Api:
     def restart_webui(self):
         if restart.is_restartable():
             restart.restart_program()
+
+    def stop_route(request):
+        shared.state.server_command = "stop"
+        return Response("Stopping.")

+ 0 - 7
webui.py

@@ -362,11 +362,6 @@ def api_only():
     api.launch(server_name="0.0.0.0" if cmd_opts.listen else "127.0.0.1", port=cmd_opts.port if cmd_opts.port else 7861)
 
 
-def stop_route(request):
-    shared.state.server_command = "stop"
-    return Response("Stopping.")
-
-
 def webui():
     launch_api = cmd_opts.api
     initialize()
@@ -404,8 +399,6 @@ def webui():
                 "redoc_url": "/redoc",
             },
         )
-        if cmd_opts.add_stop_route:
-            app.add_route("/_stop", stop_route, methods=["POST"])
 
         # after initial launch, disable --autolaunch for subsequent restarts
         cmd_opts.autolaunch = False