1.0.0初始化源代码

backend/app/main.py

@@ -0,0 +1,95 @@
+"""
+FastAPI 应用主文件
+"""
+import logging
+from contextlib import asynccontextmanager
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from loguru import logger as loguru_logger
+
+from app.core.config import settings
+from app.models.base import init_db
+from app.api.router import api_router
+from app.services.scheduler_service import scheduler_service
+
+# 配置日志
+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("Starting application...")
+    
+    # 初始化数据库
+    init_db()
+    logger.info("Database initialized")
+    
+    # 启动定时任务
+    scheduler_service.start()
+    scheduler_service.add_interaction_task()
+    scheduler_service.add_login_task(hour=8, minute=0)
+    scheduler_service.reset_daily_counters(hour=0, minute=1)
+    logger.info("Scheduler started")
+    
+    yield
+    
+    # 关闭时执行
+    logger.info("Shutting down application...")
+    scheduler_service.stop()
+
+
+# 创建 FastAPI 应用
+app = FastAPI(
+    title=settings.APP_NAME,
+    version=settings.APP_VERSION,
+    description="会会虚拟用户 AI 互动系统后端 API",
+    lifespan=lifespan
+)
+
+# 配置 CORS
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # 生产环境应该配置具体的域名
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 注册路由
+app.include_router(api_router, prefix=settings.API_PREFIX)
+
+
+@app.get("/")
+async def root():
+    """根路径"""
+    return {
+        "name": settings.APP_NAME,
+        "version": settings.APP_VERSION,
+        "status": "running"
+    }
+
+
+@app.get("/health")
+async def health_check():
+    """健康检查"""
+    return {
+        "status": "healthy",
+        "scheduler_running": scheduler_service.is_running
+    }
+
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(
+        "app.main:app",
+        host="0.0.0.0",
+        port=8000,
+        reload=settings.DEBUG
+    )