add - upload product image

apps/product/web/api/v1/viewsets/product_api.py

@@ -13,6 +13,7 @@ from apps.core.api import BaseViewSet
 from apps.core.mixins.search_mixin import DynamicSearchMixin
 from apps.core.mixins.soft_delete_mixin import SoftDeleteMixin
 from apps.product import models as product_models
+from apps.product.services.upload_product_image import UploadProductImage
 from apps.product.web.api.v1.serializers import product_serializers as product_serializers
 from apps.product.web.api.v1.serializers import quota_serializers
 from common.helpers import get_organization_by_user
@@ -69,7 +70,7 @@ class ProductCategoryViewSet(SoftDeleteMixin, viewsets.ModelViewSet, DynamicSear
             return Response(e, status=status.HTTP_204_NO_CONTENT)
 
 
-class ProductViewSet(SoftDeleteMixin, viewsets.ModelViewSet, DynamicSearchMixin):
+class ProductViewSet(SoftDeleteMixin, viewsets.ModelViewSet, DynamicSearchMixin, UploadProductImage):
     queryset = product_models.Product.objects.all()
     serializer_class = product_serializers.ProductSerializer
     filter_backends = [filters.SearchFilter]
@@ -88,6 +89,41 @@ class ProductViewSet(SoftDeleteMixin, viewsets.ModelViewSet, DynamicSearchMixin)
         serializer = self.get_serializer(queryset, many=True)
         return Response(serializer.data)
 
+    @transaction.atomic
+    def create(self, request, *args, **kwargs):
+        """
+        create product & upload image
+        """
+
+        serializer = self.serializer_class(data=request.data, context={"request": request})
+        if serializer.is_valid(raise_exception=True):
+            product = serializer.save()
+
+            # upload product image to liara
+            if 'image' in request.data.keys():
+                self.upload_image(request=request, product_id=product.id)
+
+            return Response(serializer.data, status=status.HTTP_201_CREATED)
+        return Response(serializer.errors, status=status.HTTP_403_FORBIDDEN)
+
+    def update(self, request, pk=None, *args, **kwargs):
+        """
+        update product & upload image
+        """
+
+        product = self.get_object()
+
+        serializer = self.serializer_class(product, data=request.data, context={"request": request})
+        if serializer.is_valid(raise_exception=True):
+            product = serializer.save()
+
+            # upload product image to liara
+            if 'image' in request.data.keys():
+                self.upload_image(request=request, product_id=product.id)
+
+            return Response(serializer.data, status=status.HTTP_201_CREATED)
+        return Response(serializer.errors, status=status.HTTP_403_FORBIDDEN)
+
     @action(
         methods=['get'],
         detail=True,