Stowage/msgpack-python
2
0
Fork 0
mirror of https://github.com/msgpack/msgpack-python.git synced 2025-10-19 20:03:16 +00:00
Code Issues Projects Releases Packages Wiki Activity

Support packing memoryview objects

Browse source
This commit is contained in:
folz 2015-11-12 11:49:19 +01:00
parent 8036cb4e0e
commit 31adc5a3c0
3 changed files with 33 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
msgpack/fallback.py
View file

@ -36,6 +36,8 @@ if hasattr(sys, 'pypy_version_info'):
else:
self.builder = StringBuilder()
def write(self, s):
if isinstance(s, memoryview):
s = s.tobytes()
self.builder.append(s)
def getvalue(self):
return self.builder.build()
@ -682,7 +684,7 @@ class Packer(object):
default_used = True
continue
raise PackValueError("Integer value out of range")
if self._use_bin_type and check(obj, bytes):
if self._use_bin_type and check(obj, (bytes, memoryview)):
n = len(obj)
if n <= 0xff:
self._buffer.write(struct.pack('>BB', 0xc4, n))
@ -693,7 +695,7 @@ class Packer(object):
else:
raise PackValueError("Bytes is too large")
return self._buffer.write(obj)
if check(obj, (Unicode, bytes)):
if check(obj, (Unicode, bytes, memoryview)):
if check(obj, Unicode):
if self._encoding is None:
raise TypeError(