Python实现画目录树

陈翼轩

已于 2023-10-26 17:32:04 修改

阅读量865

点赞数

文章标签： python 开发语言

于 2022-07-13 16:29:02 首次发布

本文链接：https://blog.csdn.net/weixin_45605241/article/details/125627749

版权

该博客介绍了如何使用Python实现文件夹目录树的可视化，包括在控制台打印和写入到txt文件中。通过递归函数`print_tree`和`write_tree`，程序能够根据用户输入的文件夹路径和名称，生成类似`└─dir1 ├─file1 └─dir2 └─file2`的目录结构。

摘要由CSDN通过智能技术生成

Python实现画目录树

将任意一个文件夹的路径和文件夹名称输入，得到一颗可视化的目录树，存在一个txt文件中。要用到以下四种图像进行操作。效果如下图
│ ├ ─ └
在这里插入图片描述

代码如下

import os

def get_num(path):
    dirlist = os.listdir(path)
    j=0
    for i in dirlist:
        j+=1
    return j

def print_tree(path,last):
    num=get_num(path)
    if num!=0:
        dirlist = os.listdir(path)
        j=0
        for i in dirlist:
            for k in last:
                if k=='0':
                    print("  │",end=" ")
                else:
                    print("   ", end=" ")
            j+=1
            if j<num:
                print("  ├── ", end="")
                print(i)
                dir=path+"\\"+i
                if os.path.isdir(dir):
                    print_tree(dir,last+'0')
            else:
                print("  └── ", end="")
                print(i)
                dir = path + "\\" + i
                if os.path.isdir(dir):
                    print_tree(dir,last+'1')

def write_tree(path,last,f):
    num=get_num(path)
    if num!=0:
        dirlist = os.listdir(path)
        j=0
        for i in dirlist:
            for k in last:
                if k=='0':
                    f.write("    │")
                else:
                    f.write("    ")
            j+=1
            if j<num:
                f.write("    ├── ")
                f.write(i)
                f.write('\n')
                dir=path+"\\"+i
                if os.path.isdir(dir):
                    write_tree(dir,last+'0',f)
            else:
                f.write("    └── ")
                f.write(i)
                f.write('\n')
                dir = path + "\\" + i
                if os.path.isdir(dir):
                    write_tree(dir,last+'1',f)

if __name__=='__main__':
    path = input("请输入文件夹路径（不含名称）：")
    root = input("请输入文件夹名称：")
    if len(path)==1:
        path+=':'
    #print("  └─root")
    #print_tree('D:\\root',"1")
    f = open("tree.txt", "w", encoding="utf-8")
    f.write("    └──"+root+"\n")
    write_tree(path+"\\"+root, "1",f)
    f.close()