feat(orgs): 修改OrgMemberRelationAPI，支持通过查询参数控制是否忽略已存在的数据 (#4720)

* feat(orgs): 修改OrgMemberRelationAPI，支持通过查询参数控制是否忽略已存在的数据

* feat(orgs): 修改构建数据库查询参数的问题

Co-authored-by: Bai <bugatti_it@163.com>

apps/orgs/api.py

@@ -2,9 +2,11 @@
 #
 
 from django.utils.translation import ugettext as _
+from django.db.models import Q
 from rest_framework import status, generics
 from rest_framework.views import Response
 from rest_framework_bulk import BulkModelViewSet
+from rest_framework.mixins import CreateModelMixin
 
 from common.permissions import IsSuperUserOrAppUser
 from common.drf.api import JMSBulkRelationModelViewSet
@@ -19,6 +21,8 @@ from perms.models import AssetPermission
 from orgs.utils import current_org
 from common.utils import get_logger
 from .filters import OrgMemberRelationFilterSet
+from .models import OrganizationMember
+
 
 logger = get_logger(__file__)
 
@@ -71,6 +75,57 @@ class OrgMemberRelationBulkViewSet(JMSBulkRelationModelViewSet):
     serializer_class = OrgMemberSerializer
     filterset_class = OrgMemberRelationFilterSet
 
+    @staticmethod
+    def clear_request_data(request):
+        data = request.data
+
+        ignore_already_exist = request.query_params.get('ignore_already_exist')
+        if not ignore_already_exist:
+            return data
+
+        query_params = Q()
+        for _data in data:
+            query_fields = {}
+            org = _data.get('org')
+            if org:
+                query_fields.update({'org': org})
+            user = _data.get('user')
+            if user:
+                query_fields.update({'user': user})
+            role = _data.get('role')
+            if role:
+                query_fields.update({'role': role})
+            query_params |= Q(**query_fields)
+
+        if not query_params:
+            return data
+
+        members = OrganizationMember.objects.filter(query_params)
+        members = [
+            {'org': str(member.org_id), 'user': str(member.user_id), 'role': member.role}
+            for member in members
+        ]
+        if not members:
+            return data
+
+        for member in members:
+            if member in data:
+                data.remove(member)
+        return data
+
+    def create(self, request, *args, **kwargs):
+        bulk = isinstance(request.data, list)
+
+        if not bulk:
+            return CreateModelMixin.create(self, request, *args, **kwargs)
+
+        else:
+            data = self.clear_request_data(request)
+            serializer = self.get_serializer(data=data, many=True)
+            serializer.is_valid(raise_exception=True)
+            self.perform_bulk_create(serializer)
+            return Response(serializer.data, status=status.HTTP_201_CREATED)
+
     def perform_bulk_destroy(self, queryset):
         objs = list(queryset.all().prefetch_related('user', 'org'))
         queryset.delete()