dispay message in searchcard

backend/api/account/client/serializers.py

@@ -7,6 +7,8 @@ import pytz
 class RouteSerializer(serializers.ModelSerializer):
     from_city_name = serializers.SerializerMethodField()
     to_city_name = serializers.SerializerMethodField()
+    from_country_name = serializers.SerializerMethodField()
+    to_country_name = serializers.SerializerMethodField()
     formatted_departure = serializers.SerializerMethodField()
     formatted_arrival = serializers.SerializerMethodField()
     formatted_cargo_type = serializers.SerializerMethodField()
@@ -30,6 +32,20 @@ class RouteSerializer(serializers.ModelSerializer):
         except City.DoesNotExist:
             return None
 
+    def get_from_country_name(self, obj):
+        try:
+            city = City.objects.get(id=obj.from_city_id)
+            return city.country.international_name or city.country.official_name
+        except (City.DoesNotExist, AttributeError):
+            return None
+
+    def get_to_country_name(self, obj):
+        try:
+            city = City.objects.get(id=obj.to_city_id)
+            return city.country.international_name or city.country.official_name
+        except (City.DoesNotExist, AttributeError):
+            return None
+
     def _convert_to_local_time(self, dt):
         if dt is None:
             return None