Update project

backend/app/api/routes.py

@@ -1,8 +1,23 @@
-from fastapi import APIRouter, Query
+from fastapi import APIRouter, File, Form, Query, UploadFile
 
+from app.api.analysis_schemas import AnalysisChartResponse, AnalysisReportResponse
+from app.api.capital_image_schemas import (
+    CapitalImageListResponse,
+    CapitalImageRecordDetail,
+    CapitalImageUploadResponse,
+)
+from app.api.main_capital_flow_schemas import (
+    MainCapitalFlowCreateRequest,
+    MainCapitalFlowCreateResponse,
+    MainCapitalFlowDeleteResponse,
+    MainCapitalFlowListResponse,
+    MainCapitalFlowRecognizeResponse,
+    MainCapitalFlowRecordDetail,
+)
 from app.api.schemas import (
     AShareIndexFlowResponse,
     AShareSectorFlowResponse,
+    EtfRealtimeResponse,
     HealthResponse,
     HistoryResponse,
     MetaResponse,
@@ -12,8 +27,12 @@ from app.api.schemas import (
     RulesResponse,
     SourceDiagnosticsResponse,
 )
+from app.services.analysis_service import analysis_service
 from app.services.ashare_flow_service import ashare_flow_service
 from app.services.alert_service import alert_service
+from app.services.capital_image_service import capital_image_service
+from app.services.etf_monitor_service import etf_monitor_service
+from app.services.main_capital_flow_service import main_capital_flow_service
 from app.services.monitoring_service import monitoring_service
 
 router = APIRouter()
@@ -24,6 +43,74 @@ def health() -> HealthResponse:
     return HealthResponse(status="ok")
 
 
+@router.get("/capital-images", response_model=CapitalImageListResponse)
+def capital_image_list(
+    trade_date: str | None = Query(default=None),
+    subject: str | None = Query(default=None),
+) -> CapitalImageListResponse:
+    return CapitalImageListResponse(**capital_image_service.list_records(trade_date, subject))
+
+
+@router.get("/capital-images/{record_id}", response_model=CapitalImageRecordDetail)
+def capital_image_detail(record_id: str) -> CapitalImageRecordDetail:
+    return CapitalImageRecordDetail(**capital_image_service.get_record(record_id))
+
+
+@router.post("/capital-images/upload", response_model=CapitalImageUploadResponse)
+async def capital_image_upload(
+    image: UploadFile = File(...),
+    trade_date: str | None = Form(default=None),
+    subject: str | None = Form(default=None),
+) -> CapitalImageUploadResponse:
+    return CapitalImageUploadResponse(
+        **await capital_image_service.create_record(image, trade_date=trade_date, subject=subject)
+    )
+
+
+@router.get("/main-capital-flows", response_model=MainCapitalFlowListResponse)
+def main_capital_flow_list() -> MainCapitalFlowListResponse:
+    return MainCapitalFlowListResponse(**main_capital_flow_service.list_records())
+
+
+@router.get("/main-capital-flows/{record_id}", response_model=MainCapitalFlowRecordDetail)
+def main_capital_flow_detail(record_id: str) -> MainCapitalFlowRecordDetail:
+    return MainCapitalFlowRecordDetail(**main_capital_flow_service.get_record(record_id))
+
+
+@router.post("/main-capital-flows/recognize", response_model=MainCapitalFlowRecognizeResponse)
+async def main_capital_flow_recognize(
+    image: UploadFile = File(...),
+    trade_date: str | None = Form(default=None),
+    subject: str | None = Form(default=None),
+) -> MainCapitalFlowRecognizeResponse:
+    return MainCapitalFlowRecognizeResponse(
+        **await main_capital_flow_service.recognize_image(image, trade_date=trade_date, subject=subject)
+    )
+
+
+@router.post("/main-capital-flows", response_model=MainCapitalFlowCreateResponse)
+def main_capital_flow_create(payload: MainCapitalFlowCreateRequest) -> MainCapitalFlowCreateResponse:
+    return MainCapitalFlowCreateResponse(**main_capital_flow_service.create_record(payload.model_dump()))
+
+
+@router.delete("/main-capital-flows/{record_id}", response_model=MainCapitalFlowDeleteResponse)
+def main_capital_flow_delete(record_id: str) -> MainCapitalFlowDeleteResponse:
+    return MainCapitalFlowDeleteResponse(**main_capital_flow_service.delete_record(record_id))
+
+
+@router.get("/analysis/report", response_model=AnalysisReportResponse)
+def analysis_report(q: str = Query(..., min_length=2)) -> AnalysisReportResponse:
+    return analysis_service.build_report(q)
+
+
+@router.get("/analysis/chart", response_model=AnalysisChartResponse)
+def analysis_chart(
+    q: str = Query(..., min_length=2),
+    cycle: str = Query(default="day"),
+) -> AnalysisChartResponse:
+    return analysis_service.build_chart(q, cycle)
+
+
 @router.get("/meta", response_model=MetaResponse)
 def meta() -> MetaResponse:
     return monitoring_service.get_meta()
@@ -77,3 +164,23 @@ def ashare_sector_realtime() -> AShareSectorFlowResponse:
 @router.get("/ashare/sector-flows/daily", response_model=AShareSectorFlowResponse)
 def ashare_sector_daily(trade_date: str | None = Query(default=None)) -> AShareSectorFlowResponse:
     return AShareSectorFlowResponse(**ashare_flow_service.get_sector_daily(trade_date))
+
+
+@router.get("/etf/broad/realtime", response_model=EtfRealtimeResponse)
+def etf_broad_realtime() -> EtfRealtimeResponse:
+    return EtfRealtimeResponse(**etf_monitor_service.get_group_realtime("broad"))
+
+
+@router.get("/etf/sector/realtime", response_model=EtfRealtimeResponse)
+def etf_sector_realtime() -> EtfRealtimeResponse:
+    return EtfRealtimeResponse(**etf_monitor_service.get_group_realtime("sector"))
+
+
+@router.get("/etf/broad/daily", response_model=EtfRealtimeResponse)
+def etf_broad_daily(trade_date: str | None = Query(default=None)) -> EtfRealtimeResponse:
+    return EtfRealtimeResponse(**etf_monitor_service.get_group_daily("broad", trade_date))
+
+
+@router.get("/etf/sector/daily", response_model=EtfRealtimeResponse)
+def etf_sector_daily(trade_date: str | None = Query(default=None)) -> EtfRealtimeResponse:
+    return EtfRealtimeResponse(**etf_monitor_service.get_group_daily("sector", trade_date))