實現代碼如下:
說明:現在代碼已實現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",
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命令,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回復大家的。在此也非常感謝大家對腳本之家網站的支持!
更多文章、技術交流、商務合作、聯系博主
微信掃碼或搜索:z360901061
微信掃一掃加我為好友
QQ號聯系: 360901061
您的支持是博主寫作最大的動力,如果您喜歡我的文章,感覺我的文章對您有幫助,請用微信掃描下面二維碼支持博主2元、5元、10元、20元等您想捐的金額吧,狠狠點擊下面給點支持吧,站長非常感激您!手機微信長按不能支付解決辦法:請將微信支付二維碼保存到相冊,切換到微信,然后點擊微信右上角掃一掃功能,選擇支付二維碼完成支付。
【本文對您有幫助就好】元

