""" NeoZQYY 后端 API 入口 基于 FastAPI 构建,为管理后台和微信小程序提供 RESTful API。 OpenAPI 文档自动生成于 /docs(Swagger UI)和 /redoc(ReDoc)。 """ from contextlib import asynccontextmanager from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware from app import config # CHANGE 2026-02-19 | 新增 xcx_test 路由(MVP 验证)+ wx_callback 路由(微信消息推送) # CHANGE 2026-02-23 | 新增 ops_panel 路由(运维控制面板) # CHANGE 2026-02-25 | 新增 xcx_auth 路由(小程序微信登录 + 申请 + 状态查询 + 店铺切换) # CHANGE 2026-02-26 | member_birthday 路由替换为 member_retention_clue(维客线索重构) # CHANGE 2026-02-26 | 新增 admin_applications 路由(管理端申请审核) # CHANGE 2026-02-27 | 新增 xcx_tasks / xcx_notes 路由(小程序核心业务) # CHANGE 2026-03-09 | 新增 xcx_ai_chat 路由(AI SSE 对话 + 历史对话) # CHANGE 2026-03-09 | 新增 xcx_ai_cache 路由(AI 缓存查询) from app.routers import auth, execution, schedules, tasks, env_config, db_viewer, etl_status, xcx_test, wx_callback, member_retention_clue, ops_panel, xcx_auth, admin_applications, business_day, xcx_tasks, xcx_notes, xcx_ai_chat, xcx_ai_cache from app.services.scheduler import scheduler from app.services.task_queue import task_queue from app.ws.logs import ws_router @asynccontextmanager async def lifespan(app: FastAPI): """应用生命周期:启动时拉起后台服务,关闭时优雅停止。""" # CHANGE 2026-03-07 | 启动横幅:打印关键路径,便于诊断连到了哪个实例 import sys _banner = ( "\n" "╔══════════════════════════════════════════════════════╗\n" "║ NeoZQYY Backend — 启动诊断 ║\n" "╠══════════════════════════════════════════════════════╣\n" f"║ Python: {sys.executable}\n" f"║ ROOT: {config._project_root}\n" f"║ ETL_PATH: {config.ETL_PROJECT_PATH}\n" f"║ ETL_PY: {config.ETL_PYTHON_EXECUTABLE}\n" f"║ OPS_BASE: {config.OPS_SERVER_BASE}\n" f"║ APP_DB: {config.APP_DB_NAME}\n" f"║ .env: {config._root_env}\n" "╚══════════════════════════════════════════════════════╝\n" ) print(_banner, flush=True) # 启动 task_queue.start() scheduler.start() # CHANGE 2026-02-27 | 注册触发器 job handler(核心业务模块) from app.services.trigger_scheduler import register_job from app.services import task_generator, task_expiry, recall_detector, note_reclassifier register_job("task_generator", lambda **_kw: task_generator.run()) register_job("task_expiry_check", lambda **_kw: task_expiry.run()) register_job("recall_completion_check", recall_detector.run) register_job("note_reclassify_backfill", note_reclassifier.run) # CHANGE 2026-03-10 | 注册 AI 事件处理器(消费/备注/任务分配 → AI 调用链) try: import os _api_key = os.environ.get("BAILIAN_API_KEY", "") _base_url = os.environ.get("BAILIAN_BASE_URL", "") _model = os.environ.get("BAILIAN_MODEL", "qwen-plus") if _api_key and _base_url: from app.ai.bailian_client import BailianClient from app.ai.cache_service import AICacheService from app.ai.conversation_service import ConversationService from app.ai.dispatcher import AIDispatcher, register_ai_handlers _bailian = BailianClient(api_key=_api_key, base_url=_base_url, model=_model) _dispatcher = AIDispatcher(_bailian, AICacheService(), ConversationService()) register_ai_handlers(_dispatcher) except Exception: import logging as _log _log.getLogger(__name__).warning("AI 事件处理器注册失败,AI 功能不可用", exc_info=True) yield # 关闭 await scheduler.stop() await task_queue.stop() app = FastAPI( title="NeoZQYY API", description="台球门店运营助手 — 后端 API(管理后台 + 微信小程序)", version="0.1.0", docs_url="/docs", redoc_url="/redoc", lifespan=lifespan, ) # ---- CORS 中间件 ---- # 允许来源从环境变量 CORS_ORIGINS 读取,缺省允许 Vite 开发服务器 (localhost:5173) app.add_middleware( CORSMiddleware, allow_origins=config.CORS_ORIGINS, allow_credentials=True, allow_methods=["*"], allow_headers=["*"], ) # ---- 路由注册 ---- app.include_router(auth.router) app.include_router(tasks.router) app.include_router(execution.router) app.include_router(schedules.router) app.include_router(env_config.router) app.include_router(db_viewer.router) app.include_router(etl_status.router) app.include_router(ws_router) app.include_router(xcx_test.router) app.include_router(wx_callback.router) app.include_router(member_retention_clue.router) app.include_router(ops_panel.router) app.include_router(xcx_auth.router) app.include_router(admin_applications.router) app.include_router(business_day.router) app.include_router(xcx_tasks.router) app.include_router(xcx_notes.router) app.include_router(xcx_ai_chat.router) app.include_router(xcx_ai_cache.router) @app.get("/health", tags=["系统"]) async def health_check(): """健康检查端点,用于探活和监控。""" return {"status": "ok"} # CHANGE 2026-03-07 | 诊断端点:返回关键路径配置,用于确认连到的是哪个实例 @app.get("/debug/config-paths", tags=["系统"]) async def debug_config_paths(): """返回当前后端实例的关键路径配置(仅开发环境使用)。""" import sys import os import platform from app.services.cli_builder import cli_builder as _cb from app.schemas.tasks import TaskConfigSchema as _TCS _test_cfg = _TCS(flow="api_ods_dwd", processing_mode="increment_only", tasks=["DWD_LOAD_FROM_ODS"], store_id=123) _test_cmd = _cb.build_command( _test_cfg, config.ETL_PROJECT_PATH, python_executable=config.ETL_PYTHON_EXECUTABLE, ) _test_cmd_str = " ".join(_test_cmd) return { "hostname": platform.node(), "python_executable": sys.executable, "project_root": str(config._project_root), "env_file": str(config._root_env), "etl_python_executable": config.ETL_PYTHON_EXECUTABLE, "etl_project_path": config.ETL_PROJECT_PATH, "simulated_command": _test_cmd_str, "NEOZQYY_ROOT_env": os.environ.get("NEOZQYY_ROOT", "<未设置>"), "cwd": os.getcwd(), }