diff --git a/vbos/datasets/migrations/0012_tabulardataset_unit.py b/vbos/datasets/migrations/0012_tabulardataset_unit.py new file mode 100644 index 0000000..bae6c47 --- /dev/null +++ b/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), + ), + ] diff --git a/vbos/datasets/models.py b/vbos/datasets/models.py index c143bea..6319c16 100644 --- a/vbos/datasets/models.py +++ b/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 diff --git a/vbos/datasets/serializers.py b/vbos/datasets/serializers.py index 45fc247..6db9219 100644 --- a/vbos/datasets/serializers.py +++ b/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): diff --git a/vbos/datasets/test/test_tabular_views.py b/vbos/datasets/test/test_tabular_views.py index dc7c282..79aad3d 100644 --- a/vbos/datasets/test/test_tabular_views.py +++ b/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"})