使用Python简单模拟Linux系统的tree工具

2017-08-03 董付国 Python小屋 Python小屋

Linux系统中有个tree工具可以用比较好看的形式来显示指定文件夹的目录结构。例如下图(来自于网络):

本文代码使用Python对Linux系统的tree命令简单进行了模拟,不过还不是特别像,大家可以在此基础上进行改写或者扩展。

import os
import os.path


def tree(path, depth=0):
    if depth == 0:
        print(path)
    items = os.listdir(path)
    for item in items:
        # 输出文件夹中的文件和子文件夹名
        print('|    '*depth, end='')
        print('|----', item)
        item = os.path.join(path, item)
        if os.path.isdir(item):
            # 递归遍历子目录
            tree(item, depth+1)


tree(r'g:\test')


运行结果:


---------我是分割线---------

明天(2017年8月4日)上午9:00至下午18:00之间有一波福利(我的一个在线Python视频课,共96课,时长24小时,原价99元,明天特价9元)送给关注本公众号的朋友,需要的朋友可以发送公众号消息”公众号福利“领取。