small code fix

hivesolutions · Jun 16, 2016 · 0d937c1 · 0d937c1
1 parent 23e5ad4
commit 0d937c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/budy/models/product.py b/src/budy/models/product.py
@@ -447,8 +447,8 @@ def build_images(self):
         thumbnail = thumbnail or self.get_image(size = "thumbnail")
         image = self.get_image(size = "large", order = 1)
         image = image or self.get_image(size = "large")
-        self.thumbnail_url = thumbnail.get_url()
-        self.image_url = image.get_url()
+        self.thumbnail_url = thumbnail.get_url() if thumbnail else None
+        self.image_url = image.get_url() if image else None
 
     def related(self, limit = 6, available = True):
         cls = self.__class__