added image-grabbing code to woocommerve migration,images still saved locally

utils/woocommerce.py

@@ -7,8 +7,12 @@ This file holds the functions necesary to:
 
 """
 import logging
+from io import BytesIO
 
 from woocommerce import API
+import requests
+from PIL import Image
+from django.core.files import File
 
 from companies.models import Company
 from products.models import Product
@@ -42,12 +46,9 @@ def migrate_shop_products(url, key, secret, version="wc/v3"):
     company = Company.objects.filter(web_link=url).first()
 
     if not company:
-        # logging.error(f"Could not find Company with URL: {url}")
-        # print(f"Could not find Company with URL: {url}")
-        # return None
-        # TODO: ELIMINATE THIS AFTER DEBUGGING
-        company = Company.objects.create(web_link=url)
-        logging.error(f"Created Company for testing: {url}")
+        logging.error(f"Could not find Company with URL: {url}")
+        print(f"Could not find Company with URL: {url}")
+        return None
 
     # list products
     response = wcapi.get('/products/')
@@ -102,6 +103,24 @@ def migrate_shop_products(url, key, secret, version="wc/v3"):
                 new.save()
             except Exception as e:
                 logging.error(f"Could not create product instance: {str(e)}")
+                continue
+            try:
+                # import ipdb; ipdb.set_trace()
+                # get image
+                headers={
+"User-Agent" : "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36"
+}
+                image_url = product['images'][0]['src']
+                response = requests.get(image_url, stream=True, headers=headers)
+                response.raw.decode_content = True
+                image = Image.open(response.raw)
+                # save using File object
+                img_io = BytesIO()
+                image.save(img_io, format='JPEG')
+                new.image.save(f"{new.name}-{new.sku}.jpg", File(img_io), save=False)
+                new.save()
+            except Exception as e:
+                logging.error(f"Could not add image to product: {str(e)}")
         else:
             logging.error(f"{serializer.errors}")
             continue