Image: form to change main image - fix image in sheets

1 files changed, 12 insertions, 1 deletions

diff --git a/ishtar_common/views.py b/ishtar_common/views.py
index 72418e4fa..710feb8c6 100644
--- a/ishtar_common/views.py
+++ b/ishtar_common/views.py
@@ -1748,7 +1748,18 @@ class DocumentEditView(DocumentFormMixin, UpdateView):
             if hasattr(value, 'pk'):
                 value = value.pk
             initial[k] = value
-
+        # main image initialisation
+        if document.image:
+            kwargs["main_items_fields"] = {}
+            for k in models.Document.RELATED_MODELS:
+                kwargs["main_items_fields"][k] = []
+                for related_item in getattr(document, k).all():
+                    key = "{}_{}_main_image".format(k, related_item.pk)
+                    kwargs["main_items_fields"][k].append(
+                        (key, u"{} - {}".format(
+                            _(u"Main image for"), related_item)))
+                    if related_item.main_image == document:
+                        initial[key] = True
         kwargs['initial'] = initial
         return kwargs