From 4d516fbd496096ae30191e13b7c2e378654b3759 Mon Sep 17 00:00:00 2001 From: trbs Date: Sat, 12 Sep 2015 18:22:50 +0200 Subject: allow to specify compression level for codecs which support this --- kafka/codec.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'kafka/codec.py') diff --git a/kafka/codec.py b/kafka/codec.py index 19f405b..a9373c7 100644 --- a/kafka/codec.py +++ b/kafka/codec.py @@ -22,12 +22,15 @@ def has_snappy(): return _HAS_SNAPPY -def gzip_encode(payload): +def gzip_encode(payload, compresslevel=None): + if not compresslevel: + compresslevel = 9 + with BytesIO() as buf: # Gzip context manager introduced in python 2.6 # so old-fashioned way until we decide to not support 2.6 - gzipper = gzip.GzipFile(fileobj=buf, mode="w") + gzipper = gzip.GzipFile(fileobj=buf, mode="w", compresslevel=compresslevel) try: gzipper.write(payload) finally: -- cgit v1.2.1