🔴 admin.py

🟢 0007_alter_signaltype_options_geographyunit.py
🛠️ models.py -> Added GeographyUnit model

Author: BrainIsDead

src/signals/admin.py

@@ -5,6 +5,7 @@
 
 from signals.models import (
     Geography,
+    GeographyUnit,
     Pathogen,
     Signal,
     SignalCategory,
@@ -22,6 +23,15 @@ class SignalCategoryAdmin(admin.ModelAdmin):
     search_fields: tuple[Literal['name']] = ('name',)
 
 
+@admin.register(GeographyUnit)
+class GeographyUnitAdmin(admin.ModelAdmin):
+    """
+    Admin interface for managing geography unit objects.
+    """
+    list_display: tuple[Literal['name']] = ('name', 'geo_id', 'geography',)
+    search_fields: tuple[Literal['name']] = ('name', 'display_name', 'geo_id',)
+
+
 @admin.register(Geography)
 class GeographyAdmin(admin.ModelAdmin):
     """

src/signals/migrations/0007_alter_signaltype_options_geographyunit.py

@@ -0,0 +1,35 @@
+# Generated by Django 5.0.3 on 2024-05-02 12:12
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('signals', '0006_merge_20240219_1508'),
+    ]
+
+    operations = [
+        migrations.AlterModelOptions(
+            name='signaltype',
+            options={},
+        ),
+        migrations.CreateModel(
+            name='GeographyUnit',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('created', models.DateTimeField(auto_now_add=True)),
+                ('modified', models.DateTimeField(auto_now=True)),
+                ('geo_id', models.CharField(help_text='Geo ID', max_length=128)),
+                ('name', models.CharField(help_text='Name', max_length=128)),
+                ('display_name', models.CharField(help_text='Display Name', max_length=128)),
+                ('postal', models.CharField(help_text='Postal', max_length=128, null=True)),
+                ('level', models.IntegerField(help_text='Level')),
+                ('geography', models.ForeignKey(help_text='Geography', on_delete=django.db.models.deletion.CASCADE, related_name='geography_units', to='signals.geography')),
+            ],
+            options={
+                'abstract': False,
+            },
+        ),
+    ]

src/signals/models.py

@@ -137,6 +137,47 @@ def __str__(self) -> str:
         return str(self.name)
 
 
+class GeographyUnit(TimeStampedModel):
+    """
+    A model representing a geography (geo-level) unit.
+    """
+
+    geo_id: models.CharField = models.CharField(
+        help_text=_('Geo ID'),
+        max_length=128
+    )
+    name: models.CharField = models.CharField(
+        help_text=_('Name'),
+        max_length=128
+    )
+    display_name: models.CharField = models.CharField(
+        help_text=_('Display Name'),
+        max_length=128
+    )
+    postal: models.CharField = models.CharField(
+        help_text=_('Postal'),
+        max_length=128,
+        null=True,
+    )
+    level: models.IntegerField = models.IntegerField(help_text=_('Level'))
+
+    geography: models.ForeignKey = models.ForeignKey(
+        'signals.Geography',
+        related_name='geography_units',
+        help_text=_('Geography'),
+        on_delete=models.CASCADE
+    )
+
+    def __str__(self) -> str:
+        """
+        Returns the name of the geography unit as a string.
+
+        :return: The name of the geography unit as a string.
+        :rtype: str
+        """
+        return str(self.name)
+
+
 class Signal(TimeStampedModel):
     """
     A model representing a signal.