MongoEngineJSONEncoder. Dump the data in a BaseDocument ,QuerySet of Mongo documents to JSON

encoder.py

import datetime
import json

from bson import ObjectId
from mongoengine.base import BaseDocument
from mongoengine.queryset import QuerySet


class MongoengineEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, BaseDocument):
            return obj.to_mongo()

        if isinstance(obj, QuerySet):
            return list(obj)

        if isinstance(obj, ObjectId):
            return str(obj)

        if isinstance(obj, datetime.datetime):
            return obj.isoformat()

        return json.JSONEncoder.default(self, obj)