爬虫列出文件夹内的文件夹名称

import os

def list_directories_and_save(directory, output_file):
    """
    列出指定目录下的所有子文件夹，并将文件夹名保存到文本文件中。

    参数:
    directory (str): 要列出子文件夹的目录路径。
    output_file (str): 保存文件夹名的输出文本文件路径。
    """
    try:
        # 获取指定目录下的所有子文件夹
        directories = [d for d in os.listdir(directory) if os.path.isdir(os.path.join(directory, d))]

        # 将文件夹名写入输出文件
        with open(output_file, 'w') as file:
            for d in directories:
                file.write(d + '\n')

        return "子文件夹列表生成完成。"
    except Exception as e:
        return f"发生错误: {e}"

# 示例用法
# 设置目录和输出文件路径
directory_path = '/path/to/your/directory'  # 请替换为您要列出子文件夹的目录路径
output_file_path = '/path/to/your/output_file.txt'  # 请替换为您想要保存子文件夹名的文本文件路径

# 在您的环境中运行以下代码
# result = list_directories_and_save(directory_path, output_file_path)
# print(result)