static PyObject *
pyobj2pystring(PyObject *pyjson)
{
rapidjson::Document doc;
pyobj2doc(pyjson, doc);
rapidjson::StringBuffer buffer;
rapidjson::Writer<rapidjson::StringBuffer> writer(buffer);
doc.Accept(writer);
std::string s = buffer.GetString();
return PyString_FromStringAndSize(s.data(), s.length());
}
static PyObject *
pyobj2pystring(PyObject *pyjson)
{
rapidjson::Document doc;
if (false == pyobj2doc(pyjson, doc)) {
return NULL;
}
rapidjson::StringBuffer buffer;
rapidjson::Writer<rapidjson::StringBuffer, rapidjson::Document::EncodingType, rapidjson::ASCII<> > writer(buffer);
doc.Accept(writer);
std::string s = buffer.GetString();
return PyString_FromStringAndSize(s.data(), s.length());
}
static inline bool
pyobj2doc_pair(PyObject *key, PyObject *value, rapidjson::Document& doc)
{
const char *key_string;
PyObject *utf8_item = NULL;
PyObject *pyobj = NULL;
#ifdef PY3
if (!PyUnicode_Check(key)) {
pyobj = PyObject_Str(key);
if (pyobj == NULL) {
PyErr_SetString(PyExc_TypeError, "not support key type");
return false;
}
utf8_item = PyUnicode_AsUTF8String(pyobj);
} else {
utf8_item = PyUnicode_AsUTF8String(key);
}
key_string = PyBytes_AsString(utf8_item);
#else
if (PyString_Check(key)) {
key_string = PyString_AsString(key);
} else if (PyUnicode_Check(key)) {
utf8_item = PyUnicode_AsUTF8String(key);
key_string = PyBytes_AsString(utf8_item);
} else {
PyObject *pyobj = PyObject_Str(key);
if (pyobj == NULL) {
PyErr_SetString(PyExc_TypeError, "not support key type");
return false;
}
Py_DECREF(key);
key_string = PyString_AsString(pyobj);
}
#endif
rapidjson::Value s;
s.SetString(key_string, doc.GetAllocator());
Py_XDECREF(pyobj);
Py_XDECREF(utf8_item);
rapidjson::Value _v;
if (false == pyobj2doc(value, _v, doc)) {
return false;
}
doc.AddMember(s, _v, doc.GetAllocator());
return true;
}
static inline void
pyobj2doc_pair(PyObject *key, PyObject *value, rapidjson::Document& doc)
{
const char *key_string;
#ifdef PY3
PyObject *utf8_item;
utf8_item = PyUnicode_AsUTF8String(key);
key_string = PyBytes_AsString(utf8_item);
#else
key_string = PyString_AsString(key);
#endif
rapidjson::Value s;
s.SetString(key_string, doc.GetAllocator());
rapidjson::Value _v;
pyobj2doc(value, _v, doc);
doc.AddMember(s, _v, doc.GetAllocator());
}
static inline bool
pyobj2doc_pair(PyObject *key, PyObject *value,
rapidjson::Value& doc, rapidjson::Document& root)
{
const char *key_string;
#ifdef PY3
PyObject *utf8_item;
utf8_item = PyUnicode_AsUTF8String(key);
key_string = PyBytes_AsString(utf8_item);
#else
key_string = PyString_AsString(key);
#endif
rapidjson::Value s;
s.SetString(key_string, root.GetAllocator());
rapidjson::Value _v;
if (false == pyobj2doc(value, _v, root)) {
return false;
}
doc.AddMember(s, _v, root.GetAllocator());
return true;
}
static bool
pyobj2doc(PyObject *object, rapidjson::Document& doc)
{
if (PyBool_Check(object)) {
if (Py_True == object) {
doc.SetBool(true);
}
else {
doc.SetBool(false);
}
}
else if (Py_None == object) {
doc.SetNull();
}
else if (PyFloat_Check(object)) {
doc.SetDouble(PyFloat_AsDouble(object));
}
else if (PyInt_Check(object)) {
doc.SetInt64(PyLong_AsLong(object));
}
else if (PyString_Check(object)) {
doc.SetString(PyString_AsString(object), PyString_GET_SIZE(object));
}
else if (PyUnicode_Check(object)) {
PyObject *utf8_item;
utf8_item = PyUnicode_AsUTF8String(object);
if (!utf8_item) {
PyErr_SetString(PyExc_RuntimeError, "codec error.");
return false;
}
#ifdef PY3
doc.SetString(PyBytes_AsString(utf8_item), PyBytes_GET_SIZE(utf8_item), doc.GetAllocator());
#else
doc.SetString(PyString_AsString(utf8_item), PyString_GET_SIZE(utf8_item), doc.GetAllocator());
#endif
Py_XDECREF(utf8_item);
}
else if (PyTuple_Check(object)) {
int len = PyTuple_Size(object), i;
doc.SetArray();
rapidjson::Value _v;
for (i = 0; i < len; ++i) {
PyObject *elm = PyTuple_GetItem(object, i);
if (false == pyobj2doc(elm, _v, doc)) {
return false;
}
doc.PushBack(_v, doc.GetAllocator());
}
}
else if (PyList_Check(object)) {
int len = PyList_Size(object), i;
doc.SetArray();
rapidjson::Value _v;
for (i = 0; i < len; ++i) {
PyObject *elm = PyList_GetItem(object, i);
if (false == pyobj2doc(elm, _v, doc)) {
return false;
}
doc.PushBack(_v, doc.GetAllocator());
}
}
else if (PyDict_Check(object)) {
doc.SetObject();
PyObject *key, *value;
Py_ssize_t pos = 0;
while (PyDict_Next(object, &pos, &key, &value)) {
if (false == pyobj2doc_pair(key, value, doc)) {
return false;
}
}
}
else {
PyErr_SetString(PyExc_RuntimeError, "invalid python object");
return false;
}
return true;
}
static void
pyobj2doc(PyObject *object, rapidjson::Document& doc)
{
if (PyBool_Check(object)) {
if (Py_True == object) {
doc.SetBool(true);
}
else {
doc.SetBool(false);
}
}
else if (Py_None == object) {
doc.SetNull();
}
else if (PyFloat_Check(object)) {
doc.SetDouble(PyFloat_AsDouble(object));
}
else if (PyInt_Check(object)) {
doc.SetInt(PyLong_AsLong(object));
}
else if (PyString_Check(object)) {
doc.SetString(PyString_AsString(object), PyString_GET_SIZE(object));
}
else if (PyUnicode_Check(object)) {
#ifdef PY3
PyObject *utf8_item;
utf8_item = PyUnicode_AsUTF8String(object);
if (!utf8_item) {
// TODO: error handling
printf("error\n");
}
doc.SetString(PyBytes_AsString(utf8_item),
PyBytes_Size(utf8_item), doc.GetAllocator());
Py_XDECREF(utf8_item);
#else
doc.SetString(PyBytes_AsString(object), PyBytes_GET_SIZE(object));
#endif
}
else if (PyTuple_Check(object)) {
int len = PyTuple_Size(object), i;
doc.SetArray();
rapidjson::Value _v;
for (i = 0; i < len; ++i) {
PyObject *elm = PyList_GetItem(object, i);
pyobj2doc(elm, _v, doc);
doc.PushBack(_v, doc.GetAllocator());
}
}
else if (PyList_Check(object)) {
int len = PyList_Size(object), i;
doc.SetArray();
rapidjson::Value _v;
for (i = 0; i < len; ++i) {
PyObject *elm = PyList_GetItem(object, i);
pyobj2doc(elm, _v, doc);
doc.PushBack(_v, doc.GetAllocator());
}
}
else if (PyDict_Check(object)) {
doc.SetObject();
PyObject *key, *value;
Py_ssize_t pos = 0;
while (PyDict_Next(object, &pos, &key, &value)) {
pyobj2doc_pair(key, value, doc);
}
}
else {
// TODO: error handle
}
}
