自动化之python读取目录结构转换为element-plus tree结构

def get_project_tree(start_path: str, original_path: str, tree_data:list):
    child_files = os.listdir(start_path)
    for child_file in child_files:
        if child_file in ['.gitignore', '.idea', 'venv', '__pycache__']:
            continue
        filepath = os.path.join(start_path, child_file)
        # 如果是目录则继续递归处理
        if os.path.isdir(filepath):
            dir_child = {
                "label": child_file,
                "children": []
            }
            if start_path != original_path:
                # 如果不等于，则说明当前已经处于其他的子层下,根据子层层级寻找child，并将数据存入
                calcu_now = start_path.split("\\")[len(original_path.split("\\")):]
                update_dict_value_by_path(tree_data, calcu_now, dir_child)
            else:
                tree_data.append(dir_child)
            get_project_tree(filepath, original_path, tree_data)
        # 解决根目录下可能存在文件
        elif start_path == original_path:
            if not child_file.endswith(".py"):
                continue
            final_child = {
                "label": child_file
            }
            tree_data.append(final_child)
        # 解决子目录下的文件写入对应子目录下
        elif start_path != original_path:
            if not child_file.endswith(".py"):
                continue
            # 如果不等于，则说明当前已经处于其他的子层下,根据子层层级寻找child，并将数据存入
            calcu_now = start_path.split("\\")[len(original_path.split("\\")):]
            final_child = {
                "label": child_file
            }
            update_dict_value_by_path(tree_data, calcu_now, final_child)
        else:
            raise AttributeError("存在未处理的情况")
        # return tree_data


def update_dict_value_by_path(tree_data, path, final_child, tree_tier=0):
    if isinstance(path, str):
        keys = path.split('.')
    elif isinstance(path, list):
        keys = path
    else:
        raise ValueError("不支持的路径类型")
    d = tree_data
    for child in d:
        try:
            if child["label"] != path[tree_tier]:
                # 不是要找的目录，则继续循环
                continue
        except IndexError:
            # 说明是找到了最后一层，当不做处理，继续执行
            continue
        # 已经找到了对应的目录，看是不是目标层级
        if len(path) == tree_tier + 1:
            child["children"].append(final_child)
        # 说明时找到了对应的层级目录，但是没到目标层级
        tree_tier += 1
        update_dict_value_by_path(child["children"], path, final_child, tree_tier)
    return tree_data


# 使用示例
base_dir = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
tree_data = []
get_project_tree(base_dir, base_dir, tree_data)
print(tree_data)

执行结果：