Add Geo IP location in filter users

This commit is contained in:
Gregory Soutade 2022-11-04 20:03:52 +01:00
parent c84373b2d7
commit 7ec7e0d644

View File

@ -23,6 +23,7 @@ import re
from iwla import IWLA
from iplugin import IPlugin
from display import *
from misc.geoiplookup import geoiplookup
"""
Post analysis hook
@ -67,6 +68,7 @@ Statistics creation :
visits :
remote_addr =>
filtered
geo_location
Statistics update :
visits :
@ -167,10 +169,19 @@ class IWLAPostAnalysisFilterUsers(IPlugin):
hits = self.iwla.getValidVisitors()
if len(self.filters) or len(self.ip_filters):
for (key,value) in hits.items():
if self._do_filter(key, hits):
print('Keep for {}'.format(key))
value['filtered'] = True
value['keep_requests'] = True
for (key,value) in hits.items():
# Already filtered
if value.get('filtered', False):
continue
if self._do_filter(key, hits):
value['filtered'] = True
value['keep_requests'] = True
try:
if not value.get('geo_location', None):
value['geo_location'] = geoiplookup(value['remote_ip'])
except Exception as e:
print(e)