import os.path
from os.path import isfile, isdir, join, getsize
from sys import argv, exit
import getopt

def find_big_directory(p, big_size):
    total_size = 0
    try:
        items = os.listdir(p)
    except:
        items = []
    for item in items:
        fp = join(p, item)
        if isfile(fp):
            total_size = total_size + getsize(fp)
        else:
            total_size = total_size + find_big_directory(fp, big_size)
    if total_size > big_size:
        pprint(p, total_size)
    return total_size


def find_big_file(p, big_size):
    total_size = 0
    try:
        items = os.listdir(p)
    except:
        items = []
    for item in items:
        fp = join(p, item)
        if isfile(fp):
            total_size = getsize(fp)
            if total_size > big_size:
                pprint(fp, total_size)
        else:
            find_big_file(fp, big_size)

def pprint(p, size):
    if size < 1024:
        print "[%s]: %.2f Bytes" % (p, size)
    elif size < (1024 * 1024):
        print "[%s]: %.2f Kib" % (p, float(size) / 1024)
    elif size < (1024 * 1024 * 1024):
        print "[%s]: %.2f Mib" % (p, float(size) / (1024 * 1024))
    elif size < (1024 * 1024 * 1024 * 1024):
        print "[%s]: %.2f Gib" % (p, float(size) / (1024 * 1024 * 1024))
    else:
        print "[%s]: %.2f Tib" % (p, float(size) / (1024 * 1024 * 1024 * 1024))


def usage():
    print '''
        python findbig.py [options]

        -t, --type          dir or file, find bing dir or find big file
        -d, --dir           the dir to find
        -m, --minsize       find dir or file size large than minsize
        -h, --help          show help
        -v, --version       show version
    '''
    exit(0)

if __name__ == '__main__':
    dst_dir = ""
    big_size = ""
    query_type = ""
    try:
        options, args = getopt.getopt(argv[1:], "hvt:d:m:", ["help", "type=" "dir=", "minsize="])
    except:
        usage()
    for o, v in options:
        if o in ("-h", "--help"):
            usage()
        if o in ("-v", "--version"):
            print "0.0.1 --by withrock"
            exit(0)
        if o in ("-t", "--type"):
            if v in ("dir", "file"):
                query_type = v
            else:
                usage()
        if o in ("-d", "--dir"):
            dst_dir = v
            if not isdir(dst_dir):
                print "dir invalid."
                usage()
        if o in ("-m", "--minsize"):
            try:
                big_size = long(v)
            except:
                print "minsize invalid."
                usage()
    if query_type == "" or dst_dir == "" or big_size == "":
        usage()
    if query_type == "dir":
        find_big_directory(dst_dir, big_size)
    else:
        find_big_file(dst_dir, big_size)