diff --git a/PythonProject/dao/cla_dao.py b/PythonProject/dao/cla_dao.py index f491b9f..23267a9 100644 --- a/PythonProject/dao/cla_dao.py +++ b/PythonProject/dao/cla_dao.py @@ -3,17 +3,17 @@ from fastapi import HTTPException from schema.cla_schema import claRequest -def add_class_dao(c:claRequest,session): +def add_class_dao(c: claRequest, session): try: data = c.model_dump() d = ClassManagement(**data) session.add(d) + session.commit() + session.refresh(d) + return d except: session.rollback() - return '输入有误,请重新添加班级!' - else: - session.commit() - return '添加成功!' + raise HTTPException(400, "输入有误,请重新添加班级!") def get_class_dao(n,m,session): try: