Объект не является сериализуемым JSON

У меня проблемы с Mongodb и Python (Flask).

У меня есть этот файл api.py, и я хочу, чтобы все запросы и ответы были в JSON, поэтому я реализую как таковой.

#
# Imports
#

from datetime import datetime
from flask import Flask
from flask import g
from flask import jsonify
from flask import json
from flask import request
from flask import url_for
from flask import redirect
from flask import render_template
from flask import make_response
import pymongo
from pymongo import Connection
from bson import BSON
from bson import json_util

#
# App Create
#

app = Flask(__name__)
app.config.from_object(__name__)

#
# Database
#

# connect
connection = Connection()
db = connection['storage']
units = db['storage']


#
# Request Mixins
#

@app.before_request
def before_request():
    #before
    return

@app.teardown_request
def teardown_request(exception):
    #after
    return


#
# Functions
#

def isInt(n):
    try:
        num = int(n)
        return True
    except ValueError:
        return False

def isFloat(n):
    try:
        num = float(n)
        return True
    except ValueError:
        return False

def jd(obj):
    return json.dumps(obj, default=json_util.default)

def jl(obj):
    return json.loads(obj, object_hook=json_util.object_hook)

#
# Response
#

def response(data={}, code=200):
    resp = {
        "code" : code,
        "data" : data
    }
    response = make_response(jd(resp))
    response.headers['Status Code'] = resp['code']
    response.headers['Content-Type'] = "application/json"
    return response


#
# REST API calls
#

# index
@app.route('/')
def index():
    return response()

# search
@app.route('/search', methods=['POST'])
def search(): 
    return response()

# add
@app.route('/add', methods=['POST'])
def add():
    unit = request.json
    _id = units.save(unit)
    return response(_id)

# get
@app.route('/show', methods=['GET'])
def show():
    import pdb; pdb.set_trace();
    return response(db.units.find())

#
# Error handing
#

@app.errorhandler(404)
def page_not_found(error):
    return response({},404)


#
# Run it!
#

if __name__ == '__main__':
    app.debug = True
    app.run()

Проблема здесь - данные кодирования json, поступающие в монго и обратно. Кажется, я смог "взломать" маршрут добавления, передав request.json в качестве словаря для сохранения, так что это хорошо... проблема /show. Этот код не работает... Когда я делаю некоторые записи, я получаю

TypeError: <pymongo.cursor.Cursor object at 0x109bda150> is not JSON serializable

Любые идеи? Я также приветствую любые предложения по остальной части кода, но JSON убивает меня.

Спасибо заранее!

Ответ 1

Когда вы проходите db.units.find() до response, вы передаете pymongo.cursor.Cursor объект json.dumps... и json.dumps не знает, как для сериализации его в JSON. Попробуйте получить фактические объекты, итерации по курсору, чтобы получить его результаты:

[doc for doc in db.units.find()]

Ответ 2

В то время как @ErenGüven показывает вам хороший ручной подход к решению этой проблемы сериализации json, pymongo поставляется с утилитой чтобы выполнить это для вас. Я использую это в своем собственном проекте django mongodb:

import json
from bson import json_util

json_docs = []
for doc in cursor:
    json_doc = json.dumps(doc, default=json_util.default)
    json_docs.append(json_doc)

Или просто:

json_docs = [json.dumps(doc, default=json_util.default) for doc in cursor]

И снова вернуть их из json:

docs = [json.loads(j_doc, object_hook=json_util.object_hook) for j_doc in json_docs]

Вспомогательные утилиты сообщают json, как обрабатывать пользовательские объекты mongodb.

Ответ 3

import json
from bson import json_util

docs_list  = list(db.units.find())
return json.dumps(docs_list, default=json_util.default)

Ответ 4

Чтобы закодировать документы MongoDB в JSON, я использую аналогичный подход к описанному ниже, который охватывает типы bson.objectid.ObjectId и datetime.datetime.

class CustomEncoder(json.JSONEncoder):
    """A C{json.JSONEncoder} subclass to encode documents that have fields of
    type C{bson.objectid.ObjectId}, C{datetime.datetime}
    """
    def default(self, obj):
        if isinstance(obj, bson.objectid.ObjectId):
            return str(obj)
        elif isinstance(obj, datetime.datetime):
            return obj.isoformat()
        return json.JSONEncoder.default(self, obj)

enc = CustomEncoder()
enc.encode(doc)

Что касается Курсора, вам нужно выполнить итерацию и сначала получить документы.