實現(xiàn)代碼如下:
說明:現(xiàn)在代碼已實現(xiàn)du命令的 -s 統(tǒng)計功能,統(tǒng)計與系統(tǒng)du命令稍有差別,比如用ls -ld /root結(jié)果為4096,而du計算出來是1040,用當(dāng)前python計算出來也是4096。另外暫時還沒有實現(xiàn)類似于自動根據(jù)大小顯示為人性化的單位,后續(xù)補上
#!/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="統(tǒng)計指定對象的的大小總和")parser.add_option("-a", dest="autoDisplay", action="store_true", default=False, help="根據(jù)大小自動顯示 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) #對目錄本身進行大小統(tǒng)計,和du統(tǒng)計結(jié)果有點不太一樣 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)) #對目錄里面文件進行大小統(tǒng)計 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))程序執(zhí)行結(jié)果如下:

以上所述是小編給大家介紹的Python實現(xiàn)Linux中的du命令,希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回復(fù)大家的。在此也非常感謝大家對VEVB武林網(wǎng)網(wǎng)站的支持!
新聞熱點
疑難解答
圖片精選