diff options
author | Ivan Pouzyrevsky <sandello@yandex-team.ru> | 2013-06-02 02:04:30 +0400 |
---|---|---|
committer | David Arthur <mumrah@gmail.com> | 2013-06-07 21:20:55 -0400 |
commit | d7178e188e9ce82bef702e38ef4a0eaac49cd2e4 (patch) | |
tree | 4135038e7b34e614b96c14e844ca0da3c6dea23b /kafka/codec.py | |
parent | f444caaee77f0cb61148f4e7b05670bc19729f3e (diff) | |
download | kafka-python-d7178e188e9ce82bef702e38ef4a0eaac49cd2e4.tar.gz |
Beautify codec.py
Diffstat (limited to 'kafka/codec.py')
-rw-r--r-- | kafka/codec.py | 45 |
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) |