时间戳

首页 > 解决方案 > 烧瓶棉花糖 base_fields.Function in base_fields.Nested

python - 烧瓶棉花糖 base_fields.Function in base_fields.Nested

问题描述

我正在使用烧瓶棉花糖和棉花糖 sqlalchemy

我想拥有自己的 HATEOAS 实现：对于 n 对多关系以及链接，我想知道对象的数量

为此，我有一个具有多对多关系的常规 sqlalchemy 模型：

class ParentChild(Model):
    __tablename__ = 'parrent_child'
    parent_id =Column(Integer, ForeignKey('parent.id'), primary_key=True)
    child_id = Column(Integer, ForeignKey('child.id'), primary_key=True)

class Parent(Model):
    __tablename__ = 'parent'
    id = Column(Integer, primary_key=True)
    name = Column(String())
    children = relationship('Child', secondary='parent_child', back_populates='parents')

class Child(Model):
    __tablename__ = 'child'
    id = Column(Integer, primary_key=True)
    name = Column(String())
    parents = relationship('Parent', secondary='parent_child', back_populates='children')

使用以下棉花糖模式，我设法获得了我想要的数据：

class ParentSchema(Schema):
    class Meta:
        model = Parent
    children = URLFor('api.parents_children_by_parent_id', parent_id='<id>')
    children_count = base_fields.Function(lambda obj: len(obj.children))

回报：

{
    "id" : 42,
    "name" : "Bob",
    "children" : "/api/parents/42/children",
    "children_count" : 3
}

但是当我想封装这样的字段时遇到问题：

{
     "id": 42
     "name": "bob",
     "children": {
         "link": "/api/parents/42/children",
         "count": 3
     }
}

我尝试使用base_fields.Dict：

children = base_fields.Dict(
    link = URLFor('api.parents_children_by_parent_id', parent_id='<id>'),
    count = base_fields.Function(lambda obj: len(obj.children))
)

但我明白了 TypeError: Object of type 'Child' is not JSON serializable

我尝试了各种其他解决方案，但没有成功：flask-marshmallowHyperlinks只接受超链接字典，而不接受函数。

我认为解决方案是使用 abase_fields.Nested但它打破了URLFor无法捕获'<id>'. 我在文档中找不到解决方案。

在某些时候，很难开箱即用。我错过了什么吗？任何帮助，将不胜感激。

标签： pythonflaskflask-sqlalchemyhateoasmarshmallow

解决方案

所以我找到了一个我要发布的解决方法，但我认为它可以改进。

要使用我想要的对象覆盖该children字段，我使用base_fields.Method：

class ParentSchema(Schema):
    class Meta:
        model = Parent

    children = base_fields.Method('build_children_obj')

    def build_children_obj(self, obj):
        return {
            "count": len(obj.children),
            "link": URLFor('api.parents_children_by_parent_id', parent_id=obj.id)
        }

那时，我得到TypeError: Object of type 'URLFor' is not JSON serializable

因此，在检查了_serialize方法的来源之后，URLFor我在我的（自定义）JSONEncoder 中添加了一个检查：

if isinstance(o, URLFor):
    return str(o._serialize(None, None, o))

我终于得到了我想要的有效载荷，但我觉得它不是很干净。有任何想法吗？

编辑：经过测试，我发现len(obj.children)通过加载整个子列表来获取计数的资源非常昂贵。相反，我做db.session.query(func.count(Children.id)).filter(Children.parents.any(id=obj.id)).scalar()哪个更优化。

推荐阅读