объект не сериализуется JSON

у меня возникли проблемы с Mongodb и Python (фляга).

у меня есть это 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, поступающих в и из mongo. Кажется, я смог "взломать" маршрут добавления, передав запрос.json как словарь для сохранения, так что это хорошо... проблема в / show. Этот код не работает... Когда я делаю некоторые logging I get

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

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

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

4 ответов


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

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

в то время как @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.


import json
from bson import json_util

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

чтобы кодировать документы 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)

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