python - 如何在 python Flask-peewee 中反转外键

Question

我正在使用Flask-peewee有两个类别和子类别表，我想制作一个 api 来根据 JSON 中的类别列出所有子类别。 (反向外键)

因此，如果我们有 5 个主类别和 20 个子类别，我们需要在 5 个主类别下列出 20 个子类别，因此我们应该只在 JSON 中显示 5 条记录。

例如:

[{“name”: “medical”,subCategories: [{“name”:”Medical Dental Tourism”},{“name”:”another”}]},{“name”: “Restaurants”,subCategories: [{“name”:”Cafes”},{“name”:”another”}]}]

模型.py

class Category(db.Model):
    __tablename__ = 'category'
    id = CharField(primary_key=True)
    name= CharField()

    def __unicode__(self):
        return self.image

class Subcategory(db.Model):
    __tablename__ = 'subcategory'
    id = CharField(primary_key=True)
    parent_id = ForeignKeyField(db_column='parent_id', rel_model=Category )
    name= CharField()

api.py

class CategoryResource(RestResource):
    exclude = 'created_at'

class SubcategoryResource(RestResource):
    exclude = 'created_at'
    include_resources = {'parent_id': CategoryResource} #I need this to be backwards

Answer 1

您可以使用 prepare_data() 自定义 API 输出的内容，如下所示:

class CategoryResource(RestResource):
      def prepare_data(self, obj, data):
            data["subcategories"] = []
            for item in obj.subcategories:
                data["subcategories"].append({"name" : item.name})
            return data

当然，您可以自定义以上内容以满足您的需求。如果您在子类别类中定义 lated_name ，则此方法有效:

class Subcategory(db.Model):
       name = CharField()
       category = ForeignKeyField(Category, related_name = 'subcategories')

这是一个link to the example app我曾经想出这个办法；只需运行它，您就可以看到它是如何工作的。