python脚本导出文件结构

import os

def export_directory_structure(folder_path, output_file):
    with open(output_file, 'w', encoding='utf-8') as file:
        for root, dirs, files in os.walk(folder_path):
            # 计算目录的层级，以便用缩进展示目录结构
            level = root.replace(folder_path, '').count(os.sep)
            indent = ' ' * 4 * level
            # 写入目录名称
            file.write(f'{indent}{os.path.basename(root)}/\n')
            sub_indent = ' ' * 4 * (level + 1)
            # 写入文件名称
            for f in files:
                file.write(f'{sub_indent}{f}\n')

# 使用示例
folder_path = '/path/to/your/folder'  # 替换为你想要导出的目录路径
output_file = 'directory_structure.txt'  # 导出的文本文件名称
export_directory_structure(folder_path, output_file)
print(f"目录结构已导出到 {output_file}")