2 files changed, 33 insertions, 0 deletions
diff --git a/archaeological_files/migrations/0114_main_image.py b/archaeological_files/migrations/0114_main_image.py
new file mode 100644
index 000000000..bd586e139
--- /dev/null
+++ b/archaeological_files/migrations/0114_main_image.py
@@ -0,0 +1,25 @@
+# Generated by Django 2.2.24 on 2024-03-05 12:33
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('ishtar_common', '0239_shootingangle_parent'),
+        ('archaeological_files', '0113_file_imports_updated'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='file',
+            name='main_image',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='main_image_files', to='ishtar_common.Document', verbose_name='Main image'),
+        ),
+        migrations.AddField(
+            model_name='historicalfile',
+            name='main_image',
+            field=models.ForeignKey(blank=True, db_constraint=False, null=True, on_delete=django.db.models.deletion.DO_NOTHING, related_name='+', to='ishtar_common.Document', verbose_name='Main image'),
+        ),
+    ]
diff --git a/archaeological_files/models.py b/archaeological_files/models.py
index 1f2298b32..264773c1b 100644
--- a/archaeological_files/models.py
+++ b/archaeological_files/models.py
@@ -730,6 +730,14 @@ class File(
     documents = models.ManyToManyField(
         Document, related_name="files", verbose_name=_("Documents"), blank=True
     )
+    main_image = models.ForeignKey(
+        Document,
+        related_name="main_image_files",
+        on_delete=models.SET_NULL,
+        verbose_name=_("Main image"),
+        blank=True,
+        null=True,
+    )
 
     imported_line = models.TextField(_("Imported line"), blank=True, default="")
     history = HistoricalRecords(bases=[HistoryModel])