【mongo】mongo數據轉json時特殊類型處理

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)