mirror of
https://github.com/TheFunny/ArisuAutoSweeper
synced 2025-12-16 11:15:13 +00:00
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
import threading
|
|
from multiprocessing import Event, Process
|
|
|
|
from module.logger import logger
|
|
from module.webui.setting import State
|
|
|
|
|
|
def func(ev: threading.Event):
|
|
import argparse
|
|
import asyncio
|
|
import sys
|
|
|
|
import uvicorn
|
|
|
|
if sys.platform.startswith("win"):
|
|
asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
|
|
|
|
State.restart_event = ev
|
|
|
|
parser = argparse.ArgumentParser(description="Alas FastAPI web service")
|
|
parser.add_argument(
|
|
"--host",
|
|
type=str,
|
|
help="Host to listen. Default to WebuiHost in deploy setting",
|
|
)
|
|
parser.add_argument(
|
|
"-p",
|
|
"--port",
|
|
type=int,
|
|
help="Port to listen. Default to WebuiPort in deploy setting",
|
|
)
|
|
parser.add_argument(
|
|
"-k", "--key", type=str, help="Password of alas. No password by default"
|
|
)
|
|
parser.add_argument(
|
|
"--electron", action="store_true", help="Runs by electron client."
|
|
)
|
|
parser.add_argument(
|
|
"--run",
|
|
nargs="+",
|
|
type=str,
|
|
help="Run alas by config names on startup",
|
|
)
|
|
args, _ = parser.parse_known_args()
|
|
|
|
host = args.host or State.deploy_config.WebuiHost or "0.0.0.0"
|
|
port = args.port or int(State.deploy_config.WebuiPort) or 23467
|
|
State.electron = args.electron
|
|
|
|
logger.hr("Launcher config")
|
|
logger.attr("Host", host)
|
|
logger.attr("Port", port)
|
|
logger.attr("Backend", "FastAPI")
|
|
logger.attr("Electron", args.electron)
|
|
logger.attr("Reload", ev is not None)
|
|
|
|
if State.electron:
|
|
logger.info("Electron detected, remove log output to stdout")
|
|
from module.logger.logger import console_hdlr
|
|
logger.removeHandler(console_hdlr)
|
|
|
|
# Use the new FastAPI backend
|
|
uvicorn.run(
|
|
"module.webui.fastapi_backend.main:app",
|
|
host=host,
|
|
port=port,
|
|
reload=False
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if State.deploy_config.EnableReload:
|
|
should_exit = False
|
|
while not should_exit:
|
|
event = Event()
|
|
process = Process(target=func, args=(event,))
|
|
process.start()
|
|
while not should_exit:
|
|
try:
|
|
b = event.wait(1)
|
|
except KeyboardInterrupt:
|
|
should_exit = True
|
|
break
|
|
if b:
|
|
process.kill()
|
|
break
|
|
elif process.is_alive():
|
|
continue
|
|
else:
|
|
should_exit = True
|
|
else:
|
|
func(None)
|