使用pymongo将自定义python对象编码为BSON

有没有办法告诉pymongo使用自定义编码器将python对象转换为BSON

---

具体来说，我需要将numpy阵列转换为BSON。我知道我可以手动确保每个numpy数组在发送到pymongo之前都转换为本机python数组。但这是重复的，而且容易出错。我更希望有一种方法来设置pymongo连接以自动执行此操作。

您需要编写一个

操作器。从：

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数组，则不必添加
\u type
字段。
son\u操纵器
目前已被弃用。它们将在v4.0上删除。官方建议是在将文件传递给pymongo之前对其进行转换，如