feat: 接入mysql数据库

routers/__init__.py

@@ -1,5 +1,7 @@
 from .agent import router as agent_router
+from .report import router as report_router
 
 routers = [
-    agent_router
+    agent_router,
+    report_router
 ]

routers/agent.py

@@ -1,31 +1,33 @@
-from fastapi import APIRouter
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
 
-from schema.agent import ReportRequest
+from config.database import get_db
+from schema.agent import ReportAgentRequest
 from service import agent_service
 
 router = APIRouter(prefix="/agent", tags=["Agent"])
 
 
-@router.post("/tip/generate")
-def generate_tip():
-    return agent_service.query_tip_agent()
+@router.post("/article/generate")
+def generate_article(db: Session = Depends(get_db)):
+    return agent_service.query_article_agent(db)
 
 
-@router.get("/tip/latest")
-def latest_tip():
-    return agent_service.get_latest_tip()
+@router.get("/article/latest")
+def latest_article(db: Session = Depends(get_db)):
+    return agent_service.get_latest_article(db)
 
 
 @router.post("/recipe/generate")
-def generate_recipe():
-    return agent_service.query_recipe_agent()
+def generate_recipe(db: Session = Depends(get_db)):
+    return agent_service.query_recipe_agent(db)
 
 
 @router.get("/recipe/latest")
-def latest_recipe():
-    return agent_service.get_latest_recipe()
+def latest_recipe(db: Session = Depends(get_db)):
+    return agent_service.get_latest_recipe(db)
 
 
 @router.post("/report")
-def query_report(req: ReportRequest):
+def query_report(req: ReportAgentRequest):
     return agent_service.query_report_agent(req)

routers/report.py

@@ -0,0 +1,18 @@
+from fastapi import APIRouter, Depends
+from sqlalchemy.orm import Session
+
+from config.database import get_db
+from schema.report import ReportRequest
+from service import report_service
+
+router = APIRouter(prefix="/report", tags=["Report"])
+
+
+@router.get("/{id}")
+def query_report(id: int, db: Session = Depends(get_db)):
+    return report_service.query_report(id, db)
+
+
+@router.post("")
+def upload_report(req: ReportRequest, db: Session = Depends(get_db)):
+    return report_service.upload_report(req.content, db)