mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-24 01:21:25 +00:00
92 lines
2.6 KiB
Python
92 lines
2.6 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 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(
|
|
"--cdn",
|
|
action="store_true",
|
|
help="Use jsdelivr cdn for pywebio static files (css, js). Self host cdn 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 22367
|
|
State.electron = args.electron
|
|
|
|
logger.hr("Launcher config")
|
|
logger.attr("Host", host)
|
|
logger.attr("Port", port)
|
|
logger.attr("Electron", args.electron)
|
|
logger.attr("Reload", ev is not None)
|
|
|
|
if State.electron:
|
|
# https://github.com/LmeSzinc/AzurLaneAutoScript/issues/2051
|
|
logger.info("Electron detected, remove log output to stdout")
|
|
from module.logger.logger import console_hdlr
|
|
logger.removeHandler(console_hdlr)
|
|
|
|
uvicorn.run("module.webui.app:app", host=host, port=port, factory=True)
|
|
|
|
|
|
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)
|