d_mcknight/Varken
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add logging to geo_lookup.

Browse source
This commit is contained in:
samwiseg0 2018-12-10 19:16:13 -08:00
parent 5d284a5259
commit 0e844d5a15
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
varken/helpers.py
View file

@ -39,6 +39,8 @@ def geoip_download(data_folder):
def geo_lookup(ipaddress, data_folder):
datafolder = data_folder
logging.debug('Reading GeoLite2 from %s', datafolder)
dbfile = abspath(join(datafolder, 'GeoLite2-City.mmdb'))
now = time.time()
@ -46,9 +48,13 @@ def geo_lookup(ipaddress, data_folder):
dbinfo = os.stat(dbfile)
db_age = now - dbinfo.st_ctime
if db_age > (35 * 86400):
logging.info('GeoLite2 DB is older than 35 days. Attempting to re-download...')
os.remove(dbfile)
geoip_download(datafolder)
except FileNotFoundError:
logging.error('GeoLite2 DB not found. Attempting to download...')
geoip_download(datafolder)
reader = geoip2.database.Reader(dbfile)