Add BaselineWealthCharacteristicsValueViewSet and related serializer see HEA-772

Author: girumb

apps/baseline/serializers.py

@@ -412,6 +412,52 @@ def get_wealth_group_label(self, obj):
     )
 
 
+class BaselineWealthCharacteristicsValueSerializer(serializers.ModelSerializer):
+
+    class Meta:
+        model = WealthGroupCharacteristicValue
+        fields = [
+            "id",
+            "wealth_group",
+            "wealth_group_category",
+            "wealth_group_category_name",
+            "livelihood_zone_baseline",
+            "livelihood_zone_baseline_label",
+            "wealth_characteristic",
+            "wealth_characteristic_name",
+            "characteristic_group",
+            "product",
+            "product_name",
+            "unit_of_measure",
+            "unit_of_measure_name",
+            "value",
+            "min_value",
+            "max_value",
+        ]
+
+    wealth_group_category = serializers.CharField(source="wealth_group.wealth_group_category.pk", read_only=True)
+    wealth_group_category_name = serializers.CharField(
+        source="wealth_group.wealth_group_category.name", read_only=True
+    )
+    livelihood_zone_baseline = serializers.IntegerField(
+        source="wealth_group.livelihood_zone_baseline.pk", read_only=True, allow_null=True
+    )
+    livelihood_zone_baseline_label = serializers.SerializerMethodField()
+
+    def get_livelihood_zone_baseline_label(self, obj):
+        """Get label for livelihood zone baseline."""
+        if hasattr(obj.wealth_group, "livelihood_zone_baseline") and obj.wealth_group.livelihood_zone_baseline:
+            return str(obj.wealth_group.livelihood_zone_baseline)
+        return None
+
+    wealth_characteristic_name = serializers.CharField(source="wealth_characteristic.name", read_only=True)
+    characteristic_group = serializers.CharField(
+        source="wealth_characteristic.characteristic_group", read_only=True, allow_null=True
+    )
+    product_name = serializers.CharField(source="product.name", read_only=True, allow_null=True)
+    unit_of_measure_name = serializers.CharField(source="unit_of_measure.name", read_only=True, allow_null=True)
+
+
 class LivelihoodStrategySerializer(serializers.ModelSerializer):
     class Meta:
         model = LivelihoodStrategy

apps/baseline/viewsets.py

@@ -54,6 +54,7 @@
 )
 from .serializers import (
     BaselineLivelihoodActivitySerializer,
+    BaselineWealthCharacteristicsValueSerializer,
     BaselineWealthGroupSerializer,
     ButterProductionSerializer,
     CommunityCropProductionSerializer,
@@ -512,6 +513,70 @@ class WealthGroupCharacteristicValueViewSet(BaseModelViewSet):
     filterset_class = WealthGroupCharacteristicValueFilterSet
 
 
+class BaselineWealthCharacteristicsFilterSet(filters.FilterSet):
+
+    class Meta:
+        model = WealthGroupCharacteristicValue
+        fields = ["wealth_characteristic", "wealth_group"]
+
+    livelihood_zone_baseline = filters.NumberFilter(
+        field_name="wealth_group__livelihood_zone_baseline",
+        label="Livelihood Zone Baseline ID",
+    )
+
+    country = MultiFieldFilter(
+        [
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso3166a2",
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso_en_ro_name",
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso_en_name",
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso_en_ro_proper",
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso_en_proper",
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso_fr_name",
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso_fr_proper",
+            "wealth_group__livelihood_zone_baseline__livelihood_zone__country__iso_es_name",
+        ],
+        lookup_expr="iexact",
+        label="Country",
+    )
+
+    has_value = filters.BooleanFilter(
+        method="filter_has_value",
+        label="Has Value",
+    )
+
+    def filter_has_value(self, queryset, name, value):
+        if value:
+            return queryset.exclude(value__isnull=True).exclude(value__exact="")
+        else:
+            return queryset.filter(Q(value__isnull=True) | Q(value__exact=""))
+
+
+class BaselineWealthCharacteristicsValueViewSet(BaseModelViewSet):
+    """
+    API endpoint for baseline wealth characteristics with grouping.
+    """
+
+    queryset = (
+        WealthGroupCharacteristicValue.objects.select_related(
+            "wealth_characteristic",
+            "wealth_group__wealth_group_category",
+            "wealth_group__livelihood_zone_baseline",
+            "product",
+            "unit_of_measure",
+        )
+        .filter(wealth_group__community__isnull=True)
+        .order_by(
+            "wealth_characteristic__characteristic_group",
+            "wealth_characteristic__ordering",
+            "wealth_characteristic__code",
+            "wealth_group__wealth_group_category__ordering",
+        )
+    )
+
+    serializer_class = BaselineWealthCharacteristicsValueSerializer
+    filterset_class = BaselineWealthCharacteristicsFilterSet
+
+
 class LivelihoodStrategyFilterSet(filters.FilterSet):
     class Meta:
         model = LivelihoodStrategy

hea/urls.py

@@ -11,6 +11,7 @@
 
 from baseline.viewsets import (
     BaselineLivelihoodActivityViewSet,
+    BaselineWealthCharacteristicsValueViewSet,
     BaselineWealthGroupViewSet,
     ButterProductionViewSet,
     CommunityCropProductionViewSet,
@@ -96,6 +97,9 @@
 router.register(r"baselinewealthgroup", BaselineWealthGroupViewSet)
 router.register(r"communitywealthgroup", CommunityWealthGroupViewSet)
 router.register(r"wealthgroupcharacteristicvalue", WealthGroupCharacteristicValueViewSet)
+router.register(
+    r"baselinewealthcharacteristicsvalue", BaselineWealthCharacteristicsValueViewSet, "baselinewealthcharacteristics"
+)
 router.register(r"livelihoodstrategy", LivelihoodStrategyViewSet)
 router.register(r"livelihoodactivity", LivelihoodActivityViewSet)
 router.register(r"baselinelivelihoodactivity", BaselineLivelihoodActivityViewSet)