mongomongo数据转json时特殊类型处理

Posted 2020-08-15 匡子语

mongo数据库中的有些数据类型是无法用json序列化的，比如ObjectId或者datetime.datetime类型。

可以通过json.JSONEncoder来处理

import json
import pymongo
from bson import ObjectId
import datetime

class JSONEncoder(json.JSONEncoder):
    ‘‘‘处理ObjectId,该类型无法转为json‘‘‘
    def default(self, o):
        if isinstance(o, ObjectId):
            return str(o)
        if isinstance(o, datetime.datetime):
            return datetime.datetime.strftime(o,‘%Y-%m-%d %H:%M:%S‘)
        return json.JSONEncoder.default(self, o)


if __name__ == "__main__":
    mongodb = ...
    conn = mongodb["dbname"]
    coll = conn["tbname"]
    cols = []
    for col in coll.find():
        col = JSONEncoder().encode(col)
        cols.append(col)
    data = json.dumps(cols)