Pymongo를 사용하여 사용자 정의 Python 개체를 BSON으로 인코딩하려면 SONManipulator를 작성해야 합니다. 문서에서:
SONManipulator 인스턴스를 사용하면 PyMongo에서 자동으로 적용할 변환을 지정할 수 있습니다.
from pymongo.son_manipulator import SONManipulator class Transform(SONManipulator): def transform_incoming(self, son, collection): for (key, value) in son.items(): if isinstance(value, Custom): son[key] = encode_custom(value) elif isinstance(value, dict): # Make sure we recurse into sub-docs son[key] = self.transform_incoming(value, collection) return son def transform_outgoing(self, son, collection): for (key, value) in son.items(): if isinstance(value, dict): if "_type" in value and value["_type"] == "custom": son[key] = decode_custom(value) else: # Again, make sure to recurse into sub-docs son[key] = self.transform_outgoing(value, collection) return son
그런 다음 pymongo 데이터베이스 개체에 추가합니다. −
db.add_son_manipulator(Transform())
numpy 배열을 python 배열로 자동 캐스트하려는 경우 _type 필드를 추가할 필요가 없습니다.