限制每人日报填写为一次

Router/DailyRouter.py

@@ -8,7 +8,7 @@ from docx.oxml.ns import qn
 from docx.shared import Pt, Length, Inches
 from docxtpl import DocxTemplate
 from fastapi import APIRouter, Depends, HTTPException
-from sqlalchemy import cast, DATE, func, or_
+from sqlalchemy import cast, DATE, func, or_, and_
 from sqlalchemy.orm import Session
 from starlette.responses import FileResponse
 from docx import Document
@@ -77,9 +77,15 @@ checker = SensitiveWordChecker(words)
 @router.post("/daily_add", response_model=DailySchemas.DailyAddRes, summary="添加日报")
 def daily_add(req: DailySchemas.DailyAddReq, db: Session = Depends(get_db),
               token_data: TokenData = Depends(token_data_depend)):
-    if db.query(Daily).filter(
-        Daily.type == DailyTypeEnum.部门子公司日报 and Daily.fill_user == token_data.email and cast(Daily.daily_time,
-                                                                                                    DATE) == datetime.date.today()):
+    # if db.query(Daily).filter(
+    #         Daily.type == DailyTypeEnum.部门子公司日报 and Daily.fill_user == token_data.email and cast(
+    #             Daily.daily_time,
+    #             DATE) == datetime.date.today()).first():
+    if req.type== DailyTypeEnum.部门子公司日报 and db.query(Daily).filter(
+            and_(Daily.type == DailyTypeEnum.部门子公司日报, Daily.fill_user == token_data.email), cast(
+                Daily.daily_time,
+                DATE) == datetime.date.today()
+    ).first():
         raise HTTPException(detail="今日日报已提交，若需修改请在原日报上修改", status_code=305)
     if req.content:
         bad_words = [item[1] for item in checker.check(req.content)]