Skip to content

Commit 17829b6

Browse files
committed
Create header inline rather than calling a function
1 parent c49383f commit 17829b6

File tree

1 file changed

+5
-15
lines changed

1 file changed

+5
-15
lines changed

src/isal/igzip.py

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -225,13 +225,10 @@ def write(self, data):
225225
_IGzipReader = _GzipReader
226226

227227

228-
def _create_simple_gzip_header(compresslevel: int,
229-
mtime: Optional[SupportsInt] = None) -> bytes:
230-
"""
231-
Write a simple gzip header with no extra fields.
232-
:param compresslevel: Compresslevel used to determine the xfl bytes.
233-
:param mtime: The mtime (must support conversion to a 32-bit integer).
234-
:return: A bytes object representing the gzip header.
228+
def compress(data, compresslevel: int = _COMPRESS_LEVEL_BEST, *,
229+
mtime: Optional[SupportsInt] = None) -> bytes:
230+
"""Compress data in one shot and return the compressed string.
231+
Optional argument is the compression level, in range of 0-3.
235232
"""
236233
if mtime is None:
237234
mtime = time.time()
@@ -240,14 +237,7 @@ def _create_simple_gzip_header(compresslevel: int,
240237
xfl = 4 if compresslevel == _COMPRESS_LEVEL_FAST else 0
241238
# Pack ID1 and ID2 magic bytes, method (8=deflate), header flags (no extra
242239
# fields added to header), mtime, xfl and os (255 for unknown OS).
243-
return struct.pack("<BBBBLBB", 0x1f, 0x8b, 8, 0, int(mtime), xfl, 255)
244-
245-
246-
def compress(data, compresslevel=_COMPRESS_LEVEL_BEST, *, mtime=None):
247-
"""Compress data in one shot and return the compressed string.
248-
Optional argument is the compression level, in range of 0-3.
249-
"""
250-
header = _create_simple_gzip_header(compresslevel, mtime)
240+
header = struct.pack("<BBBBLBB", 0x1f, 0x8b, 8, 0, int(mtime), xfl, 255)
251241
# use igzip_lib to compress the data without a gzip header but with a
252242
# gzip trailer.
253243
compressed = igzip_lib.compress(data, compresslevel,

0 commit comments

Comments
 (0)