python3实现linux下tree命令的结果

#!/usr/bin/python

# -*- coding:utf8 -*-

# Author:ZLF

import os

def func(cur_path):

'''

README:

python3的解释器环境

1, 递归循环遍历输出当前路径下的文件

2,可优化方向函数递归实现linux下tree命令的效果

3,文件夹|++表示普通文件|--表示

'''

dir_list = os.listdir(cur_path) # 列出当前路径下所有文件,形成列表

for filename in dir_list: # 循环列表里的每个文件

filename_path = cur_path + '/' + filename # 生成绝对路径

deep_len1=len(filename_path.split('/'))

# print(filename_path)

if os.path.isdir(filename_path): # 判断该文件是普通文件还是目录,如果是目录print('| ' * (deep_len1-deep_len-1) + '|++' + filename) # 格式化打印目录os.chdir(filename_path) # 进入目录

func(filename_path)#递归调用

os.chdir(os.path.dirname(filename_path))#返回当前目录

else:

print('| ' * (deep_len1-deep_len-1) + '|--' + os.path.basename(filename_path))

if __name__=='__main__':

cur_path = os.getcwd() # 当前绝对路径

print(os.path.basename(cur_path)) # 打印当前目录下的文件

deep_len = len(cur_path.split('/')) # 当前绝对路径以'/'分割的长度作为参考值func(cur_path)#函数调用

'''

优化方向,如果目录下都是文件没有文件夹则不输出最前面的"|"

'''