Manage main image: model, auto associate default image

1 files changed, 13 insertions, 1 deletions

diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py
index d8233ba0f..0b506e12b 100644
--- a/archaeological_operations/models.py
+++ b/archaeological_operations/models.py
@@ -36,7 +36,8 @@ from ishtar_common.models import BaseHistorizedItem, Dashboard, \
     HistoricalRecords, IshtarUser, LightHistorizedItem, \
     OperationType, Organization, OwnPerms, Person, PersonType, \
     post_delete_record_relation, post_save_cache, RelationItem, \
-    ShortMenuItem, SourceType, Town, ValueGetter, get_current_profile
+    ShortMenuItem, SourceType, Town, ValueGetter, get_current_profile, \
+    document_attached_changed
 from ishtar_common.utils import cached_label_changed, \
     force_cached_label_changed, mode
 
@@ -267,6 +268,9 @@ class ArchaeologicalSite(BaseHistorizedItem, OwnPerms, ValueGetter,
     documents = models.ManyToManyField(
         Document, related_name="sites", verbose_name=_(u"Documents"),
         blank=True)
+    main_image = models.ForeignKey(
+        Document, related_name='main_image_sites',
+        verbose_name=_(u"Main image"), blank=True, null=True)
     cached_label = models.TextField(_(u"Cached name"),
                                     null=True, blank=True, db_index=True)
 
@@ -420,6 +424,9 @@ class ArchaeologicalSite(BaseHistorizedItem, OwnPerms, ValueGetter,
 
 post_save.connect(cached_label_changed, sender=ArchaeologicalSite)
 
+m2m_changed.connect(document_attached_changed,
+                    sender=ArchaeologicalSite.documents.through)
+
 
 def get_values_town_related(item, prefix, values):
     values[prefix + 'parcellist'] = item.render_parcels()
@@ -845,6 +852,9 @@ class Operation(ClosedItem, BaseHistorizedItem, OwnPerms, ValueGetter,
     documents = models.ManyToManyField(
         Document, related_name='operations', verbose_name=_(u"Documents"),
         blank=True)
+    main_image = models.ForeignKey(
+        Document, related_name='main_image_operations',
+        verbose_name=_(u"Main image"), blank=True, null=True)
     cached_label = models.CharField(_(u"Cached name"), max_length=500,
                                     null=True, blank=True, db_index=True)
     archaeological_sites = models.ManyToManyField(
@@ -1386,6 +1396,8 @@ class Operation(ClosedItem, BaseHistorizedItem, OwnPerms, ValueGetter,
 
 
 m2m_changed.connect(force_cached_label_changed, sender=Operation.towns.through)
+m2m_changed.connect(document_attached_changed,
+                    sender=Operation.documents.through)
 
 
 def operation_post_save(sender, **kwargs):