pymongo提供一些常用的数据类型，如：数据、字符串、日期等。如果感觉还不能满足需求，那么还可以自定义数据类型。 首先定义一个类： class custom(object): def __init__(self, x): self.__x = x def x(self): return self.__x 要将自定义类型的数据存入数
pymongo提供一些常用的数据类型，如：数据、字符串、日期等。如果感觉还不能满足需求，那么还可以自定义数据类型。
首先定义一个类：
class custom(object): def __init__(self, x): self.__x = x def x(self): return self.__x
要将自定义类型的数据存入数据库中需要先进行编码；将数据从数据库读取出来后又需要再解码。
手动编码/解码我们可以定义两个方法，在插入和查询数据时进行手动的编码、解码。
def encode_custom(custom): return {_type: custom, x: custom.x()}def decode_custom(document): assert document[_type] == custom return custom(document[x])print(db.test.insert({custom: encode_custom(custom(5))}))print(db.test.find_one()['custom'])
自动编码/解码手动地进行编码虽然可行，但是还是不太方便。我们还可以使用 sonmanipulator 进行自动编码。
from pymongo.son_manipulator import sonmanipulatorclass 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 sondb.add_son_manipulator(transform())print(db.test.insert({custom: custom(5)}))print(db.test.find_one())
二进制编码我们也可以将其编码成二进制进行存储。
from bson.binary import binarydef to_binary(custom): return binary(str(custom.x()), 128)def from_binary(binary): return custom(int(binary))class transformtobinary(sonmanipulator): def transform_incoming(self, son, collection): for (key, value) in son.items(): if isinstance(value, custom): son[key] = to_binary(value) elif isinstance(value, dict): son[key] = self.transform_incoming(value, collection) return son def transform_outgoing(self, son, collection): for (key, value) in son.items(): if isinstance(value, binary) and value.subtype == 128: son[key] = from_binary(value) elif isinstance(value, dict): son[key] = self.transform_outgoing(value, collection) return sondb.add_son_manipulator(transformtobinary())print(db.test.insert({custom: custom(5)}))print(db.test.find_one())
原文地址：pymongo教程(3)——自定义数据类型, 感谢原作者分享。