Fix populate_db management command to properly add image field to product

diff --git a/diggersdigest/records/management/commands/populate_db.py b/diggersdigest/records/management/commands/populate_db.py
index 51e71b2ae305e2cc0701718a96badaf72e025358..613b619ea5806c4002d280f2e7dab550d2347139 100644 (file)
--- a/diggersdigest/records/management/commands/populate_db.py
+++ b/diggersdigest/records/management/commands/populate_db.py
@@ -282,7 +282,7 @@ class Command(BaseCommand):
                 category = Category.objects.get(pk=shop.theme)
                 prod_obj.categories.add(category)
 
-                prodvar = ProductVariation.objects.get_or_create(product=prod_obj,
+                prodvar, created_var = ProductVariation.objects.get_or_create(product=prod_obj,
                                                                  default=True)
 
                 prod_obj.save()
@@ -295,7 +295,13 @@ class Command(BaseCommand):
                     image, created = ProductImage.objects.get_or_create(
                         file=img_file,
                         product=prod_obj)
+                    prodvar.image = image
+                    prodvar.save()
 
+                prod_obj.copy_default_variation()
+                prod_obj.save()
+
+                print 'Product Image %s' % prod_obj.image
                 self.stdout.write('Product %d / %d --> %s' % (prod_obj.pk,len(shop_list), prod_obj.available))
 
             #self.stdout.write('\b\b\b\b%.4d' % obj.pk)
@@ -350,7 +356,7 @@ class Command(BaseCommand):
         #self.populate_pages()
         #self.populate_news()
 
-        #self.populate_record()
+        self.populate_record()