首页 > 解决方案 > 如何在 Python 中为递归对象创建 Mongo 文档模式?

问题描述

自引用 Python 对象 MongoDB Schema

我想达到什么目的?

我正在尝试创建的程序处理描述杂货产品类别的数据,例如:牛奶、牛肉、玉米、当地水果等。我计划将此数据存储在MongoDB实例中,因此创建一个模式会很好。

我的问题的原因是类别有子类别(乳制品:牛奶,奶酪,酸奶;肉类:牛肉,猪肉等),导致递归对象

单独的 Python

我查看了 StackOverflow 如何在 Python 中处理此类结构。

class Category():
    def _init_(self, name, amount, subcategories = ()):
        self.name = name
        self.amount = amount
        for subcategory in subcategories:
            assert isinstance(subcategory, Category)
        self.subcategories = subcategories

这似乎是对的......

架构部分

...但直到我尝试将其绑定到模式中:

class Category(mongoengine.EmbeddedDocument):
    def _init_(self, name, amount, subcategories = ()):
        self.name = name
        self.amount = amount
        for subcategory in subcategories:
            assert isinstance(subcategory, Category)
        self.subcategories = subcategories


    name = mongoengine.StringField(required=True)
    amount = mongoengine.FloatField(required=True)
    subcategories = mongoengine.EmbeddedDocumentListField(Category, default=None)
    meta = {
        'db_alias': 'core',
        'collection': 'category',
        'indexes': [
            'name',
            'amount',
            'subcategories',
        ],
        'ordering': ['-name']
    }

运行此代码会导致错误:

subbcategories = mongoengine.EmbeddedDocumentListField(Category, default=None) NameError: name 'Category' is not defined

我尝试引用父类别,而不是嵌入子类别:

parent_category = mongoengine.ObjectIdField(Category, default=None)

但我仍然收到完全相同的错误消息:

parent_category = mongoengine.ObjectIdField(Category, default=None)NameError: name 'Category' is not defined

问题本身

我应该如何为这样的递归对象定义模式?可能吗?

标签: pythonmongodbnosqlpymongomongoengine

解决方案


您可以尝试使用 str而不是在此行'Category'中键入:Category

subcategories = mongoengine.EmbeddedDocumentListField('Category', default=None)

EmbeddedDocumentListField用于EmbeddedDocument初始化。

EmbeddedDocument接受字符串。


推荐阅读