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

Use cython's embedsignature directive and enhance some docstrings.

Browse source
This commit is contained in:
INADA Naoki 2011-08-26 04:45:05 +09:00
parent 5ed2288fed
commit 938a124973
1 changed files with 28 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

49
msgpack/_msgpack.pyx
View file

@ -1,4 +1,5 @@
# coding: utf-8 # coding: utf-8
#cython: embedsignature=True
from cpython cimport * from cpython cimport *
cdef extern from "Python.h": cdef extern from "Python.h":
@ -159,17 +160,13 @@ cdef class Packer(object):
def pack(object o, object stream, default=None, encoding='utf-8', unicode_errors='strict'): def pack(object o, object stream, default=None, encoding='utf-8', unicode_errors='strict'):
"""\ """
pack(object o, object stream, default=None, encoding='utf-8', unicode_errors='strict')
pack an object `o` and write it to stream).""" pack an object `o` and write it to stream)."""
packer = Packer(default=default, encoding=encoding, unicode_errors=unicode_errors) packer = Packer(default=default, encoding=encoding, unicode_errors=unicode_errors)
stream.write(packer.pack(o)) stream.write(packer.pack(o))
def packb(object o, default=None, encoding='utf-8', unicode_errors='strict'): def packb(object o, default=None, encoding='utf-8', unicode_errors='strict'):
"""\ """
packb(object o, default=None, encoding='utf-8', unicode_errors='strict')
pack o and return packed bytes.""" pack o and return packed bytes."""
packer = Packer(default=default, encoding=encoding, unicode_errors=unicode_errors) packer = Packer(default=default, encoding=encoding, unicode_errors=unicode_errors)
return packer.pack(o) return packer.pack(o)
@ -197,10 +194,7 @@ cdef extern from "unpack.h":
def unpackb(object packed, object object_hook=None, object list_hook=None, bint use_list=0, encoding=None, unicode_errors="strict"): def unpackb(object packed, object object_hook=None, object list_hook=None, bint use_list=0, encoding=None, unicode_errors="strict"):
"""\ """
unpackb(object packed, object object_hook=None, object list_hook=None,
bint use_list=0, encoding=None, unicode_errors="strict")
Unpack packed_bytes to object. Returns an unpacked object.""" Unpack packed_bytes to object. Returns an unpacked object."""
cdef template_context ctx cdef template_context ctx
cdef size_t off = 0 cdef size_t off = 0
@ -250,21 +244,36 @@ def unpackb(object packed, object object_hook=None, object list_hook=None, bint
def unpack(object stream, object object_hook=None, object list_hook=None, bint use_list=0, encoding=None, unicode_errors="strict"): def unpack(object stream, object object_hook=None, object list_hook=None, bint use_list=0, encoding=None, unicode_errors="strict"):
"""\ """
unpack(object stream, object object_hook=None, object list_hook=None, unpack an object from stream.
bint use_list=0, encoding=None, unicode_errors="strict") """
unpack an object from stream."""
return unpackb(stream.read(), use_list=use_list, return unpackb(stream.read(), use_list=use_list,
object_hook=object_hook, list_hook=list_hook, encoding=encoding, unicode_errors=unicode_errors) object_hook=object_hook, list_hook=list_hook, encoding=encoding, unicode_errors=unicode_errors)
cdef class Unpacker(object): cdef class Unpacker(object):
"""Unpacker(read_size=1024*1024) """
Streaming unpacker. Streaming unpacker.
read_size is used like file_like.read(read_size) read_size is used like file_like.read(read_size)
example: `file_like` is a file-like object having `.read(n)` method.
When `Unpacker` initialized with `file_like`, unpacker reads serialized data
from it and `.feed()` method is not usable.
`read_size` is used as `file_like.read(read_size)`. (default: 1M)
If `use_list` is true, msgpack list is deserialized to Python list.
Otherwise, it is deserialized to Python tuple. (default: False)
`object_hook` is same to simplejson. If it is not None, it should be callable
and Unpacker calls it when deserializing key-value.
`encoding` is encoding used for decoding msgpack bytes. If it is None (default),
msgpack bytes is deserialized to Python bytes.
`unicode_errors` is used for decoding bytes.
example::
unpacker = Unpacker() unpacker = Unpacker()
while 1: while 1:
buf = astream.read() buf = astream.read()
@ -292,11 +301,9 @@ cdef class Unpacker(object):
free(self.buf); free(self.buf);
self.buf = NULL; self.buf = NULL;
def __init__(self, file_like=None, Py_ssize_t read_size=0, bint use_list=0, def __init__(self, file_like=None, Py_ssize_t read_size=1024*1024, bint use_list=0,
object object_hook=None, object list_hook=None, object object_hook=None, object list_hook=None,
encoding=None, unicode_errors='strict'): encoding=None, unicode_errors='strict'):
if read_size == 0:
read_size = 1024*1024
self.use_list = use_list self.use_list = use_list
self.file_like = file_like self.file_like = file_like
if file_like: if file_like: