summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDana Powers <dana.powers@rd.io>2016-01-25 15:37:13 -0800
committerDana Powers <dana.powers@rd.io>2016-01-25 15:37:13 -0800
commitcd956d8d7926fd294069c9c9897b5fb487862eb2 (patch)
tree385bf11447f12aef7c1a6db3793623c612bf32a5
parent7e09258409633ef3f3dbc4f45b89c1a5bb09b05c (diff)
downloadkafka-python-cd956d8d7926fd294069c9c9897b5fb487862eb2.tar.gz
Dont need context manager for BytesIO
-rw-r--r--kafka/codec.py40
1 files changed, 18 insertions, 22 deletions
diff --git a/kafka/codec.py b/kafka/codec.py
index 0109a66..9eaeeca 100644
--- a/kafka/codec.py
+++ b/kafka/codec.py
@@ -38,33 +38,29 @@ def gzip_encode(payload, compresslevel=None):
if not compresslevel:
compresslevel = 9
- with BytesIO() as buf:
+ buf = BytesIO()
- # 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", compresslevel=compresslevel)
- try:
- gzipper.write(payload)
- finally:
- gzipper.close()
+ # Gzip context manager introduced in python 2.7
+ # so old-fashioned way until we decide to not support 2.6
+ gzipper = gzip.GzipFile(fileobj=buf, mode="w", compresslevel=compresslevel)
+ try:
+ gzipper.write(payload)
+ finally:
+ gzipper.close()
- result = buf.getvalue()
-
- return result
+ return buf.getvalue()
def gzip_decode(payload):
- with BytesIO(payload) 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='r')
- try:
- result = gzipper.read()
- finally:
- gzipper.close()
-
- return result
+ buf = BytesIO(payload)
+
+ # Gzip context manager introduced in python 2.7
+ # so old-fashioned way until we decide to not support 2.6
+ gzipper = gzip.GzipFile(fileobj=buf, mode='r')
+ try:
+ return gzipper.read()
+ finally:
+ gzipper.close()
def snappy_encode(payload, xerial_compatible=True, xerial_blocksize=32*1024):