#!/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="統(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 = 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)  #對目錄本身進行大小統(tǒng)計，和du統(tǒng)計結果有點不太一樣
      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))