diff options
| author | Dana Powers <dana.powers@rd.io> | 2015-12-09 15:13:54 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2015-12-09 15:13:54 -0800 |
| commit | c4f87bce204d27cb7897baccab8454f997ce6c49 (patch) | |
| tree | 13c4a3ed98e776b9ea6121e7b923c562263e3100 /kafka/protocol/message.py | |
| parent | 58bdeb17d7e337c48ee2c14bf1f73b00eed0e727 (diff) | |
| download | kafka-python-c4f87bce204d27cb7897baccab8454f997ce6c49.tar.gz | |
Fix BytesIO import in kafka.protocol.message
Diffstat (limited to 'kafka/protocol/message.py')
| -rw-r--r-- | kafka/protocol/message.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/kafka/protocol/message.py b/kafka/protocol/message.py index 501ce47..3027ebd 100644 --- a/kafka/protocol/message.py +++ b/kafka/protocol/message.py @@ -1,3 +1,5 @@ +import io + from .struct import Struct from .types import ( Int8, Int32, Int64, Bytes, Schema, AbstractType @@ -34,7 +36,7 @@ class Message(Struct): @classmethod def decode(cls, data): if isinstance(data, bytes): - data = BytesIO(data) + data = io.BytesIO(data) fields = [field.decode(data) for field in cls.SCHEMA.fields] return cls(fields[4], key=fields[3], magic=fields[1], attributes=fields[2], crc=fields[0]) |