Merge pull request #1 from ebdavison/ebdavison-issue27

fix for issue cyberdefenders#27
gcsuaveiro · Apr 24, 2023 · 240dce1 · 240dce1
2 parents dc0b3f0 + 25ae2f0
commit 240dce1
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/mha/server.py b/mha/server.py
@@ -34,12 +34,15 @@ def getCountryForIP(line):
         if ip:
             ip = ip[0]  # take the 1st ip and ignore the rest
             if IP(ip).iptype() == 'PUBLIC':
-                r = reader.country(ip).country
-                if r.iso_code and r.name:
-                    return {
-                        'iso_code': r.iso_code.lower(),
-                        'country_name': r.name
-                    }
+                try:
+                    r = reader.country(ip).country
+                    if r.iso_code and r.name:
+                        return {
+                            'iso_code': r.iso_code.lower(),
+                            'country_name': r.name
+                        }
+                except:
+                    pass
     return dict(country=getCountryForIP)