Manage main image: model, auto associate default image

1 files changed, 7 insertions, 2 deletions

diff --git a/archaeological_context_records/models.py b/archaeological_context_records/models.py
index d7d7a618d..82527acb1 100644
--- a/archaeological_context_records/models.py
+++ b/archaeological_context_records/models.py
@@ -24,7 +24,7 @@ from django.contrib.gis.db import models
 from django.core.urlresolvers import reverse
 from django.db import connection, transaction
 from django.db.models import Q
-from django.db.models.signals import post_delete, post_save
+from django.db.models.signals import post_delete, post_save, m2m_changed
 from django.utils.translation import ugettext_lazy as _, pgettext, \
     activate, pgettext_lazy, deactivate
 from django.utils.text import slugify
@@ -35,7 +35,7 @@ from ishtar_common.models import Document, GeneralType, \
     BaseHistorizedItem, HistoricalRecords, OwnPerms, ShortMenuItem, \
     GeneralRelationType, GeneralRecordRelations, post_delete_record_relation,\
     post_save_cache, ValueGetter, BulkUpdatedItem, ExternalIdManager, \
-    RelationItem, Town, get_current_profile
+    RelationItem, Town, get_current_profile, document_attached_changed
 from archaeological_operations.models import Operation, Period, Parcel, \
     ArchaeologicalSite
 
@@ -429,6 +429,9 @@ class ContextRecord(BulkUpdatedItem, BaseHistorizedItem,
     documents = models.ManyToManyField(
         Document, related_name='context_records', verbose_name=_(u"Documents"),
         blank=True)
+    main_image = models.ForeignKey(
+        Document, related_name='main_image_context_records',
+        verbose_name=_(u"Main image"), blank=True, null=True)
     cached_label = models.TextField(_(u"Cached name"), null=True, blank=True,
                                     db_index=True)
 
@@ -681,6 +684,8 @@ class ContextRecord(BulkUpdatedItem, BaseHistorizedItem,
 
 
 post_save.connect(cached_label_changed, sender=ContextRecord)
+m2m_changed.connect(document_attached_changed,
+                    sender=ContextRecord.documents.through)
 
 
 class RelationType(GeneralRelationType):