Change GeoLite2 tar extraction

This commit is contained in:
JonnyWong16 2019-12-30 14:19:12 -08:00
parent 54bbbb36a6
commit e334a0fc8b

View file

@ -648,9 +648,9 @@ def install_geoip_db(update=False):
with tarfile.open(temp_gz, 'r:gz') as tar: with tarfile.open(temp_gz, 'r:gz') as tar:
for member in tar.getmembers(): for member in tar.getmembers():
if geolite2_db in member.name: if geolite2_db in member.name:
mmdb = tar.extractfile(member) member.name = os.path.basename(member.name)
with open(geolite2_db_path, 'wb') as db: tar.extract(member, os.path.dirname(geolite2_db_path))
db.write(mmdb.read()) mmdb = True
break break
if not mmdb: if not mmdb:
raise Exception("{} not found in gzip file.".format(geolite2_db)) raise Exception("{} not found in gzip file.".format(geolite2_db))