summaryrefslogtreecommitdiff
path: root/kafka/codec.py
diff options
context:
space:
mode:
Diffstat (limited to 'kafka/codec.py')
-rw-r--r--kafka/codec.py45
1 files changed, 21 insertions, 24 deletions
diff --git a/kafka/codec.py b/kafka/codec.py
index cb3858b..35102ca 100644
--- a/kafka/codec.py
+++ b/kafka/codec.py
@@ -1,51 +1,48 @@
from cStringIO import StringIO
import gzip
-import logging
-
-log = logging.getLogger("kafka.codec")
try:
import snappy
- hasSnappy = True
+ _has_snappy = True
except ImportError:
- log.warn("Snappy codec not available")
- hasSnappy = False
+ _has_snappy = False
+
def has_gzip():
return True
-def has_snappy():
- return hasSnappy
+def _has_snappy():
+ return _has_snappy
def gzip_encode(payload):
- buf = StringIO()
- f = gzip.GzipFile(fileobj=buf, mode='w', compresslevel=6)
- f.write(payload)
- f.close()
- buf.seek(0)
- out = buf.read()
- buf.close()
- return out
+ buffer = StringIO()
+ handle = gzip.GzipFile(fileobj=buffer, mode="w")
+ handle.write(payload)
+ handle.close()
+ buffer.seek(0)
+ result = buffer.read()
+ buffer.close()
+ return result
def gzip_decode(payload):
- buf = StringIO(payload)
- f = gzip.GzipFile(fileobj=buf, mode='r')
- out = f.read()
- f.close()
- buf.close()
- return out
+ buffer = StringIO(payload)
+ handle = gzip.GzipFile(fileobj=buffer, mode='r')
+ result = handle.read()
+ handle.close()
+ buffer.close()
+ return result
def snappy_encode(payload):
- if not hasSnappy:
+ if not _has_snappy:
raise NotImplementedError("Snappy codec not available")
return snappy.compress(payload)
def snappy_decode(payload):
- if not hasSnappy:
+ if not _has_snappy:
raise NotImplementedError("Snappy codec not available")
return snappy.decompress(payload)