utilities.py

from pyPreservica import *

def folder_desc_size(window, mline, alt_background, init_color, update_color, summary_color, username, password, tenancy, server, twofactorcb, twofactorkey, folder_report):
    mline.update('[START] GENERATING STORAGE SIZE REPORT FOR FOLDER\n', append=True, text_color_for_value=init_color)
    window.refresh()
    if twofactorcb == True:
        client = EntityAPI(username=username, password=password, tenant=tenancy, server=server, two_fa_secret_key=twofactorkey)
    else:
        client = EntityAPI(username=username, password=password, tenant=tenancy, server=server)
    folder_target = client.folder(folder_report)
    folder_size = 0
    total_assets = 0
    total_files = 0
    for asset in filter(only_assets, client.all_descendants(folder_target.reference)):
        total_assets += 1
        for representation in client.representations(asset):
            for content_object in client.content_objects(representation):
                for generation in client.generations(content_object):
                    for bitstream in generation.bitstreams:
                        total_files += 1
                        folder_size += bitstream.length
                        if (total_files % 2) == 0:
                            mline.update('[UPDATE] filename: {} | file size: {}\n'.format(bitstream.filename, bitstream.length), append=True, text_color_for_value=update_color)
                            window.refresh()
                        else:
                            mline.update('[UPDATE] filename: {} | file size: {}\n'.format(bitstream.filename, bitstream.length), append=True, text_color_for_value=update_color, background_color_for_value=alt_background)
                            window.refresh()
    folder_gb = round(folder_size / (1024 * 1024 * 1024), 2)
    folder_tb = round(folder_size / (1024 * 1024 * 1024 * 1024), 2)
    mline.update('''[SUMMARY] Storage Report for Folder
Title: {title}
Ref ID: {ref_id}
Bytes: {bytes}
GB: {gb}
TB: {tb}
Assets: {num_asset}
Files: {num_files}\n'''.format(title=folder_target.title, ref_id=folder_target.reference, bytes=folder_size, gb=folder_gb, tb=folder_tb, num_asset=total_assets, num_files=total_files), append=True, text_color_for_value=summary_color)
    window.refresh()