From 6ce2d89c617a81d2762a23265375fe49567836b8 Mon Sep 17 00:00:00 2001
From: Étienne Loks <etienne.loks@iggdrasil.net>
Date: Mon, 28 Jan 2019 19:28:43 +0100
Subject: Migrate main image: fix migration script

---
 ishtar_common/utils_migrations.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'ishtar_common/utils_migrations.py')

diff --git a/ishtar_common/utils_migrations.py b/ishtar_common/utils_migrations.py
index f51465262..7d61f36d8 100644
--- a/ishtar_common/utils_migrations.py
+++ b/ishtar_common/utils_migrations.py
@@ -117,8 +117,8 @@ def migrate_main_image(apps, app_name, model_name, verbose=False):
             sys.stdout.flush()
         q = item.documents.filter(
             image__isnull=False).exclude(image='')
-        if not q.count():
-            return
+        if not q.count():  # no image
+            continue
         # by default get the lowest pk
         item.main_image = q.order_by('pk').all()[0]
         item.skip_history_when_saving = True
-- 
cgit v1.2.3