openstack/deb-python-kafka
Code Issues Proposed changes

Fix BytesIO import in kafka.protocol.message

Browse Source
This commit is contained in:
Dana Powers
2015-12-09 15:13:54 -08:00
parent 58bdeb17d7
commit c4f87bce20
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kafka/protocol/message.py
View File

@@ -1,3 +1,5 @@
import io
from .struct import Struct from .struct import Struct
from .types import ( from .types import (
Int8, Int32, Int64, Bytes, Schema, AbstractType Int8, Int32, Int64, Bytes, Schema, AbstractType
@@ -34,7 +36,7 @@ class Message(Struct):
@classmethod @classmethod
def decode(cls, data): def decode(cls, data):
if isinstance(data, bytes): if isinstance(data, bytes):
data = BytesIO(data) data = io.BytesIO(data)
fields = [field.decode(data) for field in cls.SCHEMA.fields] fields = [field.decode(data) for field in cls.SCHEMA.fields]
return cls(fields[4], key=fields[3], return cls(fields[4], key=fields[3],
magic=fields[1], attributes=fields[2], crc=fields[0]) magic=fields[1], attributes=fields[2], crc=fields[0])