Support uploading model runs from the web client

pyproject.toml

@@ -60,6 +60,7 @@ apache-airflow-client = "^2.9.0"
 beautifulsoup4 = "^4.12.3"
 django-allauth = {extras = ["socialaccount"], version = "^0.63.2"}
 django-login-required-middleware = "^0.9.0"
+django-s3-file-field = {extras = ["minio"], version = "^1.0.1"}
 
 [tool.poetry.group.dev.dependencies]
 django-stubs = "^4.2.7"

rdwatch/core/admin.py

@@ -2,6 +2,7 @@
 
 from rdwatch.core.models import (
     ModelRun,
+    ModelRunUpload,
     Performer,
     Region,
     SatelliteFetching,
@@ -123,3 +124,15 @@ class SiteObservationAdmin(admin.ModelAdmin):
     )
     list_filter = ('timestamp',)
     raw_id_fields = ('siteeval', 'label', 'constellation', 'spectrum')
+
+
+@admin.register(ModelRunUpload)
+class ModelRunUploadAdmin(admin.ModelAdmin):
+    list_display = (
+        'id',
+        'title',
+        'performer',
+        'region',
+        'zipfile',
+        'task_id',
+    )

rdwatch/core/migrations/0033_modelrunupload.py

@@ -0,0 +1,59 @@
+# Generated by Django 5.0.8 on 2024-08-14 21:17
+
+import uuid
+
+import s3_file_field.fields
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ('core', '0032_siteevaluation_point_siteobservation_point_and_more'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='ModelRunUpload',
+            fields=[
+                (
+                    'id',
+                    models.UUIDField(
+                        default=uuid.uuid4,
+                        editable=False,
+                        primary_key=True,
+                        serialize=False,
+                    ),
+                ),
+                ('title', models.CharField(max_length=1000)),
+                (
+                    'private',
+                    models.BooleanField(
+                        default=False,
+                        help_text='Whether this model run should be private',
+                    ),
+                ),
+                (
+                    'region',
+                    models.CharField(
+                        blank=True,
+                        help_text='Override for the region this model run belongs to',
+                        max_length=1000,
+                    ),
+                ),
+                (
+                    'performer',
+                    models.CharField(
+                        blank=True,
+                        help_text='Override for the team that produced this evaluation',
+                        max_length=1000,
+                    ),
+                ),
+                ('zipfile', s3_file_field.fields.S3FileField()),
+                (
+                    'task_id',
+                    models.CharField(help_text='celery task ID', max_length=128),
+                ),
+            ],
+        ),
+    ]

rdwatch/core/models/__init__.py

@@ -1,6 +1,7 @@
 from . import lookups
 from .annotation_exports import AnnotationExport
 from .model_run import ModelRun
+from .model_run_upload import ModelRunUpload
 from .performer import Performer
 from .region import Region
 from .satellite_fetching import SatelliteFetching
@@ -12,6 +13,7 @@
     'AnnotationExport',
     'lookups',
     'ModelRun',
+    'ModelRunUpload',
     'Performer',
     'Region',
     'SiteEvaluation',

rdwatch/core/models/model_run_upload.py

@@ -0,0 +1,38 @@
+from uuid import uuid4
+
+from s3_file_field import S3FileField
+
+from django.db import models
+from django.db.models.signals import pre_delete
+from django.dispatch import receiver
+
+
+class ModelRunUpload(models.Model):
+    id = models.UUIDField(primary_key=True, default=uuid4, editable=False)
+
+    title = models.CharField(max_length=1000)
+    private = models.BooleanField(
+        default=False, help_text='Whether this model run should be private'
+    )
+    region = models.CharField(
+        max_length=1000,
+        blank=True,
+        help_text='Override for the region this model run belongs to',
+    )
+    performer = models.CharField(
+        max_length=1000,
+        blank=True,
+        help_text='Shortcode override for the team that produced this evaluation',
+    )
+    zipfile = S3FileField()
+
+    task_id = models.CharField(max_length=256, help_text='Celery task ID')
+
+    def __str__(self) -> str:
+        return f'<ModelRunUpload {self.id}>'
+
+
+@receiver(pre_delete, sender=ModelRunUpload)
+def delete_zipfile(sender, instance, **kwargs):
+    if instance.zipfile:
+        instance.zipfile.delete(save=False)