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

杰弗里·斯洛特(Jeffrey Slort)

我正在尝试使用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.

我尝试创建嵌套字典和字典列表,但是每次遇到困难时,我都会尝试创建。级别3元素最终以级别1元素或仅返回2个级别的树的形式出现。

经过一番搜索后,我发现我需要创建一个带有包含名称和子列表的节点的队列,然后将节点添加到树中。

我使用以下功能创建了队列:

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的答案,我创建了以下内容:

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
Codegeek

您是否尝试过使用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)

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章