functions

import os
import string

def get_drives():
    """
    Gets all drives on the computer
    :return:
    """
    return [c.upper()+":\\" for c in string.ascii_lowercase if os.path.isdir(c+':\\')]

def dir_getsize(directory):
    """
    Returns the size of all files within a directory (in bytes)
    :param directory:
    :return:
    """
    total_size = 0

    for file in (x for x in os.listdir(directory) if os.path.isfile(directory + "\\" + x)):
        total_size += os.path.getsize(directory + "\\" + file)
    return total_size


def folder_size(dict, folder):
    """
    Returns the the total size, in bytes, of directory including each of it's sub-directories
    :param dict:
    :param folder:
    :return:
    """

    # # Debug code ---------
    # print("DEBUG: Folder is", folder)
    # print("DEBUG: dict contains:")
    # for key in dict:
    #     print ("DEBUG:", key, "=", dict[key])
    # print()
    # print("DEBUG:", [value for key, value in dict.items() if folder in key])
    # #---------------------

    total_size = 0

    for size in [value for key, value in dict.items() if folder + "\\" in key + "\\"]:
        total_size += size

    return total_size


def formatbytes(number):
    """
    Takes a number if bytes and returns a string formatted with Bytes, KB, MB, GB
    :param number:
    :return:
    """
    if number < 1024:
        return '{:,.1f}'.format(number) + " Bytes"
    elif number < 1024**2:
        return '{:,.1f}'.format(number/1024) + " KB"
    elif number <= 1024**3:
        return '{:,.1f}'.format(number/1024**2) + " MB"
    elif number > 1024**3:
        return '{:,.1f}'.format(number/1024**3) + " GB"
    else:
        return str(number)

def return_dirs(path):
    """
    Returns a list of directories
    :param path:
    :return:
    """
    dirs = []
    for dir in os.listdir(path):
        if os.path.isdir(path + "\\" + dir):
            dirs.append(dir)
    return dirs

def print_dirs():
    """
    Prints sub directories to screen with index
    :return:
    """
    print()
    print("Current directory:", os.getcwd())
    print()
    for i, dir in enumerate(return_dirs(os.getcwd())):
        print(str(i) + ")", dir)
    print()