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,13 +923,23 @@ 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 
924934        fields  =  "__all__" 
925935
926936
927937class  MicroAppealSerializer (serializers .ModelSerializer ):
938+     atype  =  serializers .SerializerMethodField ()
939+ 
940+     @staticmethod  
941+     def  get_atype (obj ):
942+         return  obj .atype  or  AppealType (obj .atype ).label 
928943
929944    class  Meta :
930945        model  =  Appeal 
0 commit comments