任务登记模块更新

routers/api/taskRegistration/__init__.py

@@ -111,32 +111,29 @@ async def select_tasks(
 
 @router.post('/update')
 async def update_task_status(
-    request: Request,
-    db: Session = Depends(get_db),
-    user_id=Depends(valid_access_token)):  # 确保用户已验证
-    try:
-        body = await request.json()
-        task_id_to_use = body.get('task_id')
-        processing_status = body.get('processing_status')
-
-        if not task_id_to_use:
-            return Response(content="Missing required parameter 'task_id'", status_code=400)
-        if processing_status not in ['已完成', '未完成']:
-            return Response(content="processing_status must be '已完成' or '未完成'", status_code=400)
-
-        # 查询要更新的任务
-        # data_query = db.query(TaskRegistration).filter(TaskRegistration.del_flag != '2')
-        task_entry = db.query(TaskRegistration).filter(TaskRegistration.task_id == task_id_to_use).first()
-        print(task_entry)
-        print(not task_entry)
-        if not task_entry:
-            raise HTTPException(status_code=404, detail="任务不存在")
-
-        # 更新任务状态
-        task_entry.processing_status = processing_status
-        task_entry.update_time = datetime.now()  # 更新时间
+        request: Request,
+        db: Session = Depends(get_db),
+        user_id=Depends(valid_access_token)):  # 确保用户已验证
+    body = await request.json()
+    task_id_to_use = body.get('task_id')
+    processing_status = body.get('processing_status')
+
+    if not task_id_to_use:
+        return Response(content="Missing required parameter 'task_id'", status_code=400)
+    if processing_status not in ['已完成', '未完成']:
+        return Response(content="processing_status must be '已完成' or '未完成'", status_code=400)
+
+    task_entry = (db.query(TaskRegistration)
+                  .filter(TaskRegistration.del_flag != '2')
+                  .filter(TaskRegistration.task_id == task_id_to_use))
+    task_entry = task_entry.first()
+    if not task_entry:
+        raise HTTPException(status_code=404, detail="任务不存在")
+
+    task_entry.processing_status = processing_status
+    task_entry.update_time = datetime.now()
 
-        # 提交数据库事务
+    try:
         db.commit()
         return {
             "code": 200,
@@ -147,15 +144,10 @@ async def update_task_status(
                 "update_time": task_entry.update_time.isoformat()
             }
         }
-
-
-
     except Exception as e:
-        traceback.print_exc()
         db.rollback()
         raise HTTPException(status_code=500, detail=str(e))
 
-
 @router.delete('/delete')
 async def delete_task(
     request: Request,