任务登记模块更新

models/taskRegistration_base.py

@@ -32,4 +32,11 @@ class TaskRegistration(Base):
 
     # 你可以添加更多的字段和关系，例如与用户模型的外键关系
     # user_id = Column(Integer, ForeignKey('users.id'))
-    # user = relationship("User", back_populates="task_registrations")
+    # user = relationship("User", back_populates="task_registrations")
+
+class TaskUnit(Base):
+    __tablename__ = 'task_unit'
+
+    id = Column(Integer, primary_key=True, autoincrement=True, index=True,comment='id')
+    unit_name = Column(String(255), nullable=False, comment='部门名称')
+    unit_duty = Column(Text, nullable=False,comment='部门职责')

routers/api/taskRegistration/__init__.py

@@ -200,13 +200,33 @@ async def delete_task(
                 "code": 200,
                 "msg": "任务删除成功",
                 "data": {
-                    "task_id": task_entry.task_id
+                        "task_id": task_entry.task_id
+                    }
                 }
-            }
         except Exception as e:
             db.rollback()
             raise HTTPException(status_code=500, detail=f"An error occurred while deleting the task: {str(e)}")
 
+    except HTTPException as e:
+        raise e
+    except Exception as e:
+        db.rollback()
+        raise HTTPException(status_code=500, detail=f"An unexpected error occurred: {str(e)}")
+
+@router.get('/selectUnit')
+async def select_unit(
+        request: Request,
+        db: Session = Depends(get_db),
+        query: TaskQuery = Depends(),
+        user_id=Depends(valid_access_token),
+        ):
+    try:
+        data_query = db.query(TaskUnit).filter(TaskUnit.id != '2').all()
+        return {
+            "code": 200,
+            "msg": "查询成功",
+            "data": data_query
+        }
     except HTTPException as e:
         raise e
     except Exception as e: