baoyubo 2 mesi fa
parent
commit
25851448d3
1 ha cambiato i file con 5 aggiunte e 3 eliminazioni
  1. 5 3
      routers/api/emergencyPlans/contact.py

+ 5 - 3
routers/api/emergencyPlans/contact.py

@@ -399,7 +399,7 @@ async def create_contact(
             if pd.isna(info['职务']):
                 info['职务'] = None
             infos.append(info)
-
+        new_data = []
         # 创建新的预案记录
         for contact in infos:
             unit_id = db_dept.get_dept_id_by_name(db, contact['单位名称'])
@@ -418,9 +418,9 @@ async def create_contact(
                 create_by = user_id,
                 sign = ''
             )
-
+            new_data.append(new_contact)
             # 添加到数据库会话
-            db.add(new_contact)
+        db.add_all(new_data)
         # 提交
         db.commit()
 
@@ -437,6 +437,7 @@ async def create_contact(
         }
     
     except AppException as e:
+        db.rollback()
         return {
             "code": 500,
             "msg": e.msg
@@ -445,4 +446,5 @@ async def create_contact(
     except Exception as e:
         traceback.print_exc()
         # 处理异常
+        db.rollback()
         raise HTTPException(status_code=500, detail=str(e))