From ae4585ab536d6cfad4ebd11be3e23e43c1f0f106 Mon Sep 17 00:00:00 2001 From: Thomas Fillon Date: Mon, 7 Sep 2015 15:32:57 +0200 Subject: [PATCH] Fix populate_db management command to properly add image field to product --- .../records/management/commands/populate_db.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/diggersdigest/records/management/commands/populate_db.py b/diggersdigest/records/management/commands/populate_db.py index 51e71b2..613b619 100644 --- 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() -- 2.39.5