From 8646f2384ce73578621a8a7897e94ba0e1fbfd60 Mon Sep 17 00:00:00 2001
From: Mike Merrill <linted@users.noreply.github.com>
Date: Fri, 28 May 2021 11:03:56 -0400
Subject: [PATCH] Update geoip.py (#1896)

Added error handling in case `IP_ADDR_LOOKUP.get(addr)` fails.
---
 CTFd/utils/countries/geoip.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/CTFd/utils/countries/geoip.py b/CTFd/utils/countries/geoip.py
index f2e29c20..4935ff31 100644
--- a/CTFd/utils/countries/geoip.py
+++ b/CTFd/utils/countries/geoip.py
@@ -8,8 +8,8 @@ IP_ADDR_LOOKUP = maxminddb.open_database(
 
 
 def lookup_ip_address(addr):
-    response = IP_ADDR_LOOKUP.get(addr)
     try:
+        response = IP_ADDR_LOOKUP.get(addr)
         return response["country"]["iso_code"]
-    except KeyError:
+    except (KeyError, ValueError):
         return None