Added Country flags image link

Author: awaismslm

ipinfo/handler_utils.py

@@ -12,6 +12,10 @@
 # Base URL to make requests against.
 API_URL = "https://ipinfo.io"
 
+# Base URL to get country flag image link.
+# "PK" -> "https://cdn.ipinfo.io/static/images/countries-flags/PK.svg"
+COUNTRY_FLAGS_URL = "https://cdn.ipinfo.io/static/images/countries-flags/"
+
 # Used to transform incoming responses with country abbreviations into the full
 # expanded country name, e.g. "PK" -> "Pakistan".
 COUNTRY_FILE_DEFAULT = "countries.json"
@@ -75,6 +79,7 @@ def format_details(
     """
     details["country_name"] = countries.get(details.get("country"))
     details["isEU"] = details.get("country") in eu_countries
+    details["country_flag_url"] = COUNTRY_FLAGS_URL + details.get("country") + ".svg"
     details["country_flag"] = copy.deepcopy(
         countries_flags.get(details.get("country"))
     )

tests/handler_async_test.py

@@ -47,6 +47,8 @@ async def test_get_details():
     country_flag = details.country_flag
     assert country_flag["emoji"] == "🇺🇸"
     assert country_flag["unicode"] == "U+1F1FA U+1F1F8"
+    country_flag_url = details.country_flag_url
+    assert country_flag_url == "https://cdn.ipinfo.io/static/images/countries-flags/US.svg"
     country_currency = details.country_currency
     assert country_currency["code"] == "USD"
     assert country_currency["symbol"] == "$"

tests/handler_test.py

@@ -44,6 +44,8 @@ def test_get_details():
     country_flag = details.country_flag
     assert country_flag["emoji"] == "🇺🇸"
     assert country_flag["unicode"] == "U+1F1FA U+1F1F8"
+    country_flag_url = details.country_flag_url
+    assert country_flag_url == "https://cdn.ipinfo.io/static/images/countries-flags/US.svg"
     country_currency = details.country_currency
     assert country_currency["code"] == "USD"
     assert country_currency["symbol"] == "$"