# +----------------------------------------------------------------------
# | WaitAdmin(fastapi)快速开发后台管理系统
# +----------------------------------------------------------------------
# | 欢迎阅读学习程序代码,建议反馈是我们前进的动力
# | 程序完全开源可支持商用,允许去除界面版权信息
# | gitee: https://gitee.com/wafts/waitadmin-python
# | github: https://github.com/topwait/waitadmin-python
# | 官方网站: https://www.waitadmin.cn
# | WaitAdmin团队版权所有并拥有最终解释权
# +----------------------------------------------------------------------
# | Author: WaitAdmin Team <2474369941@qq.com>
# +----------------------------------------------------------------------
from typing import TypeVar, Union, List, Dict, Any

T = TypeVar("T")


class ArrayUtil:
 """ 数组工具 """

 @staticmethod
 def list_to_tree(arr: List[Dict[str, Any]], id_: str, pid: str, child: str) -> Union[T, List[Dict[str, Any]]]:
 """
 将包含字典的列表转换为树形结构。

 Args:
 arr (List[Dict[str, Any]]): 包含字典的列表,每个字典表示树中的一个节点。
 id_ (str): 字典中作为主键的字段名。
 pid (str): 字典中表示父节点ID的字段名。
 child (str): 在树形结构中,用于存储子节点的字段名。

 Returns:
 Union[List[Dict[str, Any]], Dict[str, Any]]: 转换后的树形结构,可以是列表(根节点列表)或单个字典(单根树)。

 Author:
 zero
 """
 # 使用id_字段的值作为键,将arr中的字典映射为字典。
 id_dict_map = {item[id_]: item for item in arr}
 # 用于存放顶级节点(即没有父节点的节点)的列表。
 top_level_nodes = []

 # 遍历arr中的每个字典
 for item in arr:
 # 获取父节点ID
 parent_id = item.get(pid)
 # 尝试从映射中获取父节点
 p_node = id_dict_map.get(parent_id)

 # 如果找到父节点,则将当前节点添加到父节点的子节点列表中
 if p_node:
 # 确保child字段在p_node中存在
 if child in p_node:
 p_node[child].append(item)
 else:
 p_node[child] = [item]
 # 如果没有父节点(即顶级节点),则将其添加到top_level_nodes列表中
 else:
 top_level_nodes.append(item)

 # 如果只有一个顶级节点,并且该节点有child字段,则返回这个节点(表示单根树)
 if len(top_level_nodes) == 1 and child in top_level_nodes[0]:
 if not top_level_nodes[0]:
 return []
 return top_level_nodes[0] if isinstance(top_level_nodes[0], list) else [top_level_nodes[0]]
 # 否则返回顶级节点列表(表示多根树)
 else:
 return top_level_nodes