Question

我正在尝试使用Flask创建一个Web服务，我的Web服务需要返回属于配置文件的所有活动的json编码树结构。

我的模特：

class Activity(db.Model):
    __tablename__ = 'activity'
    id = db.Column(db.Integer, primary_key=True)
    profile_id = db.Column(db.String, db.ForeignKey('profile.id'), nullable=False, index=True)
    parent_id = db.Column(db.Integer, db.ForeignKey('activity.id'), index=True)
    name = db.Column(db.String)

    parent = db.relationship('Activiteit', remote_side=[id], backref='children')

class Profile(db.Model):
    __tablename__ = 'profile'
    id = db.Column(db.Integer, primary_key=True)
    profile_name = db.Column(db.String(64))
    active = db.Column(db.Boolean, nullable=False)

    activities = db.relationship('Activity', backref="profile", lazy='dynamic')
    employees = db.relationship('Employee', backref="profile", lazy='dynamic')

我需要以下结构：

Main activity 1
     Sub activity 1
     Sub activity 2
         Subsub activity 1
Main activity 2
etc. etc.

我尝试创建嵌套的dicts和dicts列表，但每次我都遇到困难。最终为1级元素的3级元素或仅返回2级的树。

经过一些搜索，我发现我需要创建一个带有节点的队列，这些节点带有一个名字和一个子列表，然后将这些节点添加到树中。

我使用以下函数创建了Queue：

def get(self, profile_id):
    profile = models.Profile.query.get(profile_id)
    queue = {}
    for activity in profile.activities:
        queue[activity.name]= [c.name for c in activity.children]

现在我不知道如何继续，或者我是否正确的方式。任何帮助将不胜感激

修改

根据Codegeek's回答，我创建了以下内容：

def get(self, profile_id):
    activities = models.Activity.query.filter_by(profile_id=profile_id).all()
    tree = {}
    for level1 in activities:
        if level1.parent_id is None:
            tree[level1.name] = {'id': level1.id}
            for level2 in level1.children:
                tree[level1.name][level2.name] = {'id': level2.id}
                for level3 in level2.children:
                    tree[level1.name][level2.name][level3.name] = {'id': level3.id}
    return tree

Answer 1

您是否尝试过使用flask.jsonify()？

jsonify（）将字典作为参数。您可以尝试创建嵌套字典。在你的情况下，它是3级，可能看起来像：

d= { 'Activity': { 'Subactivity': { 'Subsubactivity': 'value }}}

您尚未在创建字典的地方共享代码，但您可以执行以下操作（我当然使用通用名称）

d = {}
for activity in Activities:
    for subactivity in activity:
        for subsubactivity in subactivity:
            d['Activity']['Subactivity']['subsubactivity'] = value

然后你可以将JSON重新命名为：

return flask.jsonify(**d)

如何从SQLalchemy模型创建json树结构

修改

1 个答案: