Python递归生成多叉树结构之treelib

Python 多叉树结构之treelib

• 官方文档和源码
  • https://treelib.readthedocs.io/en/latest/index.html
  • https://github.com/caesar0301/treelib
• 一篇不错的博客
  • https://blog.csdn.net/weixin_43790276/article/details/108248298

The main features of treelib includes:

• Efficient operation of node searching, O(1).
• Support common tree operations like traversing, insertion, deletion, node moving, shallow/deep copying, subtree cutting etc.
• Support user-defined data payload to accelerate your model construction.
• Pretty tree showing and text/json dump for pretty show and offline analysis.
• Compatible with Python 2 and 3.

Examples

Basic Usage

from treelib import Node, Tree
tree = Tree()
tree.create_node("Harry", "harry")  # root node
tree.create_node("Jane", "jane", parent="harry")
tree.create_node("Bill", "bill", parent="harry")
tree.create_node("Diane", "diane", parent="jane")
tree.create_node("Mary", "mary", parent="diane")
tree.create_node("Mark", "mark", parent="jane")
tree.show()

Harry
├── Bill
└── Jane
    ├── Diane
    │   └── Mary
    └── Mark

detail

create_node(tag=None, identifier=None, parent=None, data=None): 创建一个节点并直接添加到树中。
tag表示节点的标签，在控制台打印树的结构时显示的就是节点的标签，可以指定值，如果不指定值则默认等于id。
identifier表示节点的id，默认会分配一个唯一的id，也可以指定一个唯一id。
这里要注意，id是唯一的，不能重复，标签是可以重复的但最好别重复。
parent表示节点的父节点，根节点可以不指定，不过，一棵树只能有一个根节点，如果树中已经有根节点了，parent还为空会报错。
data表示节点中保存的数据，可以是各种数据类型。

Actual demand

数据库表中只存着两列直接依赖关系，需求是查询某个文件的直接依赖文件和间接依赖文件，
因为文件的依赖关系不确定有几层，打算使用树形结构来描述。本着不重复造轮子的原则，
选择treelib作为base tool.

这里需要注意的是每一个Node的 identifier 必须要唯一，因为文件可能被重复引用，所以使用别的方法(uuid)
来解决这个问题。在递归下一层的时候只用把本层的uuid传下去作为其子节点的parent属性。

from treelib import Tree, Node
import uuid


def recursion_s(sql_util, table_name, file_name, tree1, uuid_p):
    query_path_depend = """SELECT """
    res_depend = sql_util.select_all_data(query_path_depend % (table_name, file_name))
    if len(res_depend) == 0:
        return None
    for df in res_depend:
        uuid_s = uuid.uuid4()
        tree1.create_node(df[0], uuid_s, parent=uuid_p, data=df[0])
        recursion_s(sql_util, table_name, df[0], tree1, uuid_s)
    return None
	

def call()
	tree1 = Tree()
    tree1.create_node(f2, 'root', data=f2)
    for df in res_depend:
        uuid_1 = uuid.uuid4()
        tree1.create_node(df[0], uuid_1, parent='root', data=df[0])
        recursion_s(sql_util, depend[0].split('.')[0], df[0], tree1, uuid_1)

最后会输出一棵多叉树，也可根据需求转成Json格式，也可写入到文件里面。