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

1 files changed, 20 insertions, 3 deletions

diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py
index e5246d9bb..ff1cd9b5b 100644
--- a/ishtar_common/forms_common.py
+++ b/ishtar_common/forms_common.py
@@ -1156,6 +1156,9 @@ class DocumentForm(forms.ModelForm, CustomForm, ManageOldType):
     }
 
     def __init__(self, *args, **kwargs):
+        main_items_fields = {}
+        if "main_items_fields" in kwargs:
+            main_items_fields = kwargs.pop("main_items_fields")
         super(DocumentForm, self).__init__(*args, **kwargs)
         for related_key in models.Document.RELATED_MODELS_ALT:
             model = models.Document._meta.get_field(related_key).related_model
@@ -1163,6 +1166,14 @@ class DocumentForm(forms.ModelForm, CustomForm, ManageOldType):
                 model=model, remote=True, label=model._meta.verbose_name_plural,
                 required=False, long_widget=True
             )
+            if related_key in main_items_fields:
+                for field_key, label in main_items_fields[related_key]:
+                    disabled = False
+                    if kwargs.get('initial', None) and kwargs['initial'].get(
+                            field_key, False):
+                        disabled = True
+                    self.fields[field_key] = forms.BooleanField(
+                        label=label, required=False, disabled=disabled)
 
     def clean(self):
         cleaned_data = self.cleaned_data
@@ -1191,9 +1202,15 @@ class DocumentForm(forms.ModelForm, CustomForm, ManageOldType):
                     continue
                 related.remove(initial[pk])
             for new_pk in new:
-                if new_pk in initial.keys():
-                    continue
-                related.add(related.model.objects.get(pk=new_pk))
+                related_item = related.model.objects.get(pk=new_pk)
+                if new_pk not in initial.keys():
+                    related.add(related_item)
+                key = "{}_{}_main_image".format(related_key, related_item.pk)
+                if self.cleaned_data.get(key, []) and \
+                        related_item.main_image != item:
+                    related_item.skip_history_when_saving = True
+                    related_item.main_image = item
+                    related_item.save()
         item = models.Document.objects.get(pk=item.pk)
         item.skip_history_when_saving = True
         item.save()  # resave to regen the attached items