Get Python File Size in Readable Format

pythonFileSize.py

def size(path, kilo=1000):
    total_size = 0
    for dirpath, dirnames, filenames in os.walk(path):
        for i in filenames:
            f = os.path.join(dirpath, i)
            total_size += os.path.getsize(f)
    if total_size < kilo:
        return str(total_size)
    elif total_size < kilo ** 2:
        return str(int(total_size / kilo)) + "KB " + str(int(total_size % kilo)) + "B"
    elif total_size < kilo ** 3:
        return (
            str(int(total_size / kilo ** 2))
            + "MB "
            + str(int((total_size % kilo ** 2) / kilo))
            + "KB "
            + str(int(total_size % kilo))
            + "B"
        )
    else:
        return (
            str(int(total_size / kilo ** 3))
            + "GB "
            + str(int((total_size % kilo ** 3) / kilo ** 2))
            + "MB "
            + str(int((total_size % kilo ** 2) / kilo))
            + "KB "
            + str(int(total_size % kilo))
            + "B"
        )