# Directory Size
# Simple utility to calculate the size of each sub directory in a specified directory.
#
#
#
#--------------------------------------------------------------------------------------------------------

from operator import itemgetter #used for sorting nested list
from pyDSFunctions import * #import functions


SelectedPath = 'M:\\'


while True:

    #Ask user to select a Directory
    SelectedPath = MainMenu(SelectedPath)

    #Calculate size of objects in selected directory
    FolderSizes = GetDirectorySizes(SelectedPath, Verbose = True)


    #Calculate total size of directory
    TotalFolderSize = 0
    for item in FolderSizes:
        TotalFolderSize += item[1]

    #sort largest first
    FolderSizes = sorted(FolderSizes, key=itemgetter(1), reverse = True)


    #Display Results
    ClearScreen()
    print('Using: ' + SelectedPath)
    print()


    for item in FolderSizes:
        itemSize = str(format(ConvertByteSize(item[1])[0],',.2f')) + ' ' + ConvertByteSize(item[1])[1]
        itemPercent = str(format((item[1]/TotalFolderSize)*100,',.0f')) + '%'
        printline = '{0:24}' + (' ' * 6) + '{1:12}' + (' ' * 6) + '{2}'
        print(printline.format(item[0],itemSize,itemPercent))



    print ()
    print('Total Directory Size: ' + str(format(ConvertByteSize(TotalFolderSize)[0],',.2f')) + ' ' +ConvertByteSize(TotalFolderSize)[1])
    print ()

    if input('Perform new search? (no to exit): ').lower == 'no': break