Handle AddressNotFoundError in nodemapper.py (#25)

dockerfiles/nodemapper.py

@@ -12,6 +12,7 @@ from os import environ as env
 
 import requests
 import geoip2.database
+from geoip2.errors import AddressNotFoundError
 from flask import Flask, make_response
 
 
@@ -24,8 +25,11 @@ NODE_PORT = env.get('NODE_PORT', 18083)
 
 def get_geoip(ip):
     """Takes an IP address and determines GeoIP data"""
-    with geoip2.database.Reader("./geoip.mmdb") as reader:
-        return reader.city(ip)
+    try:
+        with geoip2.database.Reader("./geoip.mmdb") as reader:
+            return reader.city(ip)
+    except AddressNotFoundError:
+        return None
 
 
 @app.route("/metrics")
@@ -35,6 +39,9 @@ def nodes():
     peer_list = requests.get(f'http://{NODE_HOST}:{NODE_PORT}/get_peer_list').json()
     def add_peer(host, status):
         geo = get_geoip(host)
+        if geo is None:
+            return
+
         geostr = 'geoip{{latitude="{lat}", longitude="{lon}", country_code="{country_code}", country_name="{country_name}", status="{status}"}} 1'
         if geostr not in peers:
             if 'en' in geo.continent.names:
@@ -54,4 +61,4 @@ def nodes():
     data = '\n'.join(peers)
     response = make_response(data, 200)
     response.mimetype = "text/plain"
-    return response
+    return response