From c4f87bce204d27cb7897baccab8454f997ce6c49 Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@rd.io>
Date: Wed, 9 Dec 2015 15:13:54 -0800
Subject: Fix BytesIO import in kafka.protocol.message

---
 kafka/protocol/message.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'kafka')

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])
-- 
cgit v1.2.1