66from drf_spectacular .utils import extend_schema_field
77from rest_framework import serializers
88
9- from api .models import Appeal , Country , Region
9+ from api .models import Appeal , AppealType , Country , Region
1010from api .serializers import (
1111 MiniCountrySerializer ,
1212 RegoCountrySerializer ,
@@ -867,6 +867,7 @@ class Meta:
867867class 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
920925class 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