soutade/iwla
1
0
Fork 0
Code Issues Pull Requests Releases 2 Activity

Check for presence of country_code field to avoid key error exception

Browse Source
This commit is contained in:
Gregory Soutade 2016-02-07 10:17:32 +01:00
parent 19e9b9d4fb
commit 9bb188b6fc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
plugins/display/ip_to_geo.py
View File

@ -70,11 +70,11 @@ class IWLADisplayTopGeo(IPlugin):
cc = None cc = None
host_name = host.split(' ')[0] # hostname or ip host_name = host.split(' ')[0] # hostname or ip
if host_name in self.valid_visitors.keys(): if host_name in self.valid_visitors.keys():
cc = self.valid_visitors[host_name]['country_code'] cc = self.valid_visitors[host_name].get('country_code', None)
else: else:
for visitor in self.valid_visitors.values(): for visitor in self.valid_visitors.values():
if visitor['remote_addr'] == host_name: if visitor['remote_addr'] == host_name:
cc = visitor['country_code'] cc = visitor.get('country_code', None)
break break
if not cc or cc == 'ip': return None if not cc or cc == 'ip': return None
icon = '<img src="/%s/flags/%s.png"/>' % (self.icon_path, cc) icon = '<img src="/%s/flags/%s.png"/>' % (self.icon_path, cc)