products: Add route to delete product images.

Author: prah23

products/serializers.py

@@ -76,7 +76,7 @@ class Meta:
         fields = ['product', 'image', 'mini_description']
 
 
-class ProductImageCreateSerializer(serializers.ModelSerializer):
+class ProductImageCRUDSerializer(serializers.ModelSerializer):
     class Meta:
         model = ProductImage
         fields = '__all__'

products/urls.py

@@ -4,6 +4,7 @@
     SubCategoryCreateView,
     ProductCreateView,
     ProductImageCreateView,
+    ProductImageDeleteView,
     CategoryUpdateDeleteView,
     SubCategoryUpdateDeleteView,
     PopularProductListView,
@@ -23,6 +24,7 @@
 urlpatterns = [
     path('create/product/', ProductCreateView.as_view()),
     path('create/product/add_image/', ProductImageCreateView.as_view()),
+    path('delete/product/delete_image/<pi_id>/', ProductImageDeleteView.as_view()),
     path('create/category/', CategoryCreateView.as_view()),
     path('create/subcategory/', SubCategoryCreateView.as_view()),
     path('modify/category/<category>/', CategoryUpdateDeleteView.as_view()),

products/views.py

@@ -11,7 +11,7 @@
     ProductSerializer,
     ParticularProductSerializer,
     ProductImageSerializer,
-    ProductImageCreateSerializer,
+    ProductImageCRUDSerializer,
     ReviewRatingSerializer,
     ReviewRatingCreateSerializer
 )
@@ -54,10 +54,31 @@ def create(self, request, *args, **kwargs):
 class ProductImageCreateView(CreateAPIView):
     queryset = ProductImage.objects.all()
     permission_classes = [IsAuthenticated, IsArtist]
-    serializer_class = ProductImageCreateSerializer
+    serializer_class = ProductImageCRUDSerializer
     parser_classes = [FormParser, MultiPartParser]
 
 
+class ProductImageDeleteView(APIView):
+    permission_classes = [IsAuthenticated, IsArtist]
+    serializer_class = ProductImageCRUDSerializer
+    parser_classes = [JSONParser]
+
+    def delete(self, request, pi_id):
+        user = request.user.id
+        try:
+            obj = ProductImage.objects.get(id=pi_id, product__artist=user)
+            obj.delete()
+            return Response({
+                'status': 'success',
+                'details': 'Successfully deleted product image.'
+            }, status=200)
+        except:
+            return Response({
+                'status': 'error',
+                'details': 'Error in deleting product image.'
+            }, status=400)
+
+
 class CategoryUpdateDeleteView(RetrieveUpdateDestroyAPIView):
     serializer_class = CategorySerializer
     parser_classes = [JSONParser]