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