Add unit column to TabularDataset table (#20)

* Add unit column to TabularDataset table

* Lint code

Author: willemarcel

vbos/datasets/migrations/0012_tabulardataset_unit.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.2.5 on 2025-10-06 18:53
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("datasets", "0011_rename_data_tabularitem_metadata_and_more"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="tabulardataset",
+            name="unit",
+            field=models.CharField(blank=True, max_length=50, null=True),
+        ),
+    ]

vbos/datasets/models.py

@@ -147,6 +147,7 @@ class TabularDataset(models.Model):
     )
     type = models.CharField(max_length=55, choices=TYPE_CHOICES, default="baseline")
     source = models.CharField(max_length=155, blank=True, null=True)
+    unit = models.CharField(max_length=50, blank=True, null=True)
 
     def __str__(self):
         return self.name

vbos/datasets/serializers.py

@@ -66,7 +66,16 @@ class TabularDatasetSerializer(serializers.ModelSerializer):
 
     class Meta:
         model = TabularDataset
-        fields = ["id", "name", "created", "updated", "cluster", "type", "source"]
+        fields = [
+            "id",
+            "name",
+            "created",
+            "updated",
+            "cluster",
+            "type",
+            "source",
+            "unit",
+        ]
 
 
 class TabularItemSerializer(serializers.ModelSerializer):

vbos/datasets/test/test_tabular_views.py

@@ -20,6 +20,7 @@ def setUp(self):
             cluster=Cluster.objects.create(name="Statistics"),
             source="Government",
             type="estimated_damage",
+            unit="Vatu (VUV)",
         )
         self.url = reverse("datasets:tabular-list")
 
@@ -35,6 +36,7 @@ def test_tabular_datasets_list(self):
         assert req.data.get("results")[1]["cluster"] == "Statistics"
         assert req.data.get("results")[0]["type"] == "baseline"
         assert req.data.get("results")[1]["type"] == "estimated_damage"
+        assert req.data.get("results")[1]["unit"] == "Vatu (VUV)"
 
     def test_raster_datasets_list_filter(self):
         req = self.client.get(self.url, {"cluster": "transportation"})