Mongoengine自定义查询集
Mongoengine自定义查询集
提问于 2015-06-17 20:01:14
在从ObjectId数据库查询时，我试图将ISODate和MongoDB转换为字符串表示。
def mongo_to_dict(obj, exclude_fields):
return_data = []
if obj is None:
return None
if isinstance(obj, Document):
return_data.append(("_id", str(obj.id)))
for field_name in obj._fields:
if field_name in exclude_fields:
continue
if field_name in ("id",):
continue
data = obj._data[field_name]
if isinstance(obj._fields[field_name], ListField):
return_data.append((field_name, list_field_to_dict(data)))
elif isinstance(obj._fields[field_name], EmbeddedDocumentField):
return_data.append((field_name, mongo_to_dict(data, [])))
elif isinstance(obj._fields[field_name], DictField):
return_data.append((field_name, data))
else:
return_data.append((field_name, mongo_to_python_type(obj._fields[field_name], data)))
return dict(return_data)
def list_field_to_dict(list_field):
return_data = []
for item in list_field:
if isinstance(item, EmbeddedDocument):
return_data.append(mongo_to_dict(item, []))
else:
return_data.append(mongo_to_python_type(item, item))
return return_data
def mongo_to_python_type(field, data):
if isinstance(field, DateTimeField):
return time.mktime(data.timetuple()) * 1000
elif isinstance(field, ComplexDateTimeField):
return field.to_python(data).isoformat()
elif isinstance(field, StringField):
return str(data)
elif isinstance(field, FloatField):
return float(data)
elif isinstance(field, IntField):
return int(data)
elif isinstance(field, BooleanField):
return bool(data)
elif isinstance(field, ObjectIdField):
return str(data)
elif isinstance(field, DecimalField):
return data
else:
return str(data)
class Portfolio(Document):
meta = {'collection': 'Portfolios'}
PortfolioName = StringField()
LastUpdateDate = DateTimeField(default=datetime.datetime.now())
RecentActivity = ListField(default=[])
def to_dict(self):
return mongo_to_dict(self, [])现在，当我创建这样一个Portfolio对象时
a = Portfolio(PortfolioName='BB Visa').save()当我试图得到像a.to_dict()这样的对象的a.to_dict()约束时，它工作得很好。
{'_id': '5581cf9129e457241a32e8f7', 'PortfolioName': 'BB Visa', 'RecentActivity': [], 'LastUpdateDate': 1434570641000.0}但问题是我希望to_dict()在类级别上操作，而不是在对象级别上操作。
所以当我试图定义这样的自定义查询集时
class Portfolio(Document):
meta = {'collection': 'Portfolios'}
PortfolioName = StringField()
LastUpdateDate = DateTimeField(default=datetime.datetime.now())
RecentActivity = ListField(default=[])
@queryset_manager
def to_dict(self, queryset):
return mongo_to_dict(self, [])现在，运行以下命令Portfolio.to_dict()将生成如下所示的错误(最近一次调用是最后一次)：
File "/home/ajay/PycharmProjects/solveit/test.py", line 102, in <module>
print(Portfolio.to_dict())
File "/home/ajay/.pyenv/versions/3.4.3/lib/python3.4/site-packages/mongoengine/queryset/manager.py", line 43, in __get__
queryset = self.get_queryset(owner, queryset)
File "/home/ajay/PycharmProjects/solveit/test.py", line 83, in to_dict
return mongo_to_dict(self, [])
File "/home/ajay/PycharmProjects/solveit/test.py", line 28, in mongo_to_dict
data = obj._data[field_name]
TypeError: 'member_descriptor' object is not subscriptable我理解QuerySet管理器将类传递到函数中，这就是错误的原因。如何解决这个问题。
Stack Overflow用户
发布于 2015-06-18 20:32:27
实际上，我不确定按照文档的说法，把这个定义为文档原因是正确的。
mongoengine.queryset.queryset_manager(func)装饰器，允许您在文档类上定义自定义QuerySet管理器。管理器必须是接受文档类作为其第一个参数的函数，而接受QuerySet作为其第二个参数的函数。方法函数应该返回一个QuerySet，可能与传入的相同，但以某种方式修改了。
我建议添加使用queryset的一些单独的utils函数：
def convert_queryset_to_list_of_dicts(queryset):
return [mongo_to_dict(obj) for obj in queryset]或者，如果您仍然希望将它作为@queryset_manager，那么：
...
@queryset_manager
def to_dict(self, queryset):
return [mongo_to_dict(obj) for obj in queryset]
...此外，还有一个mongoengine to_mongo的现成方法。您可以这样做：[obj.to_mongo() for obj in queryset]。不适用吗？
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/30901274
复制相关文章
相似问题
腾讯云开发者
