实现代码如下:
说明:现在代码已实现du命令的 -s 统计功能,统计与系统du命令稍有差别,比如用ls -ld /root结果为4096,而du计算出来是1040,用当前python计算出来也是4096。另外暂时还没有实现类似于自动根据大小显示为人性化的单位,后续补上
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
|
#!/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" , 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 = 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命令,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对服务器之家网站的支持!