添加接收单完成接口

mods/receive/mods/receive_order/router.py

@@ -1,9 +1,12 @@
+from datetime import datetime
+
 from fastapi import APIRouter, Depends, HTTPException
 from sqlalchemy.orm import Session
 
 from ...common import get_db
 from . import crud
 from . import schemas
+from ...schemas import ReceiveOrderState
 
 router = APIRouter(tags=["接收单"], prefix='/receive_order')
 
@@ -49,11 +52,14 @@ def receive_order_query(req: schemas.ReceiveOrderQueryReq, db: Session = Depends
     return schemas.ReceiveOrderQueryRes(count=count, items=items)
 
 
-@router.post("/all", summary="获取所有接收单", response_model=schemas.ReceiveOrderAllRes)
+@router.post("/finish_order", summary="完成接受单")
-def receive_order_all(req: schemas.ReceiveOrderAllReq, db: Session = Depends(get_db)):
+def finish_order(req: schemas.ReceiveOrderId, db: Session = Depends(get_db)):
-    query = crud.receive_order_all(db)
+    item = crud.receive_order_get(db, req)
-    items = [schemas.ReceiveOrderInfo(**item.to_full_dict(include=req.include,
+    for file in item.files:
-                                                          ex_include=req.ex_include,
+        if not file.file_path:
-                                                          relation_use_id=req.relation_use_id)) for item in query]
+            raise HTTPException(status_code=303, detail='存在未收集文件')
-    return schemas.ReceiveOrderAllRes(items=items)
+    item.state = ReceiveOrderState.finish
+    item.finish_time = datetime.now()
+    db.commit()
+    return '成功'
 #########