From a51a4442ce0347bb9a5fe971b5bf699ba706b7e5 Mon Sep 17 00:00:00 2001
From: Mojtaba-z <mojtaba.zolfaghari30@gmail.com>
Date: Sat, 16 Aug 2025 16:45:58 +0330
Subject: [PATCH] edit assignment

---
 .../web/api/v1/serilaizers/client.py          |  2 +-
 .../web/api/v1/serilaizers/device.py          |  5 ++--
 apps/pos_device/web/api/v1/viewsets/device.py | 23 ++++++++++++++++++-
 3 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/apps/pos_device/web/api/v1/serilaizers/client.py b/apps/pos_device/web/api/v1/serilaizers/client.py
index cce4aa6..0555ca0 100644
--- a/apps/pos_device/web/api/v1/serilaizers/client.py
+++ b/apps/pos_device/web/api/v1/serilaizers/client.py
@@ -13,7 +13,7 @@ class POSClientSerializer(ModelSerializer):
         org = attrs['organization']
 
         # check for duplicate organization client
-        if org:
+        if not self.instance and org:
             if self.Meta.model.objects.filter(organization=org).exists():
                 raise APIException("قبلا کلاینت با این سازمان ثبت شده است", code=status.HTTP_403_FORBIDDEN)  # noqa
 
diff --git a/apps/pos_device/web/api/v1/serilaizers/device.py b/apps/pos_device/web/api/v1/serilaizers/device.py
index 28ef8af..71946ec 100644
--- a/apps/pos_device/web/api/v1/serilaizers/device.py
+++ b/apps/pos_device/web/api/v1/serilaizers/device.py
@@ -56,8 +56,9 @@ class DeviceAssignmentSerializer(ModelSerializer):
         device = attrs['device']
         client = attrs['client']
 
-        if self.Meta.model.objects.filter(device=device, client=client).exists():
-            raise pos_exceptions.DeviceAlreadyAssigned()
+        if not self.instance:
+            if self.Meta.model.objects.filter(device=device, client=client).exists():
+                raise pos_exceptions.DeviceAlreadyAssigned()
 
         return attrs
 
diff --git a/apps/pos_device/web/api/v1/viewsets/device.py b/apps/pos_device/web/api/v1/viewsets/device.py
index 99a827b..736a686 100644
--- a/apps/pos_device/web/api/v1/viewsets/device.py
+++ b/apps/pos_device/web/api/v1/viewsets/device.py
@@ -177,7 +177,7 @@ class DeviceAssignmentViewSet(viewsets.ModelViewSet):
             # if client will be an organization
             if request.data['client_data']['is_organization']:
                 client = pos_models.POSClient.objects.filter(
-                        organization_id=request.data['client_data']['organization']
+                    organization_id=request.data['client_data']['organization']
                 )
 
                 if client.exists():
@@ -207,6 +207,27 @@ class DeviceAssignmentViewSet(viewsets.ModelViewSet):
             return Response(serializer.data, status=status.HTTP_201_CREATED)
         return Response(serializer.errors, status=status.HTTP_403_FORBIDDEN)
 
+    @transaction.atomic
+    def update(self, request, pk=None, *args, **kwargs):
+        """ edit assignment """
+        assignment = self.get_object()
+
+        if 'client_data' in request.data.keys():
+            client = CustomOperations().custom_update(
+                request=request,
+                view=POSClientViewSet(),
+                data_key='client_data',
+                obj_id=request.data['client_data']['id']
+            )
+            request.data.update({'client': client['id']})
+
+        serializer = self.serializer_class(data=request.data, instance=assignment, partial=True)
+        if serializer.is_valid():
+            assignment = serializer.save()
+
+            return Response(serializer.data, status=status.HTTP_200_OK)
+        return Response(serializer.errors, status=status.HTTP_403_FORBIDDEN)
+
     @action(
         methods=['get'],
         detail=False,