代码
import os
import argparsedef get_directory_tree(root_dir, show_hidden=False):"""递归获取目录树结构参数:root_dir: 根目录路径show_hidden: 是否显示隐藏目录（以.开头的目录）返回:目录树结构的嵌套字典"""# 检查目录是否存在if not os.path.isdir(root_dir):raise ValueError(f"目录不存在: {root_dir}")tree = {"name": os.path.basename(root_dir), "children": []}try:# 获取目录下的所有条目entries = os.listdir(root_dir)for entry in entries:entry_path = os.path.join(root_dir, entry)# 跳过文件，只处理目录if not os.path.isdir(entry_path):continue# 如果不显示隐藏目录，且目录以.开头，则跳过if not show_hidden and entry.startswith('.'):continue# 递归处理子目录tree["children"].append(get_directory_tree(entry_path, show_hidden))except PermissionError:# 没有权限访问的目录，标记为[权限不足]tree["name"] += " [权限不足]"except Exception as e:# 其他错误tree["name"] += f" [错误: {str(e)}]"return treedef print_tree(tree, indent="", last=True):"""打印目录树结构参数:tree: 由get_directory_tree返回的目录树字典indent: 缩进字符串last: 是否为最后一个子节点"""# 确定当前行的前缀prefix = "└── " if last else "├── "print(indent + prefix + tree["name"])# 计算下一级的缩进if last:new_indent = indent + " "else:new_indent = indent + "│ "# 递归打印子目录children = tree["children"]for i, child in enumerate(children):print_tree(child, new_indent, i == len(children) - 1)def main():# 解析命令行参数parser = argparse.ArgumentParser(description='显示目录的树形结构')parser.add_argument('directory', nargs='?', default=fr'H:\2025年10月13日——数据集处理', help='要显示的目录路径（默认当前目录）')parser.add_argument('-a', '--all', action='store_true', help='显示隐藏目录')args = parser.parse_args()try:# 获取绝对路径root_dir = os.path.abspath(args.directory)print(f"目录树结构: {root_dir}")print("-" * (len(root_dir) + 5))# 生成并打印目录树tree = get_directory_tree(root_dir, args.all)print_tree(tree)except Exception as e:print(f"错误: {e}")if __name__ == "__main__":main()print(f"finished")
