Pine
ac2df72b21
- 删 UserIdentity 模型/IdentityRepository/db.identities，账号统一单一角色(users.role) - auth.py 去 select-identity/_identity_summaries，登录恒返回 identities=[]；_login_response_for_user 按单角色签发 - 删多端口路由 rbac_government/investor/developer 及其 main 挂载 - seed 去多身份回填(_migrate_identities/_ensure_port_agents/多端身份/端口标签映射) - dependencies: require_port 去端口隔离(单角色放宽)、optional_current_user 修 identity_id 多余参数 - user_admin_service create_user 不再建端口身份；schemas 去 SelectIdentityRequest - 新增迁移 0014_drop_user_identities(drop user_identities 表) - 测试改写为单角色契约(登录空 identities、无 select-identity、账号级智能体) Co-Authored-By: Claude <noreply@anthropic.com>
103 lines
3.2 KiB
Python
103 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""FastAPI 应用入口（接口层组装）。
|
|
|
|
启动方式（项目根目录）：
|
|
uv run uvicorn app.main:app --host 127.0.0.1 --port 8090
|
|
或使用脚本：``uv run python -m app``
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from app import config
|
|
from app.infrastructure.repositories import Database
|
|
from app.api.routers import agents as agents_router
|
|
from app.api.routers import auth as auth_router
|
|
from app.api.routers import bootstrap as bootstrap_router
|
|
from app.api.routers import rbac_admin as rbac_admin_router
|
|
from app.api.routers import rbac_opc as rbac_opc_router
|
|
from app.api.routers import rbac_operator as rbac_operator_router
|
|
from app.api.routers import rbac_org as rbac_org_router
|
|
from app.api.routers import rbac_ecosystem as rbac_ecosystem_router
|
|
from app.api.routers import rbac_portals as rbac_portals_router
|
|
from app.api.routers import templates as templates_router
|
|
from app.api.routers import relay as relay_router
|
|
|
|
APP_NAME = "云超服Agents 演示后端"
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""启动时初始化数据库（建表 + 幂等种子）。
|
|
|
|
若 ``app.state.db`` 已由外部注入（如测试夹具），则直接复用，
|
|
不重新创建，避免覆盖调用方准备的数据库实例。
|
|
"""
|
|
db = getattr(app.state, "db", None)
|
|
if db is None:
|
|
db = Database()
|
|
await db.initialize()
|
|
app.state.db = db
|
|
yield
|
|
if db is not None and db._owns_session:
|
|
await db.close()
|
|
# 释放引擎连接池，避免连接残留导致下次启动 initialize() 被锁挂起/连接泄漏
|
|
engine = getattr(db, "_engine", None)
|
|
if engine is not None:
|
|
try:
|
|
await engine.dispose()
|
|
except Exception: # noqa: BLE001
|
|
pass
|
|
|
|
|
|
app = FastAPI(
|
|
title=APP_NAME,
|
|
description=(
|
|
"云超服核心服务端：身份/平台 API + 培训子应用（/api/*），统一对外 opc.pinesound.cn。"
|
|
"异步四层架构（接口/业务/领域/基础设施）。"
|
|
),
|
|
version="0.1.0",
|
|
lifespan=lifespan,
|
|
)
|
|
|
|
# 演示服务允许任意来源（实际生产由主后端网关收敛）。
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
app.include_router(auth_router.router)
|
|
app.include_router(agents_router.router)
|
|
app.include_router(templates_router.router)
|
|
app.include_router(bootstrap_router.router)
|
|
app.include_router(rbac_admin_router.router)
|
|
app.include_router(rbac_opc_router.router)
|
|
app.include_router(rbac_operator_router.router)
|
|
app.include_router(rbac_org_router.router)
|
|
app.include_router(rbac_portals_router.router)
|
|
app.include_router(rbac_ecosystem_router.router)
|
|
app.include_router(relay_router.router)
|
|
|
|
|
|
@app.get("/health", tags=["meta"], summary="健康检查")
|
|
async def health():
|
|
return {
|
|
"status": "ok",
|
|
"service": "pineagents-demo-server",
|
|
"auth_enabled": config.AUTH_ENABLED,
|
|
}
|
|
|
|
|
|
@app.get("/", tags=["meta"], include_in_schema=False)
|
|
async def root():
|
|
return {
|
|
"service": APP_NAME,
|
|
"docs": "/docs",
|
|
"health": "/health",
|
|
}
|