Python实现Linux中的du命令

时间:2021-05-22

实现代码如下:

说明:现在代码已实现du命令的 -s 统计功能,统计与系统du命令稍有差别,比如用ls -ld /root结果为4096,而du计算出来是1040,用当前python计算出来也是4096。另外暂时还没有实现类似于自动根据大小显示为人性化的单位,后续补上

#!/usr/bin/python#*-*coding:utf8*-*import sysimport osfrom optparse import OptionParser#使用选项帮助信息可以使用中文reload(sys)sys.setdefaultencoding("utf-8")#定义选项和帮助信息usage = sys.argv[0] + " [选项]... [对象]..."parser = OptionParser(usage)parser.add_option("-s", dest="sum", action="store_true", default=False, help="统计指定对象的的大小总和")parser.add_option("-a", dest="autoDisplay", action="store_true", default=False, help="根据大小自动显示 k,KB,MB,GB 等单位")options, args = parser.parse_args()#判断文件或目录是否存在def noFile(i): if not os.path.exists(i): sys.stderr.write(i + "\tis not exists\n") exit(1)sum = 0for i in args: noFile(i) if os.path.isfile(i): size = os.path.getsize(i) sum += size print("%d\t%s" %(size, i)) if os.path.isdir(i): dir = os.walk(i) for x, y, z in dir: size = os.path.getsize(x) #对目录本身进行大小统计,和du统计结果有点不太一样 sum += size if options.sum: pass else: print("%d\t%s" %(size, x)) for f in z: size = os.path.getsize(os.path.join(x, f)) #对目录里面文件进行大小统计 sum += size if options.sum: pass else: print("%d\t%s" %(size, os.path.join(x, f))) """如果加上-s选项,这里就输入总的大小""" if options.sum: print("%d\t%s" %(sum, i))

程序执行结果如下:

以上所述是小编给大家介绍的Python实现Linux中的du命令,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对网站的支持!

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章