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

open/close file for windows issues

Browse source
This commit is contained in:
Nicholas St. Germain 2019-02-12 12:44:32 -06:00
parent 4eda93a4bf
commit 1fd6eef44b
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
varken/helpers.py
View file

@ -24,7 +24,14 @@ class GeoIPHandler(object):
self.update()
self.logger.info('Opening persistent connection to GeoLite2 DB...')
self.reader = Reader(self.dbfile)
self.reader = None
self.reader_manager(action='open')
def reader_manager(self, action=None):
if action == 'open':
self.reader = Reader(self.dbfile)
else:
self.reader.close()
def lookup(self, ipaddress):
ip = ipaddress
@ -47,8 +54,10 @@ class GeoIPHandler(object):
if dbdate < first_wednesday_date < today:
self.logger.info("Newer GeoLite2 DB available, Updating...")
self.reader_manager(action='close')
remove(self.dbfile)
self.download()
self.reader_manager(action='open')
else:
td = first_wednesday_date - today
if td.days < 0: