Merge pull request #21 from alpacahq/fix-py2-write

Fix python2 write
alpacahq · Sep 11, 2018 · cb24bb5 · cb24bb5
2 parents ecfe895 + 3de7892
commit cb24bb5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pymarketstore/client.py b/pymarketstore/client.py
@@ -4,6 +4,7 @@
 import re
 import requests
 import logging
+import six
 
 from .jsonrpc import JsonRpcClient, MsgpackRpcClient
 from .results import QueryReply
@@ -103,7 +104,8 @@ def write(self, recarray, tbk, isvariablelength=False):
         ]
         data['names'] = recarray.dtype.names
         data['data'] = [
-            bytes(memoryview(recarray[name]))
+            bytes(buffer(recarray[name])) if six.PY2
+                else bytes(memoryview(recarray[name]))
             for name in recarray.dtype.names
         ]
         data['length'] = len(recarray)