sam_gui.py

import PySimpleGUI as sg
import sam_db
import serial_formatter
import sam_records_csv_reports
import sam_records_json_reports

sg.theme('Dark Blue 3')  # please make your windows colorful


def main():
    # Main menu
    primary_layout = [
        [sg.Text('Select one of the following options:')],
        [sg.Button('Add Software Vendor', size=(20, 0)),
        sg.Button('Add Serial Number', size=(20, 0)),
        sg.Button('Add Product Key', size=(20, 0))],
        [sg.Button('Update Software Vendor', size=(20, 0)),
        sg.Button('Update Serial Number', size=(20, 0)),
        sg.Button('Update Product Key', size=(20, 0))],
        [sg.Button('View Vendor List', size=(20, 0)),
        sg.Button('Get Reports', size=(20, 0)),
        sg.Button('Delete Records', size=(20, 0), visible=False)],
        [sg.HorizontalSeparator()],
        [sg.Button('Exit')]
        ]
    primary_window = sg.Window('Main Menu', primary_layout)

    while True:
        # Main menu
        event, values = primary_window.read()

        if event in ('Exit', None):
            print("Exiting Program")
            exit()

        if event == 'Add Software Vendor':
            while True:
                sv_event = (sg.popup_get_text("Enter a new software vendor: "))
                if sv_event in ('Cancel', None):
                    break
                elif sv_event.lower() in [v.lower() for v in sam_db.view_vendors()]:
                    add_sv_layout = [[sg.Text("That vendor already exists. Do you want to add more licenses for it?")],
                                     [sg.Button('Yes'), sg.Button('No')]]
                    add_sv_window = sg.Window(layout=add_sv_layout, title="Confirm Vendor Addition")
                    while True:
                        add_sv_event, add_sv_value = add_sv_window.read()
                        if add_sv_event in ('Cancel', None, 'No'):
                            add_sv_window.close()
                            break
                        if add_sv_event == 'Yes':
                            add_sv_window.close()
                            continue
                try:
                    vendor_amount = (sg.popup_get_text("How many licenses do you want to add?"))
                    if vendor_amount is None or vendor_amount == 'Exit':
                        break
                    if vendor_amount:
                        if not vendor_amount.isdigit():
                            raise TypeError(sg.popup("Please enter an integer"))
                except TypeError:
                    event = 'Add Software Vendor'
                else:
                    if vendor_amount:
                        sam_db.soft_vend_enter(sv_event, int(vendor_amount))
                        break

        def serial_number_common(enumerated_sn_list, gui_sn_list, title):
            # functionality shared by 'Add Serial Number' and 'Update Serial Number'
            sn_set_out = set()
            listbox_width = max([len(row) for row in gui_sn_list])
            add_sn_layout = [[sg.T("Pick Vendor(s):")],
                            [sg.Listbox(values=gui_sn_list, size=(listbox_width, 10), select_mode='multiple',
                                        key='SELECTION', enable_events=True, metadata=())],
                            [sg.Listbox(values=["Selected rows go here"], size=(listbox_width, 10), key='UPDATE')],
                            [sg.B("Go"), sg.B("Cancel")]]
            add_sn_window = sg.Window(layout=add_sn_layout, title=title,
                                      element_padding=((10, 10), (5, 5)), size=(None, None))
            while True:
                sn_event, sn_value = add_sn_window.read()
                if sn_event is None or sn_event == 'Cancel':
                    add_sn_window.close()
                    break
                if sn_event == 'SELECTION':  # add/remove selected to right/lower box
                    # replace existing metadata with a tuple of the indexes of highlighted items in the Listbox:
                    add_sn_window.Element('SELECTION').metadata = add_sn_window.Element('SELECTION').GetIndexes()
                    # copy the pretty display of items into a second Listbox:
                    add_sn_window.Element('UPDATE').Update(sn_value['SELECTION'])
                if sn_event == 'Go':
                    initial_sn = sg.popup_get_text("Enter Serial Number: ")
                    if initial_sn is None:
                        continue
                    for row in enumerated_sn_list:
                        # compare complete namedtuple rows to index values of selected gui rows:
                        if enumerated_sn_list.index(row) in add_sn_window.Element('SELECTION').metadata:
                            # send row to validate serial against regex, request new serial number until validated
                            if serial_formatter.sn_checker(row, initial_sn) != row:
                                try_sn = sg.popup_get_text(str(f"{initial_sn} is not a valid serial number for "
                                                                f"{row[1].Vendor}, Product Key: {row[1].Product_Key}"))
                                while serial_formatter.sn_checker(row, try_sn) != row and try_sn is not None:
                                    try_sn = sg.popup_get_text(str(f"{try_sn} is not a valid serial number for "
                                                                    f"{row[1].Vendor}, Product Key: {row[1].Product_Key}"))
                                if try_sn is not None:
                                    row_sn_mod = row[1]._replace(Serial_Number=try_sn)  # enumeration discarded
                                    sn_set_out.add(row_sn_mod)
                            else:
                                if initial_sn == '':
                                    initial_sn = None  # not allowing empty strings as sns
                                row_sn_mod = row[1]._replace(Serial_Number=initial_sn)  # enumeration discarded
                                sn_set_out.add(row_sn_mod)
                    add_sn_window.Element('UPDATE').Update(["Updated Items:"] + [f"Vendor: {row.Vendor}, Serial Number:"
                                                            f" {row.Serial_Number}, Product Key: {row.Product_Key}" for
                                                            row in sn_set_out])
                    print("sn_set_out post-mod:", sn_set_out)  # for testing purposes
                    sam_db.sn_pk_updater(sn_set_out, sam_db.sql_update_sn)
                    sn_set_out = set()  # empty the set for re-use

        def product_key_common(enumerated_pk_list, gui_pk_list, title):
            # functionality shared by 'Add Product Key' and 'Update Product Key'
            pk_set_out = set()  # collect rows with new pks in here
            listbox_width = max([len(row) for row in gui_pk_list])
            add_pk_layout = [[sg.Text("Pick Vendor(s):")],
                             [sg.Listbox(values=gui_pk_list, size=(listbox_width, 10), select_mode='multiple', key='SELECTION',
                                         enable_events=True, metadata=())],
                             [sg.Listbox(values=["Selected rows go here"], size=(listbox_width, 10), key='UPDATE')],
                             [sg.Button("Go"), sg.B("Cancel")]]
            add_pk_window = sg.Window(layout=add_pk_layout, title=title,
                                      element_padding=((10, 10), (5, 5)), size=(None, None))
            while True:
                pk_event, pk_value = add_pk_window.read()
                if pk_event is None or pk_event == 'Cancel':  # always check for closed window
                    add_pk_window.close()
                    break
                if pk_event == 'SELECTION':  # add/remove selected to right/lower box
                    # replace existing metadata with a tuple of the indexes of highlighted items in the Listbox:
                    add_pk_window.Element('SELECTION').metadata = add_pk_window.Element('SELECTION').GetIndexes()
                    # copy the pretty display of items into a second Listbox:
                    add_pk_window.Element('UPDATE').Update(pk_value['SELECTION'])
                if pk_event == 'Go':
                    initial_key = sg.popup_get_text("Enter Product Key: ")
                    if initial_key is None:
                        continue
                    for row in enumerated_pk_list:
                        # compare complete namedtuple rows to index values of selected gui rows:
                        if enumerated_pk_list.index(row) in add_pk_window.Element('SELECTION').metadata:
                            # send row to validate product key against regex, request new product key until validated
                            if serial_formatter.pk_checker(row, initial_key) != row:
                                try_key = sg.popup_get_text(str(f"{initial_key} is not a valid product key for "
                                                                f"{row[1].Vendor}, Serial Number: {row[1].Serial_Number}"))
                                while serial_formatter.pk_checker(row, try_key) != row and try_key is not None:
                                    try_key = sg.popup_get_text(str(f"{try_key} is not a valid product key for "
                                                                    f"{row[1].Vendor}, Serial Number: {row[1].Serial_Number}"))
                                if try_key is not None:
                                    row_pk_mod = row[1]._replace(Product_Key=try_key)  # enumeration discarded
                                    pk_set_out.add(row_pk_mod)
                            else:
                                if initial_key == '':
                                    initial_key = None  # not allowing empty strings as product keys
                                row_pk_mod = row[1]._replace(Product_Key=initial_key)  # enumeration discarded
                                pk_set_out.add(row_pk_mod)
                    add_pk_window.Element('UPDATE').Update(
                        ["Updated Items:"] + [f"Vendor: {row.Vendor}, Serial Number: "
                                              f"{row.Serial_Number}, Product Key: {row.Product_Key}" for
                                              row in pk_set_out])
                    print("pk_set_out post-mod:", pk_set_out)  # for testing purposes
                    sam_db.sn_pk_updater(pk_set_out, sam_db.sql_update_product_key)
                    pk_set_out = set()  # empty the set for re-use

        if event == 'Add Serial Number':
            # adds serial numbers (sn) to entries that have none
            sn_list = sam_db.view_all_none_sn_namedtuple()
            # the sn_list index is hardcoded to the items so that we can see the exact items that wil be updated:
            enumerated_sn_list = list(enumerate(sn_list))
            # make a subset of data for a pretty display:
            gui_sn_list = [f"Vendor: {row.Vendor}, Product Key: {row.Product_Key}" for row in sn_list]
            title = "Add Serial Number"
            serial_number_common(enumerated_sn_list, gui_sn_list, title)

        if event == 'Add Product Key':
            # adds product keys (pk) to entries that have none
            pk_list = sam_db.view_all_none_pk_namedtuple()  # original list
            # the pk_list index is hardcoded to the items so that we can see the exact items that wil be updated:
            enumerated_pk_list = list(enumerate(pk_list))
            # make a subset of data for a pretty display:
            gui_pk_list = [f"Vendor: {row.Vendor}, Serial Number: {row.Serial_Number}" for row in pk_list]
            title = "Add Product Key"
            product_key_common(enumerated_pk_list, gui_pk_list, title)

        if event == 'Update Software Vendor':
            # get distinct vendor names, set button size to length of longest name
            vendor_list = sam_db.view_vendors()
            vendor_buttons = [sg.Button(vendor, size=(max(len(vendor) for vendor in vendor_list), 1),)
                              for vendor in vendor_list]
            usv_layout = [[sg.Text("Select a vendor name to modify:")]]
            # append buttons to usv_layout in a set number per gui row:
            grid_width = 9
            i = 0
            j = grid_width
            while sum([len(sublists) for sublists in usv_layout])-1 < len(vendor_buttons):
                sublist = vendor_buttons[i:j]
                usv_layout.append(sublist)
                i += grid_width
                j += grid_width
            usv_window = sg.Window(layout=usv_layout, title="Update Software Vendor")
            while True:
                usv_event, usv_values = usv_window.read()
                if usv_event is None or usv_event == 'Exit':
                    usv_window.close()
                    break
                else:
                    usv_vendor = sg.popup_get_text(f"Enter the corrected name of: {usv_event}")
                    if usv_vendor is None or usv_vendor == 'Exit':
                        event = 'Update Software Vendor'
                    else:
                        sg.theme('HotDogStand')  # eye-catching warning screen
                        usv_confirm_layout = [[sg.Text(f"ARE YOU SURE YOU WANT TO CHANGE {usv_event} TO {usv_vendor}?")],
                                               [sg.Button('Yes'), sg.Button('No')]]
                        usv_confirm_window = sg.Window(layout=usv_confirm_layout, title="CAUTION")
                        while True:
                            usvc_event, usvc_values = usv_confirm_window.read()
                            if usvc_event == 'Yes':
                                sam_db.update_vendor(usv_vendor, usv_event)
                                usv_confirm_window.close()
                                sg.theme('Dark Blue 3')  # reset theme
                            else:
                                usv_confirm_window.close()
                                sg.theme('Dark Blue 3')  # reset theme
                                break

        if event == 'Update Serial Number':
            # replaces serial numbers (sn) for entries that have one
            sn_list = sam_db.view_all_sn_namedtuple()
            # the sn_list index is hardcoded to the items so that we can see the exact items that wil be updated:
            enumerated_sn_list = list(enumerate(sn_list))
            # make a subset of data for a pretty display:
            gui_sn_list = [f"Vendor: {row.Vendor}, Serial Number: {row.Serial_Number}, Product Key: {row.Product_Key}"
                           for row in sn_list]
            title = "Update Serial Number"
            serial_number_common(enumerated_sn_list, gui_sn_list, title)

        if event == 'Update Product Key':
            # replaces product keys (pk) for entries that have one
            pk_list = sam_db.view_all_pk_namedtuple()  # original list
            # the pk_list index is hardcoded to the items so that we can see the exact items that wil be updated:
            enumerated_pk_list = list(enumerate(pk_list))
            # make a subset of data for a pretty display:
            gui_pk_list = [f"Vendor: {row.Vendor}, Serial Number: {row.Serial_Number}, Product Key: {row.Product_Key}"
                           for row in pk_list]
            title = "Update Product Key"
            product_key_common(enumerated_pk_list, gui_pk_list, title)

        if event == 'View Vendor List':
            vendor_list = sam_db.view_vendors()
            listbox_width = max(26, max([len(row) for row in vendor_list]))
            listbox_height = min(20, len(vendor_list))
            vendor_query_layout = [[sg.T("All Vendors:")],
                                    [sg.Listbox(values=vendor_list, size=(listbox_width, listbox_height),
                                               enable_events=True, key='SELECTION')]]
            vendor_query_window = sg.Window(layout=vendor_query_layout, title="All Vendors")
            while True:
                vq_event, vq_value = vendor_query_window.read()
                if vq_event is None or vq_event == 'Exit':
                    vendor_query_window.close()
                    break
                else:
                    sg.Popup([f"Vendor: {row.Vendor}, Serial Number: {row.Serial_Number}, Product Key: {row.Product_Key}"
                            for row in sam_db.view_vendor_info(vq_value['SELECTION'])], title="Licenses")

        if event == 'Get Reports':
            # get distinct vendor names, set button size to length of longest name
            vendor_list = sam_db.view_vendors()
            button_width = max(len(vendor) for vendor in vendor_list)
            vendor_buttons = [sg.Button(vendor, size=(button_width, 1))
                              for vendor in vendor_list]
            report_layout = [[sg.Text("Which report would you like?")],
                             [sg.B("All Vendors")],
                             [sg.HorizontalSeparator()],
                             [sg.HorizontalSeparator()],
                            [sg.B("Cancel")]]
            # append buttons to report_layout in a set number per gui row:
            grid_width = 9
            i = 0
            j = grid_width
            while sum([len(sublists) for sublists in report_layout]) -3 < len(vendor_buttons):
                sublist = vendor_buttons[i:j]
                report_layout.insert(-2, sublist)
                i += grid_width
                j += grid_width
            report_window = sg.Window(layout=report_layout, title="Report Selector")
            while True:
                gr_event, gr_value = report_window.read()
                if gr_event is None or gr_event == 'Cancel':
                    report_window.close()
                    break
                else:
                    format_layout = [[sg.T("Select Format:")], [sg.B("CSV"), sg.B("JSON")]]
                    format_window = sg.Window(layout=format_layout, title="Select Format")
                    while True:
                        format_event, format_values = format_window.read()
                        if format_event is None:
                            format_window.close()
                            break
                        if format_event == "CSV":
                            if gr_event == "All Vendors":
                                sam_records_csv_reports.all_vendors_report()
                            else:
                                sam_records_csv_reports.one_vendor_report(gr_event)
                            format_window.close()
                            break
                        if format_event == "JSON":
                            if gr_event == "All Vendors":
                                sam_records_json_reports.all_vendors_report()
                            else:
                                sam_records_json_reports.one_vendor_report(gr_event)
                            format_window.close()
                            break

        if event == 'Delete Records':
            sg.Popup("Coming in a Future Release")


if __name__ == '__main__':
    main()