Skip to content

Commit 17d0645

Browse files
committed
Add appealtype name to ops-learning
1 parent 3967220 commit 17d0645

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

per/serializers.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from drf_spectacular.utils import extend_schema_field
77
from rest_framework import serializers
88

9-
from api.models import Appeal, Country, Region
9+
from api.models import Appeal, AppealType, Country, Region
1010
from api.serializers import (
1111
MiniCountrySerializer,
1212
RegoCountrySerializer,
@@ -867,6 +867,7 @@ class Meta:
867867
class MiniAppealSerializer(serializers.ModelSerializer):
868868
name = serializers.SerializerMethodField()
869869
start_date = serializers.SerializerMethodField()
870+
atype = serializers.SerializerMethodField()
870871
dtype = serializers.SerializerMethodField()
871872
country = serializers.SerializerMethodField()
872873
region = serializers.SerializerMethodField()
@@ -881,6 +882,10 @@ def get_name(obj):
881882
def get_start_date(obj):
882883
return obj.start_date and obj.start_date.year
883884

885+
@staticmethod
886+
def get_atype(obj):
887+
return obj.atype or AppealType(obj.atype).label
888+
884889
@staticmethod
885890
def get_dtype(obj):
886891
return obj.dtype and obj.dtype.name
@@ -918,6 +923,11 @@ class Meta:
918923

919924

920925
class FullAppealSerializer(serializers.ModelSerializer):
926+
atype = serializers.SerializerMethodField()
927+
928+
@staticmethod
929+
def get_atype(obj):
930+
return obj.atype or AppealType(obj.atype).label
921931

922932
class Meta:
923933
model = Appeal

0 commit comments

Comments
 (0)