Fixes Thriftpy#302 -- deprecation warnings when array.tostring() is c…

…alled on Python 3.3+.
terwilliger42 · Jun 29, 2017 · 7ad8056 · 7ad8056
1 parent b30e41d
commit 7ad8056
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/thriftpy/protocol/compact.py b/thriftpy/protocol/compact.py
@@ -58,7 +58,10 @@ def write_varint(trans, n):
         else:
             out.append((n & 0xff) | 0x80)
             n = n >> 7
-    data = array.array('B', out).tostring()
+
+    # 3.2+ uses tobytes(). Thriftpy doesn't support Python 3 pre-3.3
+    a = array.array('B', out)
+    data = a.tobytes() if PY3 else a.tostring()
 
     if PY3:
         trans.write(data)