#!/usr/bin/env python
import sys
import os
import pwd
from argparse import ArgumentParser 

MINSIZE=0

def setup_options():
    parser=ArgumentParser(description="List file usage")
    parser.add_argument('startpath', action="append", nargs='+')

    options=parser.parse_args()
    return options

def add_recurse(db,options,start):
    for path,dirs,files in os.walk(start,followlinks=False):
        files=clean_syms(files,path)
        for f in files:
            filename=os.path.join(path,f)
            stats=os.stat(filename)
            owner=stats.st_uid
            size=stats.st_size
            if not owner in db:
                db[owner]=0
            db[owner]+=size
    return db

def clean_syms(files,path):
    nonsyms=[]
    for f in files:
        if not os.path.islink(os.path.join(path,f)):
            nonsyms.append(f)
    return nonsyms

def print_db(db):
    namedb=dict()
    for u in db:
        try:
            namedb[pwd.getpwuid(u).pw_name]=humanize_size(db[u])
        except:
            namedb[str(u)]=humanize_size(db[u])
    names=namedb.keys()
    names.sort()
    for n in names:   
        print(n+": "+namedb[n])

def humanize_size(size,precision=1):
    if size==None:
        return 'nan'
    suffixes=['B','KB','MB','GB','TB']
    suffixIndex = 0
    defPrecision=0
    while size > 1024:
        suffixIndex += 1 #increment the index of the suffix
        size = size/1024.0 #apply the division
        defPrecision=precision
    return "%.*f%s"%(defPrecision,size,suffixes[suffixIndex])

def main():
    options=setup_options();
    db=dict()
    for f in options.startpath[0]:
        db=add_recurse(db,options,f)
    print_db(db)
    sys.exit(0)

main()