Преобразовать объект в MongoDB вернуться в словарь
Я использую рамки бутылки вместе с mongoengine. У меня есть модель заказов :
class OrderDetail(Option):
orderDetailsQty = FloatField()
def to_dict(self):
return mongo_to_dict_helper(self)
class Order(Document):
userName = StringField(required=True)
orderDate = DateTimeField()
orderStatus = ListField(EmbeddedDocumentField(Status))
orderDetails = ListField(EmbeddedDocumentField(OrderDetail))
orderComments = ListField(EmbeddedDocumentField(Comment))
isActive = BooleanField()
def to_dict(self):
orderObj = mongo_to_dict_helper(self)
orderDetailList = []
for orderDetail in orderObj["orderDetails"]:
orderDetailList.append(orderDetail.__dict__)
orderObj["OrderDetails"] = orderDetailList
return (self)
когда MongoDB запрашивается, я получаю объект, который затем преобразуется в dict, используя следующую функцию:
def mongo_to_dict_helper(obj):
return_data = []
for field_name in obj._fields:
if field_name in ("id",):
continue
data = obj._data[field_name]
if isinstance(obj._fields[field_name], StringField):
return_data.append((field_name, str(data)))
elif isinstance(obj._fields[field_name], FloatField):
return_data.append((field_name, float(data)))
elif isinstance(obj._fields[field_name], IntField):
return_data.append((field_name, int(data)))
elif isinstance(obj._fields[field_name], ListField):
return_data.append((field_name, int(data)))
else:
# You can define your logic for returning elements
pass
return dict(return_data)
Я нашел эту функцию после долгого поиска в интернете. Позже выяснилось, что эта функция также терпит неудачу при определении члена как ListField(EmbeddedDocumentField(obj)).
Я также попытался написать условие для ловли конкретный случай EmbeddedDocumentField:
elif isinstance(obj._fields[field_name], EmbeddedDocumentField):
return_data.append(mongo_to_dict_helper(data))
но это тоже не помогло.
у кого-нибудь есть решение для этой проблемы ?
4 ответов
расширение ответов @alexvassel и @z0r, вызывая .to_mongo()
преобразует объект в сын экземпляр. Как только вы его получите, вы можете назвать его .to_dict()
метод преобразования его в словарь.
например... (qset
- это queryset, который возвращается из mongoengine, например,Posts.objects.all()
).
sons = [ob.to_mongo() for ob in qset]
for son in sons:
print str(son.to_dict())
объединение всех остальных ответов,
import json
dict = {'data':[json.loads(ob.to_json()) for ob in qset]}