tree_str = '' def save_data(tree,filename='tree.txt'): with open(filename,'w',encoding='utf-8') as f: f.write(tree) def generate_tree(pathname,n=0): global tree_str if pathname.is_file(): tree_str += ' |' * n + '-' * 4 + pathname.name + '\n' elif pathname.is_dir(): tree_str += ' |' * n + '-' * 4 + \ str(pathname.relative_to(pathname.parent)) + '\\' + '\n' for cp in pathname.iterdir(): generate_tree(cp, n + 1) if __name__ == '__main__': pathname = Path("")#在这里输入目标目录 generate_tree(pathname) print(tree_str)

这段代码是一个用于生成指定目录下文件和文件夹的树形结构，并将结果保存到指定文件中的程序。具体来说，它包括了两个函数：

1. generate_tree(pathname,n=0)：这个函数接受一个Path对象和一个整数n作为参数，用于生成树形结构。当传入的pathname表示一个文件时，它会在tree_str字符串中添加一个以'|'和'-'组成的标记，表示文件在树形结构中的位置。当传入的pathname表示一个文件夹时，它会在tree_str字符串中添加一个以'|'和'-'组成的标记，并且递归调用generate_tree函数，将子文件夹和子文件全部遍历一遍。

2. save_data(tree,filename='tree.txt')：这个函数接受一个字符串tree和一个文件名filename作为参数，将字符串tree保存到指定文件中。

在主函数中，首先指定了目标目录pathname，然后调用generate_tree函数生成树形结构，并将结果保存到tree_str字符串中。最后输出tree_str字符串，可以看到目录下的文件和文件夹结构。