Mã hóa các đối tượng python tùy chỉnh như BSON với pymongo

Question

Có cách nào để cho pymongo sử dụng bộ mã hóa tùy chỉnh để chuyển đổi các đối tượng python sang BSON không?

Cụ thể là tôi cần chuyển đổi các mảng có nhiều mảng thành BSON. Tôi biết tôi có thể tự đảm bảo mỗi mảng numpy được chuyển đổi thành một mảng python bản địa trước khi gửi nó đến pymongo. Nhưng điều này là lặp đi lặp lại và dễ bị lỗi. Tôi muốn có một cách để thiết lập kết nối pymongo của tôi để làm điều này tự động.

Answer 1

Bạn cần phải viết SONManipulator. Từ docs:

Các phiên bản SONManipulator cho phép bạn chỉ định các phép biến đổi được PyMongo áp dụng tự động.

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 

sau đó thêm nó vào đối tượng cơ sở dữ liệu pymongo của bạn:

db.add_son_manipulator(Transform()) 

Lưu ý bạn không cần phải thêm lĩnh vực _type nếu bạn muốn âm thầm bỏ mảng NumPy đến một mảng python.