feat: poll in django admin, pipeline FK, migrations

backend/apps/api/v1/admin.py

@@ -29,6 +29,7 @@
     ReorderColumnsForm,
     ReorderTablesForm,
     TableInlineForm,
+    PollInlineForm,
     UpdateInlineForm,
 )
 from backend.apps.api.v1.models import (
@@ -53,6 +54,7 @@
     ObservationLevel,
     Organization,
     Pipeline,
+    Poll,
     QualityCheck,
     RawDataSource,
     Status,
@@ -229,6 +231,19 @@ def datetime_ranges(self, cov):
         """Show datetime ranges in coverage inline"""
         return [str(dt) for dt in cov.datetime_ranges.all()]
 
+class PollInline(admin.StackedInline):
+    model = Poll
+    form = PollInlineForm
+    extra = 0
+    fields = [
+        "id",
+        "entity",
+        "frequency",
+        "latest",
+    ]
+    autocomplete_fields = [
+        "entity",
+    ]
 
 class UpdateInline(admin.StackedInline):
     model = Update
@@ -699,15 +714,15 @@ class RawDataSourceAdmin(TabbedTranslationAdmin):
         "languages",
         "area_ip_address_required",
     ]
-    inlines = [CoverageInline]
+    inlines = [CoverageInline, PollInline]
 
 
 class InformationRequestAdmin(TabbedTranslationAdmin):
     list_display = ["__str__", "dataset", "created_at", "updated_at"]
     search_fields = ["__str__", "dataset__name"]
     readonly_fields = ["id", "created_at", "updated_at"]
     autocomplete_fields = ["dataset"]
-    inlines = [CoverageInline, ObservationLevelInline]
+    inlines = [CoverageInline, ObservationLevelInline, PollInline]
 
 
 class CoverageTypeAdminFilter(admin.SimpleListFilter):
@@ -830,6 +845,23 @@ class LanguageAdmin(TabbedTranslationAdmin):
         "slug",
     ]
 
+class PollAdmin(admin.ModelAdmin):
+    readonly_fields = [
+        "id",
+    ]
+    list_display = [
+        "__str__",
+    ]
+    search_fields = [
+        "entity",
+        "raw_data_source",
+        "information_request",
+    ]
+    autocomplete_fields = [
+        "entity",
+        "raw_data_source",
+        "information_request",
+    ]
 
 class UpdateAdmin(admin.ModelAdmin):
     readonly_fields = [
@@ -1014,6 +1046,7 @@ class QualityCheckAdmin(TabbedTranslationAdmin):
 admin.site.register(ObservationLevel, ObservationLevelAdmin)
 admin.site.register(Organization, OrganizationAdmin)
 admin.site.register(Pipeline)
+admin.site.register(Poll, PollAdmin)
 admin.site.register(RawDataSource, RawDataSourceAdmin)
 admin.site.register(Status, StatusAdmin)
 admin.site.register(Table, TableAdmin)

backend/apps/api/v1/forms/__init__.py

@@ -6,6 +6,7 @@
     CoverageInlineForm,
     ObservationLevelInlineForm,
     TableInlineForm,
+    PollInlineForm,
     UpdateInlineForm,
 )
 from backend.apps.api.v1.forms.reorder_columns_form import ReorderColumnsForm  # noqa: F401

backend/apps/api/v1/forms/admin_form.py

@@ -8,6 +8,7 @@
     Coverage,
     ObservationLevel,
     Table,
+    Poll,
     Update,
 )
 
@@ -106,6 +107,11 @@ class Meta(UUIDHiddenIdForm.Meta):
         ]
 
 
+class PollInlineForm(UUIDHiddenIdForm):
+    class Meta(UUIDHiddenIdForm.Meta):
+        model = Poll
+        fields = "__all__"
+
 class UpdateInlineForm(UUIDHiddenIdForm):
     class Meta(UUIDHiddenIdForm.Meta):
         model = Update

backend/apps/api/v1/migrations/0037_poll_pipeline.py

@@ -0,0 +1,19 @@
+# Generated by Django 4.2.13 on 2024-08-21 05:40
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('v1', '0036_datetimerange_unit'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='poll',
+            name='pipeline',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.PROTECT, related_name='polls', to='v1.pipeline'),
+        ),
+    ]

backend/apps/api/v1/models.py

@@ -742,6 +742,13 @@ class Poll(BaseModel):
     )
     frequency = models.IntegerField(blank=True, null=True)
     latest = models.DateTimeField(blank=True, null=True)
+    pipeline = models.ForeignKey(
+        "Pipeline",
+        blank=True,
+        null=True,
+        on_delete=models.PROTECT,
+        related_name="polls"
+    )
     raw_data_source = models.ForeignKey(
         "RawDataSource",
         blank=True,