"""Independent sample API for synthetic work orders and monitoring data.""" from typing import get_args from uuid import UUID from fastapi import Depends, FastAPI, Query, Request from fastapi.exceptions import RequestValidationError from fastapi.middleware.cors import CORSMiddleware from fastapi.responses import JSONResponse from app.auth.project_dependencies import get_metadata_repository from app.infra.db.metadb.repositories.metadata_repository import MetadataRepository from .auth import Identity, identity from .config import SampleSettings from .errors import SampleFailure from .routing import resolve_store_routing from .store import TRANSITIONS from .schemas import ( AlertPage, Context, CreateTestDataRecord, CreateWorkOrder, DeleteResource, DeletedResource, DemoDataset, Observations, PointPage, Receipt, SampleError, Scenario, ScenarioCommand, ScenarioState, TestDataRecord, TestDataRecordPage, UpdateTestDataRecord, UpdateWorkOrder, WorkOrder, WorkOrderPage, ) async def store(request: Request, who: Identity = Depends(identity), repository: MetadataRepository = Depends(get_metadata_repository)): from .store import PostgresStore routing = await resolve_store_routing( who, repository, timeseries=request.url.path.endswith("/monitoring/observations"), ) return PostgresStore(request.app.state.settings, routing=routing) def create_app(settings: SampleSettings | None = None) -> FastAPI: app = FastAPI(title="TJWater 外围系统模拟接入", version="1.0.0", description="测试监测与核查反馈样板，不是正式工单平台或设备控制服务。") app.state.settings = settings or SampleSettings() app.add_middleware(CORSMiddleware, allow_origins=app.state.settings.cors_origins, allow_credentials=False, allow_methods=["GET", "POST", "PATCH", "DELETE"], allow_headers=["Authorization", "Content-Type", "X-Project-Id"]) @app.exception_handler(SampleFailure) async def failure(request, exc): return JSONResponse(status_code=exc.status, content=exc.payload()) @app.exception_handler(RequestValidationError) async def invalid(request, exc): return JSONResponse(status_code=422, content=SampleFailure("SCHEMA_INVALID", "请求字段不符合样板契约").payload()) @app.get("/health") def health(): return {"service": "peripheral-sample", "origin": "synthetic", "storage": "postgresql", "status": "up", "storage_verified": False} prefix = "/api/sample/v1" errors = {code: {"model": SampleError} for code in (401, 403, 404, 409, 422, 502, 503)} @app.get(prefix + "/context", response_model=Context, responses=errors) def context(request: Request, who: Identity = Depends(identity), db=Depends(store)): return {"protocol_version": "tjwater-peripheral-sample@1", "origin": "synthetic", "project_id": who.project_id, "user_id": who.user_id, "can_write": who.can_write, "scenarios": list(get_args(Scenario)) if request.app.state.settings.enable_faults else ["normal"], "state_transitions": TRANSITIONS, "managed_resources": db.managed_resource_capabilities()} @app.get(prefix + "/scenario", response_model=ScenarioState, responses=errors) def scenario(who: Identity = Depends(identity), db=Depends(store)): return db.scenario(who) @app.post(prefix + "/scenario", response_model=ScenarioState, responses=errors) def configure_scenario(command: ScenarioCommand, who: Identity = Depends(identity), db=Depends(store)): return db.set_scenario(who, command) @app.get(prefix + "/requests/{request_id}", response_model=Receipt, responses=errors) def receipt(request_id: UUID, who: Identity = Depends(identity), db=Depends(store)): return db.receipt(who, request_id) @app.get(prefix + "/work-orders", response_model=WorkOrderPage, responses=errors) def work_orders(limit: int = Query(20, ge=1, le=100), cursor: UUID | None = None, who: Identity = Depends(identity), db=Depends(store)): return db.list_work_orders(who, limit, cursor) @app.post(prefix + "/work-orders", response_model=WorkOrder, status_code=201, responses=errors) def create_work_order(command: CreateWorkOrder, who: Identity = Depends(identity), db=Depends(store)): return db.create_work_order(who, command) @app.get(prefix + "/work-orders/{work_order_id}", response_model=WorkOrder, responses=errors) def work_order(work_order_id: UUID, who: Identity = Depends(identity), db=Depends(store)): return db.get_work_order(who, work_order_id) @app.patch(prefix + "/work-orders/{work_order_id}", response_model=WorkOrder, responses=errors) def update_work_order(work_order_id: UUID, command: UpdateWorkOrder, who: Identity = Depends(identity), db=Depends(store)): return db.update_work_order(who, work_order_id, command) @app.delete(prefix + "/work-orders/{work_order_id}", response_model=DeletedResource, responses=errors) def delete_work_order(work_order_id: UUID, command: DeleteResource, who: Identity = Depends(identity), db=Depends(store)): return db.delete_work_order(who, work_order_id, command) @app.get(prefix + "/test-data", response_model=TestDataRecordPage, responses=errors) def test_data(limit: int = Query(20, ge=1, le=100), cursor: UUID | None = None, who: Identity = Depends(identity), db=Depends(store)): return db.list_test_data(who, limit, cursor) @app.post(prefix + "/test-data", response_model=TestDataRecord, status_code=201, responses=errors) def create_test_data(command: CreateTestDataRecord, who: Identity = Depends(identity), db=Depends(store)): return db.create_test_data(who, command) @app.get(prefix + "/test-data/{record_id}", response_model=TestDataRecord, responses=errors) def test_data_record(record_id: UUID, who: Identity = Depends(identity), db=Depends(store)): return db.get_test_data(who, record_id) @app.patch(prefix + "/test-data/{record_id}", response_model=TestDataRecord, responses=errors) def update_test_data(record_id: UUID, command: UpdateTestDataRecord, who: Identity = Depends(identity), db=Depends(store)): return db.update_test_data(who, record_id, command) @app.delete(prefix + "/test-data/{record_id}", response_model=DeletedResource, responses=errors) def delete_test_data(record_id: UUID, command: DeleteResource, who: Identity = Depends(identity), db=Depends(store)): return db.delete_test_data(who, record_id, command) @app.get(prefix + "/monitoring/points", response_model=PointPage, responses=errors) def points(limit: int = Query(100, ge=1, le=500), cursor: str | None = Query(None, max_length=128), who: Identity = Depends(identity), db=Depends(store)): return db.points(who, limit, cursor) @app.get(prefix + "/monitoring/observations", response_model=Observations, responses=errors) def observations(point_id: str, start: str, end: str, who: Identity = Depends(identity), db=Depends(store)): return db.observations(who, point_id, start, end) @app.get(prefix + "/monitoring/alerts", response_model=AlertPage, responses=errors) def alerts(who: Identity = Depends(identity), db=Depends(store)): return db.alerts(who) @app.get(prefix + "/demo-data", response_model=DemoDataset, responses=errors) def demo_data(who: Identity = Depends(identity), db=Depends(store)): return db.demo_data(who) return app app = create_app()