【python实用小脚本-84】从HR到技术博主：用Python绘制文件夹树状图，让复杂结构一目了然（建议收藏）

本文链接：https://blog.youkuaiyun.com/kylner/article/details/148255093

场景故事

在人力资源管理工作中，我常常需要整理和归档大量的文件，包括员工档案、培训资料、绩效记录等。这些文件通常被存储在复杂的文件夹结构中，查找特定文件时，就像在迷宫中寻找出口，费时又费力。直到有一天，我发现了一个痛点：手动浏览文件夹结构不仅效率低下，还容易遗漏重要文件。

于是，我决定用Python来解决这个问题，编写了一个能够生成文件夹树状图的脚本。通过输入目标文件夹路径，脚本可以自动遍历文件夹结构，并以树状图的形式清晰地展示出来。这个过程虽然需要一些技术知识，但当我看到脚本成功运行并呈现出清晰的文件夹结构时，所有的努力都变得值得。现在，我不仅能快速找到所需文件，还能更好地管理文件夹结构，提升工作效率。

代码核心价值解析

核心代码展示

以下是一个Python脚本，用于生成文件夹的树状图结构。它利用了os库来操作文件系统，并通过递归的方式遍历文件夹结构。

import os
import argparse

# 生成树状图的模式字符串
def get_pattern(level: int) -> str:
    spaces = 2 * (level + 1)  # 计算缩进空格数量
    separators_length = spaces // 2  # 计算分隔符数量
    pattern = ''  # 初始化模式字符串

    for i in range(separators_length):
        pattern += '  |'  # 添加分隔符
    pattern += '__'  # 添加结尾符号
    return pattern

# 打印文件夹结构
def print_structure(path: str, level: int) -> None:
    if level == 0:  # 如果是根目录，直接打印路径
        print(path)
    else:
        if '/' in path:  # 如果路径中包含斜杠，只打印最后一部分
            print(f'{get_pattern(level - 1)}{path.split("/")[-1]}/')
        else:
            print(f'{get_pattern(level)}{path}/')  # 打印目录名

    contents = os.listdir(path)  # 获取当前目录下的内容

    for content in contents:
        full_path = os.path.join(path, content)  # 拼接完整路径
        if os.path.isfile(full_path):  # 如果是文件，直接打印文件名
            print(f'{get_pattern(level)}{content}')
        else:  # 如果是目录，递归调用
            if args.spacing:  # 如果开启了间距选项，打印空行
                print(get_pattern(level)[:-2])
            print_structure(full_path, level + 1)

# 主函数
def main(args: argparse.Namespace) -> None:
    dir_path = args.directory  # 获取目标目录
    if not os.path.isdir(dir_path):  # 检查路径是否有效
        print('ERROR: can not find the path specified!')
        exit(0)
    print_structure(dir_path, 0)  # 打印目录结构

if __name__ == '__main__':
    parser = argparse.ArgumentParser()  # 创建解析器
    parser.add_argument('-d', '--directory', metavar='', help='目标目录路径 [默认为当前目录]', type=str, default='./')
    parser.add_argument('-s', '--spacing', action='store_true', help='在文件夹之间添加垂直间距')
    args = parser.parse_args()  # 解析参数
    main(args)  # 调用主函数

这段代码的核心功能是通过递归遍历文件夹结构，并以树状图的形式打印出来。get_pattern()函数用于生成树状图的模式字符串，print_structure()函数负责递归遍历和打印文件夹内容。

代码执行流程图

核心代码价值分析

价值分析("文件夹树状图生成")

✅ 三维价值评估

时间收益：节省15分钟/次 → 年省75小时
误差消除：避免手动查找文件时的遗漏
扩展潜力：改造为图形化界面工具仅需增加少量代码

✅ HR专业视角
该脚本实质是“文件管理”的技术映射，如：

自动化处理 ≈ 文件归档与检索
异常检测 ≈ 文件完整性检查
日志记录 ≈ 文件操作追溯

关键技术解剖台

os库的跨界解读

▍HR眼中的技术价值

在人力资源管理中，文件管理是一个重要的环节。良好的文件管理系统可以帮助HR快速找到所需文件，提高工作效率。而os库作为Python的一个标准库，能够方便地操作文件系统，就像为HR提供了一个强大的文件管理工具，帮助他们更好地管理文件和资料。

▍工程师的实现逻辑

# 核心代码
import os
import argparse

def get_pattern(level: int) -> str:
    spaces = 2 * (level + 1)
    separators_length = spaces // 2
    pattern = ''
    for i in range(separators_length):
        pattern += '  |'
    pattern += '__'
    return pattern

def print_structure(path: str, level: int) -> None:
    if level == 0:
        print(path)
    else:
        if '/' in path:
            print(f'{get_pattern(level - 1)}{path.split("/")[-1]}/')
        else:
            print(f'{get_pattern(level)}{path}/')

    contents = os.listdir(path)
    for content in contents:
        full_path = os.path.join(path, content)
        if os.path.isfile(full_path):
            print(f'{get_pattern(level)}{content}')
        else:
            if args.spacing:
                print(get_pattern(level)[:-2])
            print_structure(full_path, level + 1)

def main(args: argparse.Namespace) -> None:
    dir_path = args.directory
    if not os.path.isdir(dir_path):
        print('ERROR: can not find the path specified!')
        exit(0)
    print_structure(dir_path, 0)

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-d', '--directory', metavar='', help='目标目录路径 [默认为当前目录]', type=str, default='./')
    parser.add_argument('-s', '--spacing', action='store_true', help='在文件夹之间添加垂直间距')
    args = parser.parse_args()
    main(args)

技术三棱镜

原理类比：os.listdir()类似于人力资源管理中的“文件检索”，通过列出目录内容帮助我们找到所需文件。
参数黑盒：args.directory参数相当于HR的“目标文件夹”，需要根据实际情况进行配置。
避坑指南：在使用os库时，忘记检查路径是否存在可能导致程序崩溃，就像在人力资源管理中，没有做好文件备份可能导致重要资料丢失。

▍复杂度可视化

扩展应用场景

场景迁移实验室

生成扩展案例("文件夹树状图生成")

✅ 场景拓展矩阵

案例1：文件夹树状图生成 → 图形化界面版本改造指南

# 关键参数替换公式
import tkinter as tk
from tkinter import filedialog

def select_directory():
    return filedialog.askdirectory()

▶️ 改造收益：降低使用门槛，适合非技术用户

案例2：文件夹树状图生成 + 文件搜索功能跨界融合

# 组合技实现方案
import re
def search_files(path, keyword):
    for content in os.listdir(path):
        full_path = os.path.join(path, content)
        if re.search(keyword, content):
            print(full_path)
        if os.path.isdir(full_path):
            search_files(full_path, keyword)

▶️ 创新价值：帮助用户快速定位包含特定关键词的文件

总结

本文介绍了一个Python脚本，用于生成文件夹的树状图结构。它利用了os库来操作文件系统，并通过递归的方式清晰地展示文件夹结构。这个脚本不仅能够帮助我们快速了解文件夹的组织方式，还能通过扩展功能满足更多个性化需求。它是一个很好的实践项目，适合有一定Python基础的读者学习和应用。同时，它也展示了技术如何帮助我们更好地管理文件，提升工作效率。

源码获取

完整代码已开源，包含详细的注释文档：
🔗 [GitCode仓库] https://gitcode.com/laonong-1024/python-automation-scripts
📥 [备用下载] https://pan.quark.cn/s/654cf649e5a6 提取码：f5VG