Fix #326, catch a geoip2 database error in GraphQL

rsmusllp · Oct 10, 2018 · 1970ea2 · 1970ea2
1 parent 42fc8d7
commit 1970ea2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/king_phisher/server/graphql/types/misc.py b/king_phisher/server/graphql/types/misc.py
@@ -37,6 +37,7 @@
 import king_phisher.geoip as geoip
 import king_phisher.ipaddress as ipaddress
 
+import geoip2.errors
 import graphene.types.utils
 import graphql.language.ast
 
@@ -110,7 +111,10 @@ def from_ip_address(cls, ip_address):
 		ip_address = ipaddress.ip_address(ip_address)
 		if ip_address.is_private:
 			return
-		result = geoip.lookup(ip_address)
+		try:
+			result = geoip.lookup(ip_address)
+		except geoip2.errors.AddressNotFoundError:
+			result = None
 		if result is None:
 			return
 		return cls(**result)