修改前序树遍历为Python结构

Question

你好，我正在尝试把一种叫做“修改前序树遍历”的结构用Python表示出来，然后输出成json格式。我的目标是把这个树展示在jstree上。

假设我有一个表格，像这样显示：http://imrannazar.com/Modified-Preorder-Tree-Traversal（每一行也有一个parent_id）

    Node ID   Name       Left MPTT value     Right MPTT value       ParentID
     1       (Root)             1                 16                   -1
     2       Articles           2                 11                    1
     5       Fiction            3                 8                     2
     7       Fantasy            4                 5                     5
     8       Sci-fi             6                 7                     5
     6       Reference          9                 10                    2
     3       Portfolio          12                13                    1
     4       Contact            14                15                    1

jstree所需的json格式是这样的：

    [
      {
       "data" : "Root",
       "children" : [ 
           {
              "data":"Articles",
               "children : [
                             {"data":"Fiction"},
                             {"data":"Reference"}
                            ]
           },
           {"data":"Portfolio"},
           {"data":"Contact"}]
      },
    ]

我想知道如何把上面的表格转换成Python格式，以便输出这个json。

我考虑过用一个嵌套字典来实现，像下面这样：

    class NestedDict(dict):
        def __missing__(self, key):
            return self.setdefault(key, NestedDict())

不过我不太确定需要什么样的算法。

非常感谢任何帮助。

谢谢！

Answer 1

你真的应该自己试试看，展示一下你做了什么，哪里不工作。不过我有几分钟的空闲时间，所以……

要解析这个表格，你可以使用csv模块。我就不多说了，这个你可以自己去研究。

这可能不是最好的解决方案，但这样做是可以的：

datain = (
    (1,'Root',1,16,-1),
    (2,'Articles',2,11,1),
    (5,'Fiction',3,8,2),
    (7,'Fantasy',4,5,5),
    (8,'Sci-fi',6,7,5),
    (6,'Reference',9,10,2),
    (3,'Portfolio',12,13,1),
    (4,'Contact',14,15,1),
    )

def convert_to_json(data):
    node_index = dict()
    parent_index = dict()
    for node in data:
        node_index[node[0]] = node
        parent_index.setdefault(node[4],[]).append(node)

    def process_node(index):
        result = { 'data' : node_index[index][1] }
        for node in parent_index.get(index,[]):
            result.setdefault('children',[]).append(process_node(node[0]))
        return result

    node = process_node(1)
    return [node]

返回结果：

[
    {
        'data': 'Root',
        'children': [
            {
                'data': 'Articles',
                'children': [
                    {
                        'data': 'Fiction',
                        'children': [
                            { 'data': 'Fantasy' },
                            { 'data': 'Sci-fi' }
                            ]
                    },
                    { 'data': 'Reference' }
                ]
            }, 
            { 'data': 'Portfolio' },
            { 'data': 'Contact' }
        ]
    }
]