如何在 Pymongo 中将自定义 python 对象编码为 BSON？

要在 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 字段。

更新于： 2020 年 6 月 16 日

335 次阅读

开启您的 职业生涯

完成课程并获得认证

开始了解
广告