2 files changed, 46 insertions, 1 deletions
diff --git a/archaeological_context_records/migrations/0020_auto_20180423_1838.py b/archaeological_context_records/migrations/0020_auto_20180423_1838.py
new file mode 100644
index 000000000..fb0c7463f
--- /dev/null
+++ b/archaeological_context_records/migrations/0020_auto_20180423_1838.py
@@ -0,0 +1,39 @@
+# -*- coding: utf-8 -*-
+# Generated by Django 1.11.10 on 2018-04-23 18:38
+from __future__ import unicode_literals
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ishtar_common', '0048_auto_20180423_1838'),
+        ('archaeological_context_records', '0019_auto_20180403_1120'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ContextRecordImage',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('is_main', models.BooleanField(default=False, verbose_name='Main image')),
+            ],
+        ),
+        migrations.AlterField(
+            model_name='contextrecord',
+            name='images',
+            field=models.ManyToManyField(blank=True, through='archaeological_context_records.ContextRecordImage', to='ishtar_common.IshtarImage', verbose_name='Images'),
+        ),
+        migrations.AddField(
+            model_name='contextrecordimage',
+            name='context_record',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='archaeological_context_records.ContextRecord'),
+        ),
+        migrations.AddField(
+            model_name='contextrecordimage',
+            name='image',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='ishtar_common.IshtarImage'),
+        ),
+    ]
diff --git a/archaeological_context_records/models.py b/archaeological_context_records/models.py
index f6253bf68..bdb21ea39 100644
--- a/archaeological_context_records/models.py
+++ b/archaeological_context_records/models.py
@@ -325,7 +325,7 @@ class ContextRecord(BulkUpdatedItem, BaseHistorizedItem,
     point = models.PointField(_(u"Point (3D)"), blank=True, null=True, dim=3)
     polygon = models.PolygonField(_(u"Polygon"), blank=True, null=True)
     images = models.ManyToManyField(IshtarImage, verbose_name=_(u"Images"),
-                                    blank=True)
+                                    blank=True, through='ContextRecordImage')
     cached_label = models.TextField(_(u"Cached name"), null=True, blank=True,
                                     db_index=True)
     PARENT_SEARCH_VECTORS = ['operation']
@@ -565,6 +565,12 @@ class ContextRecord(BulkUpdatedItem, BaseHistorizedItem,
 post_save.connect(cached_label_changed, sender=ContextRecord)
 
 
+class ContextRecordImage(models.Model):
+    image = models.ForeignKey(IshtarImage, on_delete=models.CASCADE)
+    context_record = models.ForeignKey(ContextRecord, on_delete=models.CASCADE)
+    is_main = models.BooleanField(_(u"Main image"), default=False)
+
+
 class RelationType(GeneralRelationType):
     class Meta:
         verbose_name = _(u"Relation type")