util: Handle cases where a country may not have an iso_code

This commit is contained in:
OatmealDome
2025-08-21 23:16:54 +00:00
committed by OatmealDome
parent 4e86ba8c69
commit 5bb86b5712

View File

@@ -12,7 +12,12 @@ SECRET_KEY_LENGTH = 10
def get_ip_region(ip):
reader = geoip2.database.Reader(settings.GEOIP_DATABASE_PATH)
try:
return reader.country(ip).country.iso_code.lower()
iso_code = reader.country(ip).country.iso_code
if iso_code is None:
return None
return iso_code.lower()
except geoip2.errors.AddressNotFoundError:
return None