Create Table Rosters From Seating Chart.py

# -*- coding: utf-8 -*-
"""
Created on Sat Feb  9 12:13:07 2019

@author: Josh
"""
  
import datetime
import tkinter
from tkinter.filedialog import askopenfilename
import pickle
import xlrd
from utils import writeTables


def formatNameWithoutFamilyName(guest):
    if '-' in guest:
        guest = guest.split('-')[-1].strip()
    return guest    


## read guest data from pickle, but read seating chart from excel because it might be changed manually
def readSeatingChart():
    # Make a top-level instance and hide since it is ugly and big.
    root = tkinter.Tk()
    root.withdraw()

    # Make it almost invisible - no decorations, 0 size, top left corner.
    root.overrideredirect(True)
    root.geometry('0x0+0+0')

    # Show window again and lift it to top so it can get focus,
    # otherwise dialogs will end up behind the terminal.
    root.deiconify()
    root.lift()
    root.focus_force()

    file_path = askopenfilename(parent=root)  # Or some other dialog

    wb = xlrd.open_workbook(file_path)
    xl_sheet = wb.sheet_by_index(0)
    print('readSeatingChart: processing', xl_sheet.name)

    # Extract column headings from the first row
    column_headings = [xl_sheet.cell(0, col).value for col in range(xl_sheet.ncols)]

    # Create tables List of sets of guests from excel file
    tablesDict = {}
    for row_idx in range(1, xl_sheet.nrows):  # Iterate through rows, excluding the header
        tableNum = xl_sheet.cell(row_idx, 0).value
        guestName = xl_sheet.cell(row_idx, 1).value.strip()

        if tableNum not in tablesDict:
            tablesDict[tableNum] = set()

        tablesDict[tableNum].add(guestName)

    tablesList = []
    for tableNum in sorted(tablesDict):
        tablesList.append(tablesDict[tableNum])

    return tablesList, column_headings


def ask_for_column(column_headings):
    # Ask the user to select a column heading
    print("Available Column Headings:")
    for index, heading in enumerate(column_headings, 1):
        print(f"{index}. {heading}")

    selected_index = input("Enter the number corresponding to the column heading you want to select: ")

    try:
        selected_index = int(selected_index)
        if 1 <= selected_index <= len(column_headings):
            selected_heading = column_headings[selected_index - 1]
            print(f"You selected: {selected_heading}")
        else:
            print("Invalid input. Please enter a valid number.")
    except ValueError:
        print("Invalid input. Please enter a number.")
    return selected_index


if __name__ == "__main__":    
    timestamp=datetime.datetime.now().strftime('%Y-%m-%d_%H%M%S')
    with open('guest_data.p', 'rb') as f:    
        guests, antiRequests, emails = pickle.load(f)    

    tables, headings = readSeatingChart()

    # ask_for_column(headings)

    writeTables(tables, emails, timestamp, True, extraColumn=headings)
    writeTables(tables, emails, timestamp, False, extraColumn=headings)