python - pymongo を使用してカスタム python オブジェクトを BSON としてエンコードする

Question

カスタムエンコーダーを使用してPythonオブジェクトをBSONに変換するようにpymongoに指示する方法はありますか?

具体的には、numpy 配列を BSON に変換する必要があります。すべての numpy 配列が pymongo に送信される前に、ネイティブの python 配列に変換されることを手動で確認できることはわかっています。しかし、これは反復的でエラーが発生しやすいものです。これを自動的に行うようにpymongo接続をセットアップする方法があればいいのにと思います。

Answer 1

を書く必要があります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())

_typenumpy 配列を python 配列に暗黙的にキャストする場合は、フィールドを追加する必要がないことに注意してください。