aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-19 06:54:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update geoip.py (#1896)

Browse Source 
Added error handling in case `IP_ADDR_LOOKUP.get(addr)` fails.
This commit is contained in:
Mike Merrill
2021-05-28 11:03:56 -04:00
committed by GitHub
parent f00e69d619
commit 8646f2384c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
CTFd/utils/countries/geoip.py
View File

@@ -8,8 +8,8 @@ IP_ADDR_LOOKUP = maxminddb.open_database(
def lookup_ip_address(addr): def lookup_ip_address(addr):
response = IP_ADDR_LOOKUP.get(addr)
try: try:
response = IP_ADDR_LOOKUP.get(addr)
return response["country"]["iso_code"] return response["country"]["iso_code"]
except KeyError: except (KeyError, ValueError):
return None return None