Python——生成项目文件树状图

                                              Python——生成项目文件树状图

 

目标：用readme文档可能会用到文件夹的目录结构，如何用python快速实现？

如果不想用代码实现，可以看看tree命令，一条命令解决问题，参考另一篇博客：tree命令的安装与使用。

# 显示文件夹树状目录
import os
import os.path

def dfs_showdir(path, depth):
    if depth == 0:
        print("root:[" + path + "]")

    for item in os.listdir(path):
        if '.git' not in item:
            print("|      " * depth + "+--" + item)
            newitem = path +'/'+ item
            if os.path.isdir(newitem):
                dfs_showdir(newitem, depth +1)


if __name__ == '__main__':
    path = ""             # 文件夹路径
    dfs_showdir(path, 0)  # 显示文件夹的树状结构

运行结果：