Fix missing country in whois lookup causing error

2018-03-02 15:39:05 -08:00
parent 66167d5960
commit 9d08717c83
1 changed files with 1 additions and 1 deletions
--- a/plexpy/helpers.py
+++ b/plexpy/helpers.py
@@ -646,7 +646,7 @@ def whois_lookup(ip_address):
        countries = ipwhois.utils.get_countries()
        nets = whois['nets']
        for net in nets:
-            net['country'] = countries[net['country']]
+            net['country'] = countries.get(net['country'])
            if net['postal_code']:
                 net['postal_code'] = net['postal_code'].replace('-', ' ')
    except ValueError as e: