。

本文将介绍如何使用FastAPI封装一个通用的树形结构生成函数#xff0c;支持动态选择字段#xff0c…使用FastAPI封装树形结构生成函数高效处理层级数据

在Web开发中树形结构是一种常见的数据组织形式常用于菜单、分类、组织结构等场景。

本文将介绍如何使用FastAPI封装一个通用的树形结构生成函数支持动态选择字段并递归构建树形数据。

需求分析

我们需要实现一个函数能够根据给定的模型和父节点ID递归生成树形结构。

同时支持动态选择需要返回的字段以满足不同场景的需求。

实现思路

递归查询从根节点pid0开始逐层查询子节点。

动态字段选择通过fields参数指定需要返回的字段。

树形结构构建将查询到的节点数据组织成树形结构。

代码实现

fields.CharField(max_length255)pid

父节点ID#

model.filter(pidpid).values(*fields)tree

[]for

childrentree.append(node_data)return

tree#

register_tortoise(app,db_urlsqlite://:memory:,modules{models:

[__main__]},generate_schemasTrue,

)if

Category是一个简单的分类模型包含id、name和pid字段。

pid表示父节点ID根节点的pid为0。

create_tree函数

通过model.filter(pidpid).values(*fields)查询当前层级的节点。

使用递归调用create_tree获取子节点并将结果添加到children字段中。

返回完整的树形结构。

FastAPI路由

/tree接口调用create_tree函数返回生成的树形结构。

数据库初始化

使用register_tortoise初始化数据库并自动生成表结构。

使用示例

启动FastAPI服务后访问/tree接口将返回如下格式的树形结构

{tree:

:propsdefaultPropsdefault-expand-all

/tree总结

通过封装create_tree函数我们可以轻松生成树形结构并支持动态选择字段。

这种方法适用于各种层级数据的场景如菜单、分类、组织结构等。

结合FastAPI和Tortoise

希望本文对你有所帮助如果有任何问题或建议欢迎在评论区交流讨论。