Avoid doubly-compressing zstandard files

This commit is contained in:
Sybren A. Stüvel
2022-02-03 17:54:03 +01:00
parent 2bf155a5f9
commit e5c7e38b44
5 changed files with 18 additions and 8 deletions
@@ -73,7 +73,7 @@ class Compression(enum.Enum):
def open(path: pathlib.Path, mode: str, buffer_size: int) -> DecompressedFileInfo:
"""Open the file, decompressing it into a temporary file if necesssary."""
fileobj = path.open(mode, buffering=buffer_size) # typing.IO[bytes]
compression = _find_compression_type(fileobj)
compression = find_compression_type(fileobj)
if compression == Compression.UNRECOGNISED:
fileobj.close()
@@ -113,7 +113,7 @@ def open(path: pathlib.Path, mode: str, buffer_size: int) -> DecompressedFileInf
)
def _find_compression_type(fileobj: typing.IO[bytes]) -> Compression:
def find_compression_type(fileobj: typing.IO[bytes]) -> Compression:
fileobj.seek(0, os.SEEK_SET)
# This assumes that all magics are not longer than "BLENDER".