From 4c4e637cc328a469ab810d841d55f9e8c99b6ae4 Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Mon, 3 Apr 2023 14:01:38 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20rparty=20delete=E5=90=8C=E6=AD=A5dept?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/rpm/views.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/apps/rpm/views.py b/apps/rpm/views.py
index 329155e0..7dc2339a 100644
--- a/apps/rpm/views.py
+++ b/apps/rpm/views.py
@@ -37,6 +37,13 @@ class RpartyViewSet(CustomModelViewSet):
         if user.is_authenticated and user.type == 'remployee':
             queryset = queryset.filter(dept=user.belong_dept)
         return queryset
+    
+    def destroy(self, request, *args, **kwargs):
+        obj = self.get_object()
+        if obj.dept:
+            obj.dept.delete()
+        obj.delete()
+        return Response(status=204)
 
     @action(methods=['post'], detail=True, perms_map={'post': 'rparty:assgin'}, serializer_class=RpartyAssignSerializer)
     @transaction.atomic