142 lines
3.9 KiB
Python
142 lines
3.9 KiB
Python
"""WPS Bot Game - FastAPI主应用"""
|
||
import logging
|
||
import argparse
|
||
import os
|
||
from fastapi import FastAPI
|
||
from fastapi.responses import JSONResponse
|
||
from contextlib import asynccontextmanager
|
||
import asyncio
|
||
|
||
from config import APP_CONFIG, SESSION_TIMEOUT, SetWebhookURL, GetWebhookURL
|
||
from core.middleware import ConcurrencyLimitMiddleware
|
||
from core.database import get_db
|
||
from routers import callback, health, private
|
||
|
||
# 配置日志
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
||
)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
"""应用生命周期管理"""
|
||
# 启动时
|
||
logger.info("=" * 50)
|
||
logger.info("WPS Bot Game 启动中...")
|
||
logger.info("=" * 50)
|
||
|
||
# 初始化数据库
|
||
db = get_db()
|
||
logger.info(f"数据库初始化完成: {db.db_path}")
|
||
|
||
# 启动后台清理任务
|
||
cleanup_task = asyncio.create_task(periodic_cleanup())
|
||
logger.info("后台清理任务已启动")
|
||
|
||
logger.info("应用启动完成!")
|
||
|
||
yield
|
||
|
||
# 关闭时
|
||
logger.info("应用正在关闭...")
|
||
cleanup_task.cancel()
|
||
try:
|
||
await cleanup_task
|
||
except asyncio.CancelledError:
|
||
pass
|
||
|
||
db.close()
|
||
logger.info("应用已关闭")
|
||
|
||
|
||
async def periodic_cleanup():
|
||
"""定期清理过期会话"""
|
||
while True:
|
||
try:
|
||
await asyncio.sleep(300) # 每5分钟执行一次
|
||
db = get_db()
|
||
db.cleanup_old_sessions(SESSION_TIMEOUT)
|
||
except asyncio.CancelledError:
|
||
break
|
||
except Exception as e:
|
||
logger.error(f"清理任务错误: {e}", exc_info=True)
|
||
|
||
|
||
# 创建FastAPI应用
|
||
app = FastAPI(**APP_CONFIG, lifespan=lifespan)
|
||
|
||
# 添加并发限制中间件
|
||
app.add_middleware(ConcurrencyLimitMiddleware)
|
||
|
||
# 注册路由
|
||
app.include_router(callback.router, prefix="/api", tags=["callback"])
|
||
app.include_router(health.router, tags=["health"])
|
||
app.include_router(private.router, prefix="/api", tags=["private"])
|
||
|
||
|
||
@app.get("/")
|
||
async def root():
|
||
"""根路径"""
|
||
return JSONResponse({
|
||
"message": "WPS Bot Game API",
|
||
"version": "1.0.0",
|
||
"status": "running"
|
||
})
|
||
|
||
|
||
@app.exception_handler(Exception)
|
||
async def global_exception_handler(request, exc):
|
||
"""全局异常处理"""
|
||
logger.error(f"未捕获的异常: {exc}", exc_info=True)
|
||
return JSONResponse(
|
||
status_code=500,
|
||
content={"error": "Internal Server Error", "detail": str(exc)}
|
||
)
|
||
|
||
|
||
def parse_args():
|
||
"""解析命令行参数"""
|
||
parser = argparse.ArgumentParser(description='WPS Bot Game')
|
||
parser.add_argument('--webhook-url', '-w',
|
||
help='WPS Webhook URL')
|
||
parser.add_argument('--host', '-H',
|
||
default='0.0.0.0',
|
||
help='服务器主机地址 (默认: 0.0.0.0)')
|
||
parser.add_argument('--port', '-p',
|
||
type=int, default=11000,
|
||
help='服务器端口 (默认: 11000)')
|
||
parser.add_argument('--workers',
|
||
type=int, default=1,
|
||
help='工作进程数 (默认: 1)')
|
||
parser.add_argument('--log-level',
|
||
default='info',
|
||
choices=['debug', 'info', 'warning', 'error'],
|
||
help='日志级别 (默认: info)')
|
||
return parser.parse_args()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
|
||
# 解析命令行参数
|
||
args = parse_args()
|
||
|
||
# 如果提供了webhook URL，设置环境变量
|
||
if args.webhook_url:
|
||
SetWebhookURL(args.webhook_url)
|
||
logger.info(f"设置Webhook URL: {args.webhook_url}")
|
||
|
||
# 启动服务器
|
||
uvicorn.run(
|
||
"app:app",
|
||
host=args.host,
|
||
port=args.port,
|
||
workers=args.workers,
|
||
limit_concurrency=5,
|
||
log_level=args.log_level
|
||
)
|
||
|