Added extra small image in api

Author: MySecondLanguage

nxtbn/product/admin_types.py

@@ -12,12 +12,16 @@ class ProductVariantNonPaginatedType(DjangoObjectType):
     display_name = graphene.String()
     humanize_price = graphene.String()
     variant_thumbnail = graphene.String()
+    variant_thumbnail_xs = graphene.String()
 
     def resolve_humanize_price(self, info):
         return self.humanize_total_price()
 
     def resolve_variant_thumbnail(self, info):
         return self.variant_thumbnail(info.context)
+    
+    def resolve_variant_thumbnail_xs(self, info):
+        return self.variant_thumbnail_xs(info.context)
     class Meta:
         model = ProductVariant
         fields = (
@@ -46,12 +50,16 @@ class ProductGraphType(DjangoObjectType):
     db_id = graphene.Int(source="id")
     all_variants = graphene.List(ProductVariantNonPaginatedType)
     product_thumbnail = graphene.String()
+    product_thumbnail_xs = graphene.String()
 
     def resolve_all_variants(self, info):
         return self.variants.all()
 
     def resolve_product_thumbnail(self, info):
         return self.product_thumbnail(info.context)
+    
+    def resolve_product_thumbnail_xs(self, info):
+        return self.product_thumbnail_xs(info.context)
     class Meta:
         model = Product
         fields = (

nxtbn/product/models.py

@@ -221,7 +221,18 @@ def product_thumbnail(self, request):
             full_url = request.build_absolute_uri(image_url)
             return full_url
         return None
-
+    
+    def product_thumbnail_xs(self, request):
+        """
+        Returns the URL of the first image associated with the product. 
+        If no image is available, returns None.
+        """
+        first_image = self.images.first()  # Get the first image if it exists
+        if first_image and hasattr(first_image, 'image_xs') and first_image.image_xs:
+            image_url = first_image.image_xs.url
+            full_url = request.build_absolute_uri(image_url)
+            return full_url
+        return None
 
 
 
@@ -385,6 +396,24 @@ def variant_thumbnail(self, request):
                 full_url = request.build_absolute_uri(image_url)
                 return
         return None
+    
+    def variant_thumbnail_xs(self, request):
+        """
+        Returns the URL of the first image associated with the product variant. 
+        If no image is available, returns None.
+        """
+        if self.image and hasattr(self.image, 'image_xs') and self.image.image_xs:
+            image_url = self.image.image_xs.url
+            full_url = request.build_absolute_uri(image_url)
+            return full_url
+        
+        if self.product.images.exists():
+            first_image = self.product.images.first()
+            if first_image and hasattr(first_image, 'image_xs') and first_image.image_xs:
+                image_url = first_image.image_xs.url
+                full_url = request.build_absolute_uri(image_url)
+                return
+        return None
 
 
     def get_valid_stock(self): # stocks that available for sell