diff --git a/PythonProject/dao/cla_dao.py b/PythonProject/dao/cla_dao.py new file mode 100644 index 0000000..272c810 --- /dev/null +++ b/PythonProject/dao/cla_dao.py @@ -0,0 +1,85 @@ +from model.all_model import ClassManagement +from fastapi import HTTPException +from schema.cla_schema import claRequest + + +def add_class_dao(c:claRequest,session): + try: + data = c.model_dump() + d = ClassManagement(**data) + session.add(d) + except: + session.rollback() + return '输入有误,请重新添加班级!' + else: + session.commit() + return '添加成功!' + +def get_class_dao(session): + try: + class_list=session.query(ClassManagement).filter(ClassManagement.delete_status==0).all() + return class_list + except: + raise HTTPException(status_code=406,detail="输入查询信息有误,请重新输入!") + +def update_class_dao(id:int,req:ClassManagement,session): + try: + session.query(ClassManagement).filter(ClassManagement.class_id==id)\ + .filter(ClassManagement.delete_status==0).update(req.model_dump(exclude_unset=True)) + session.commit() + except: + raise HTTPException(status_code=400,detail="更新格式错误或参数缺失,请重新输入!") + return '更新成功,请核对信息!' + +def delete_class(id:int,session): + try: + session.query(ClassManagement).filter(ClassManagement.class_id==id,ClassManagement.delete_status==0)\ + .update({'delete_status': 1}) + session.commit() + except: + raise HTTPException(status_code=404, detail="删除异常,请重新核对之后删除!") + return '删除成功,请核对信息!' + +def get_class_for_delete(id: int, session): + try: + obj = ( + session.query(ClassManagement).filter(ClassManagement.class_id == id,ClassManagement.delete_status == 0).first() + ) + if not obj: + raise HTTPException(status_code=404, detail="班级不存在") + except HTTPException: + raise + except Exception: + raise HTTPException(status_code=500, detail="查询异常,请重新核对!") + + return { + "class_id": obj.class_id, + "class_name": obj.class_name, + "start_class_date": obj.start_class_date, + "head_teacher_id": obj.head_teacher_id, + "class_teacher_id": obj.class_teacher_id, + } + + +def hard_delete_class(id: int, session): + try: + obj = ( + session.query(ClassManagement).filter(ClassManagement.class_id == id,ClassManagement.delete_status == 0) + .first() + ) + if not obj: + raise HTTPException(status_code=404, detail="班级不存在") + + session.delete(obj) + session.commit() + except HTTPException: + raise + except Exception: + session.rollback() + raise HTTPException(status_code=500, detail="删除异常,请重新核对之后删除!") + + return '删除成功,请核对信息!' + + + +