From 5acc65a6f09fc9d759020cc7518d3992ba540139 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com>
Date: Thu, 15 Sep 2022 17:23:39 +0800
Subject: [PATCH] =?UTF-8?q?=E7=9B=B8=E5=85=B3=E6=96=B9=E5=88=9B=E5=BB=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/rpm/serializers.py | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/apps/rpm/serializers.py b/apps/rpm/serializers.py
index c5387b0d..3730c758 100644
--- a/apps/rpm/serializers.py
+++ b/apps/rpm/serializers.py
@@ -24,28 +24,28 @@ class RpartyCreateUpdateSerializer(CustomModelSerializer):
             'belong_dept': {'required': True}
         }
 
+    @transaction.atomic
     def create(self, validated_data):
         instance = super().create(validated_data)
-        with transaction.atomic():
-            dept = Dept.objects.get_or_create(name=instance.name,
-                                              defaults={
-                                                  "name": instance.name,
-                                                  "parent": instance.belong_dept,
-                                                  "type": "rparty"
-                                              })
-            sync_dahua_dept(dept)
+        dept = Dept.objects.get_or_create(name=instance.name,
+                                          defaults={
+                                              "name": instance.name,
+                                              "parent": instance.belong_dept,
+                                              "type": "rparty"
+                                          })
+        sync_dahua_dept(dept)
         instance.dept = dept
         instance.save()
         return instance
 
+    @transaction.atomic
     def update(self, instance, validated_data):
         instance = super().update(instance, validated_data)
-        with transaction.atomic():
-            dept = instance.dept
-            dept.name = instance.name
-            dept.parent = instance.belong_dept  # 重新规划归属部门
-            dept.save()
-            sync_dahua_dept(dept)
+        dept = instance.dept
+        dept.name = instance.name
+        dept.parent = instance.belong_dept  # 重新规划归属部门
+        dept.save()
+        sync_dahua_dept(dept)
         return instance