Fix "dumping" message to work correctly for .gz files

This commit is contained in:
Matthias 2019-12-25 10:35:23 +01:00
parent abc6b9459a
commit 8f214aec89
1 changed files with 3 additions and 1 deletions

View File

@ -47,14 +47,16 @@ def file_dump_json(filename: Path, data, is_zip=False) -> None:
:param data: JSON Data to save
:return:
"""
logger.info(f'dumping json to "{filename}"')
if is_zip:
if filename.suffix != '.gz':
filename = filename.with_suffix('.gz')
logger.info(f'dumping json to "{filename}"')
with gzip.open(filename, 'w') as fp:
rapidjson.dump(data, fp, default=str, number_mode=rapidjson.NM_NATIVE)
else:
logger.info(f'dumping json to "{filename}"')
with open(filename, 'w') as fp:
rapidjson.dump(data, fp, default=str, number_mode=rapidjson.NM_NATIVE)