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: