changed from if condition to smaller api and added is_arduino flag value

esim-cloud-backend/saveAPI/views.py

@@ -32,36 +32,26 @@ class StateSaveView(APIView):
     @swagger_auto_schema(request_body=StateSaveSerializer)
     def post(self, request, *args, **kwargs):
         logger.info('Got POST for state save ')
-
+        esim_libraries = None
         if request.data.get('esim_libraries'):
             esim_libraries = json.loads(request.data.get('esim_libraries'))
-            img = Base64ImageField(max_length=None, use_url=True)
-            filename, content = img.update(request.data['base64_image'])
-            state_save = StateSave(
-                data_dump=request.data.get('data_dump'),
-                description=request.data.get('description'),
-                name=request.data.get('name'),
-                owner=request.user
-            )
-            state_save.base64_image.save(filename, content)
-            print(state_save)
+        img = Base64ImageField(max_length=None, use_url=True)
+        filename, content = img.update(request.data['base64_image'])
+        state_save = StateSave(
+            data_dump=request.data.get('data_dump'),
+            description=request.data.get('description'),
+            name=request.data.get('name'),
+            owner=request.user,
+            is_arduino=True if esim_libraries == None else False,
+        )
+        state_save.base64_image.save(filename, content)
+        if esim_libraries:
             state_save.esim_libraries.set(esim_libraries)
-            try:
-                state_save.save()
-                return Response(StateSaveSerializer(state_save).data)
-            except Exception:
-                return Response(status=status.HTTP_400_BAD_REQUEST)
-
-        data = request.data
-        if request.data.get('esim_libraries', None):
-            data['esim_libraries'] = ",".join(request.data['esim_libraries'])
-        print(data)
-        serializer = StateSaveSerializer(
-            data=data, context={'request': self.request})
-        if serializer.is_valid():
-            serializer.save(owner=self.request.user)
-            return Response(serializer.data)
-        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
+        try:
+            state_save.save()
+            return Response(StateSaveSerializer(state_save).data)
+        except Exception:
+            return Response(status=status.HTTP_400_BAD_REQUEST)
 
 
 class StateFetchUpdateView(APIView):