anonfunc/JellyPy
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change GeoLite2 tar extraction

Browse Source
This commit is contained in:
JonnyWong16
2019-12-30 14:19:12 -08:00
parent 54bbbb36a6
commit e334a0fc8b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
plexpy/helpers.py
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))