Python入门之自各儿动手实现tree工具

Python入门之自己动手实现tree工具

完成目标：

使用python实现一个简单的tree工具，完成对指定目录的tree功能，通过选项可以打开目录所占磁盘大小的统计信息，通过选项可以关闭文件的打印（即仅显示文件夹）。若不指定目录，以当前工作目录进行处理。

涉及内容：

1、python基础

2、使用argparse模块处理命令行参数

3、使用os模块进行文件相关的内容处理

4、使用递归思想进行文件系统的打印

5、文件大小的格式化打印（B、KB、MB、GB、TB）

实现效果图：

存在问题：

指定路径中存在乱码或非中英文字符时报错。（测试在包含韩文文件的目录os.listdir()报错）

代码如下：

# -*- coding: utf8 -*-import osimport sysimport argparsedef get_prefix(fs): u'''根据目录标志位列表获取前置的显示标志 ''' s = '' for i in range(0,len(fs)): last = (i == len(fs)-1) if fs[i] and last: s = s + '|-' elif fs[i] and not last: s = s + '| ' elif not fs[i] and last: s = s + '.-' else: s = s + ' ' if i != len(fs)-1: s = s + ' ' else: s = s + '-' return sdef print_tree(current,fs,if_size,if_nofile): u'''深度递归指定目录并打印 ''' global d_count,f_count,size fs.append(True) dirs = os.listdir(current) for i in range(0,len(dirs)): abs_file = os.path.realpath(current+os.sep+dirs[i]) if i == len(dirs)-1: fs[len(fs)-1] = False if fs[len(fs)-1] and if_nofile: last_flag = True for j in range(i+1,len(dirs)): if os.path.isdir(os.path.realpath(current+os.sep+dirs[j])): last_flag = False break if last_flag: fs[len(fs)-1] = False if os.path.isdir(abs_file): print '%s%s' % (get_prefix(fs),dirs[i]) d_count = d_count + 1 print_tree(abs_file,fs[:],if_size,if_nofile) else: if if_size: size = size + os.path.getsize(abs_file) if not if_nofile: print '%s%s' % (get_prefix(fs),dirs[i]) f_count = f_count + 1def print_total(dir_count,file_count,if_size=False,file_size=0): u'''打印汇总信息 ''' if not if_size: print u'共有%d个文件夹,%d个文件' % (d_count,f_count) else: unit = ('B','KB','MB','GB','TB') i = len(str(size))/3 if i > 4: i = 4 unit_size = 1024 ** i f_size = round(float(size)/unit_size,2) print u' 共有%d个文件夹,%d个文件,大小 %s %s' % (d_count,f_count,f_size,unit[i]) print ''parser = argparse.ArgumentParser(description=u'以图形显示指定路径的文件夹结构')parser.add_argument('paths',metavar='path',nargs='*',help=u'指定路径')parser.add_argument('--size','-s',action='store_true',help=u'计算大小')parser.add_argument('--nofile','-f',action='store_true',help=u'忽略文件')args = parser.parse_args()f_count = 0d_count = 0size = 0if len(args.paths)==0 : print u'未指定目标路径,默认以当前路径执行...' print os.getcwd() print_tree(os.getcwd(),[],args.size==True,args.nofile==True) print_total(d_count,f_count,args.size==True,size)else: for path in args.paths: f_count = 0 d_count = 0 size = 0 if not os.path.exists(path): print u'跳过',path,u',找不到指定路径.' print '' continue path = os.path.realpath(path) print path print_tree(os.path.normpath(path),[],args.size==True,args.nofile==True) print_total(d_count,f_count,args.size==True,size)