edit assignment

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,