From 1fd6eef44b0fca71e7886baf7d800a7fa400e368 Mon Sep 17 00:00:00 2001 From: "Nicholas St. Germain" Date: Tue, 12 Feb 2019 12:44:32 -0600 Subject: [PATCH] open/close file for windows issues --- varken/helpers.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/varken/helpers.py b/varken/helpers.py index 93817be..8657d43 100644 --- a/varken/helpers.py +++ b/varken/helpers.py @@ -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: