mirror of
https://github.com/msgpack/msgpack-c.git
synced 2025-12-24 06:14:33 +01:00
Ues more suitable type when packing.
This commit is contained in:
@@ -7,6 +7,8 @@ cdef extern from "Python.h":
|
|||||||
ctypedef struct PyObject
|
ctypedef struct PyObject
|
||||||
cdef object PyString_FromStringAndSize(const_char_ptr b, Py_ssize_t len)
|
cdef object PyString_FromStringAndSize(const_char_ptr b, Py_ssize_t len)
|
||||||
char* PyString_AsString(object o)
|
char* PyString_AsString(object o)
|
||||||
|
int PyMapping_Check(object o)
|
||||||
|
int PySequence_Check(object o)
|
||||||
|
|
||||||
cdef extern from "stdlib.h":
|
cdef extern from "stdlib.h":
|
||||||
void* malloc(size_t)
|
void* malloc(size_t)
|
||||||
@@ -37,7 +39,7 @@ cdef extern from "pack.h":
|
|||||||
void msgpack_pack_raw_body(msgpack_packer* pk, char* body, size_t l)
|
void msgpack_pack_raw_body(msgpack_packer* pk, char* body, size_t l)
|
||||||
|
|
||||||
|
|
||||||
cdef class Packer:
|
cdef class Packer(object):
|
||||||
"""Packer that pack data into strm.
|
"""Packer that pack data into strm.
|
||||||
|
|
||||||
strm must have `write(bytes)` method.
|
strm must have `write(bytes)` method.
|
||||||
@@ -99,7 +101,8 @@ cdef class Packer:
|
|||||||
msgpack_pack_map(&self.pk, len)
|
msgpack_pack_map(&self.pk, len)
|
||||||
|
|
||||||
cdef __pack(self, object o):
|
cdef __pack(self, object o):
|
||||||
cdef long long intval
|
cdef long long llval
|
||||||
|
cdef long longval
|
||||||
cdef double fval
|
cdef double fval
|
||||||
cdef char* rawval
|
cdef char* rawval
|
||||||
|
|
||||||
@@ -110,11 +113,11 @@ cdef class Packer:
|
|||||||
elif o is False:
|
elif o is False:
|
||||||
msgpack_pack_false(&self.pk)
|
msgpack_pack_false(&self.pk)
|
||||||
elif isinstance(o, long):
|
elif isinstance(o, long):
|
||||||
intval = o
|
llval = o
|
||||||
msgpack_pack_long_long(&self.pk, intval)
|
msgpack_pack_long_long(&self.pk, llval)
|
||||||
elif isinstance(o, int):
|
elif isinstance(o, int):
|
||||||
intval = o
|
longval = o
|
||||||
msgpack_pack_long_long(&self.pk, intval)
|
msgpack_pack_long_long(&self.pk, longval)
|
||||||
elif isinstance(o, float):
|
elif isinstance(o, float):
|
||||||
fval = o
|
fval = o
|
||||||
msgpack_pack_double(&self.pk, fval)
|
msgpack_pack_double(&self.pk, fval)
|
||||||
@@ -127,12 +130,12 @@ cdef class Packer:
|
|||||||
rawval = o
|
rawval = o
|
||||||
msgpack_pack_raw(&self.pk, len(o))
|
msgpack_pack_raw(&self.pk, len(o))
|
||||||
msgpack_pack_raw_body(&self.pk, rawval, len(o))
|
msgpack_pack_raw_body(&self.pk, rawval, len(o))
|
||||||
elif isinstance(o, dict):
|
elif PyMapping_Check(o):
|
||||||
msgpack_pack_map(&self.pk, len(o))
|
msgpack_pack_map(&self.pk, len(o))
|
||||||
for k,v in o.iteritems():
|
for k,v in o.iteritems():
|
||||||
self.pack(k)
|
self.pack(k)
|
||||||
self.pack(v)
|
self.pack(v)
|
||||||
elif isinstance(o, tuple) or isinstance(o, list):
|
elif PySequence_Check(o):
|
||||||
msgpack_pack_array(&self.pk, len(o))
|
msgpack_pack_array(&self.pk, len(o))
|
||||||
for v in o:
|
for v in o:
|
||||||
self.pack(v)
|
self.pack(v)
|
||||||
|
|||||||
Reference in New Issue
Block a user