更改新增dao bug

2026-09-22 23:01:59 +08:00
parent f2e525b5a5
commit 6e189d0c65
2 changed files with 10 additions and 3 deletions
@@ -4,6 +4,12 @@ from sqlalchemy.orm import Session
 #新增班级
 def add_class_dao(c,db):
    try:
+        exist_class = db.query(ClassInfo).filter(
+            ClassInfo.num == c["num"],
+            ClassInfo.is_delete == 0
+        ).first()
+        if exist_class:
+            return False
        c1 = ClassInfo(**c)
        db.add(c1)
    except:
@@ -11,6 +17,7 @@ def add_class_dao(c,db):
        return False
    else:
        db.commit()
+        db.refresh(c1)
        return True