import argparse
import signal
import sys
from api import *
import uvicorn

# argparse
parser = argparse.ArgumentParser(description="Process some integers.")
parser.add_argument("--port", type=int, default=7074, help="The port number.")
args = parser.parse_args()


# 通过信号来退出服务，否则会出现终端显示退出后，实际进程仍在运行
# 定义信号处理函数
def signal_handler(sig, frame):
    print("[python] [flask] Received signal to terminate the server:", sig)
    sys.exit(0)


# 注册信号处理函数
signal.signal(signal.SIGTERM, signal_handler)
signal.signal(signal.SIGINT, signal_handler)

if __name__ == "__main__":
    # 以api方式启动服务会出现警告，请忽略
    # app.run(port=args.port,use_reloader=True)
    uvicorn.run(app="main:app", host="127.0.0.1", port=int(args.port), reload=True)
    # 或许flask内置的stdio与node.js stdio有冲突，导致控制台无法显示信息。
    # 如果想要查看控制台输出，请单独启动服务 npm run dev-python
    print("python server is running at port:", args.port)