Stowage/msgpack-python
2
0
Fork 0
mirror of https://github.com/msgpack/msgpack-python.git synced 2025-11-01 01:50:54 +00:00
Code Issues Projects Releases Packages Wiki Activity

Add Packer.pack_pairs.

Browse source
This commit is contained in:
INADA Naoki 2012-12-10 21:26:41 +09:00
parent 0c7ab7c344
commit 537a2ab3f2
2 changed files with 44 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

39
msgpack/_msgpack.pyx
View file

@ -186,18 +186,49 @@ cdef class Packer(object):
self.pk.length = 0
return buf
cpdef pack_array_header(self, size_t size):
msgpack_pack_array(&self.pk, size)
def pack_array_header(self, size_t size):
cdef int ret = msgpack_pack_array(&self.pk, size)
if ret == -1:
raise MemoryError
elif ret: # should not happen
raise TypeError
buf = PyBytes_FromStringAndSize(self.pk.buf, self.pk.length)
self.pk.length = 0
return buf
cpdef pack_map_header(self, size_t size):
msgpack_pack_map(&self.pk, size)
def pack_map_header(self, size_t size):
cdef int ret = msgpack_pack_map(&self.pk, size)
if ret == -1:
raise MemoryError
elif ret: # should not happen
raise TypeError
buf = PyBytes_FromStringAndSize(self.pk.buf, self.pk.length)
self.pk.length = 0
return buf
def pack_map_pairs(self, object pairs):
"""
Pack *pairs* as msgpack map type.
*pairs* should sequence of pair.
(`len(pairs)` and `for k, v in *pairs*:` should be supported.)
"""
cdef int ret = msgpack_pack_map(&self.pk, len(pairs))
if ret == 0:
for k, v in pairs:
ret = self._pack(k)
if ret != 0: break
ret = self._pack(v)
if ret != 0: break
if ret == -1:
raise MemoryError
elif ret: # should not happen
raise TypeError
buf = PyBytes_FromStringAndSize(self.pk.buf, self.pk.length)
self.pk.length = 0
return buf
def pack(object o, object stream, default=None, encoding='utf-8', unicode_errors='strict'):
"""
pack an object `o` and write it to stream)."""