Skip to content

Commit e72d18a

Browse files
committed
Standardize wealth_group_category and wealth_group_category_name - see HEA-659
1 parent a8b9e5b commit e72d18a

5 files changed

Lines changed: 20 additions & 2 deletions

File tree

apps/baseline/models.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -554,6 +554,12 @@ def natural_key(self):
554554
class Meta:
555555
verbose_name = _("Wealth Group")
556556
verbose_name_plural = _("Wealth Groups")
557+
ordering = [
558+
"livelihood_zone_baseline__livelihood_zone__code",
559+
"livelihood_zone_baseline__reference_year_end_date",
560+
"wealth_group_category__ordering",
561+
"community__name",
562+
]
557563
constraints = [
558564
models.UniqueConstraint(
559565
fields=("livelihood_zone_baseline", "wealth_group_category", "community"),

apps/baseline/serializers.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -270,6 +270,7 @@ class Meta:
270270
"community",
271271
"community_name",
272272
"wealth_group_category",
273+
"wealth_group_category_name",
273274
"percentage_of_households",
274275
"average_household_size",
275276
]
@@ -296,6 +297,7 @@ def get_livelihood_zone_baseline_label(self, obj):
296297
source_organization_name = serializers.CharField(
297298
source="livelihood_zone_baseline.source_organization.name", read_only=True
298299
)
300+
wealth_group_category_name = serializers.CharField(source="wealth_group_category.name", read_only=True)
299301

300302

301303
class BaselineWealthGroupSerializer(WealthGroupSerializer):
@@ -1554,7 +1556,8 @@ class Meta:
15541556
"id",
15551557
"name",
15561558
"description",
1557-
"wealth_group_category_code",
1559+
"wealth_group_category",
1560+
"wealth_group_category_name",
15581561
"main_livelihood_category",
15591562
"bss",
15601563
"currency",
@@ -1601,7 +1604,8 @@ def field_to_database_path(field_name):
16011604
"valid_to_date": "valid_to_date",
16021605
"livelihood_strategy_pk": "livelihood_strategies__pk",
16031606
"livelihood_activity_pk": "livelihood_strategies__livelihoodactivity__pk",
1604-
"wealth_group_category_code": "livelihood_strategies__livelihoodactivity__wealth_group__wealth_group_category__code", # NOQA: E501
1607+
"wealth_group_category": "livelihood_strategies__livelihoodactivity__wealth_group__wealth_group_category__code", # NOQA: E501
1608+
"wealth_group_category_name": f"livelihood_strategies__livelihoodactivity__wealth_group__wealth_group_category__name_{language_code}", # NOQA: E501
16051609
"kcals_consumed": "livelihood_strategies__livelihoodactivity__kcals_consumed",
16061610
"income": "livelihood_strategies__livelihoodactivity__income",
16071611
"expenditure": "livelihood_strategies__livelihoodactivity__expenditure",

apps/baseline/viewsets.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -370,6 +370,12 @@ class WealthGroupViewSet(BaseModelViewSet):
370370
)
371371
serializer_class = WealthGroupSerializer
372372
filterset_class = WealthGroupFilterSet
373+
ordering = [
374+
"livelihood_zone_baseline__livelihood_zone__code",
375+
"livelihood_zone_baseline__reference_year_end_date",
376+
"wealth_group_category__ordering",
377+
"community__name",
378+
]
373379

374380

375381
class BaselineWealthGroupFilterSet(filters.FilterSet):

apps/metadata/admin.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ class ReferenceDataAdmin(admin.ModelAdmin):
2323
*translation_fields("name"),
2424
"aliases",
2525
*translation_fields("description"),
26+
"ordering",
2627
)
2728
list_display = (
2829
"code",

apps/metadata/models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ def __str__(self):
8989

9090
class Meta:
9191
abstract = True
92+
ordering = ["ordering", "code"]
9293

9394
class ExtraMeta:
9495
identifier = ["name_en"]

0 commit comments

Comments
 (0)