python实现linux命令输出两列_Python实现Linux中的du命令

实现代码如下：

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

#!/usr/bin/python

#*-*coding:utf8*-*

import sys

import os

from 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",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 = 0

for 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命令，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对编程小技巧网站的支持！