uhhhhh / main.py
BarBar288's picture
Create main.py
7d89bd4 verified
import tkinter as tk
from tkinter import filedialog, messagebox
from random import shuffle, choice
class WheelOfNames:
def __init__(self):
self.root = tk.Tk()
self.root.title("Wheel of Names")
self.root.geometry("600x400")
self.names = []
# Create the menu bar
self.menu_bar = tk.Menu(self.root)
self.root.config(menu=self.menu_bar)
# File menu
self.file_menu = tk.Menu(self.menu_bar, tearoff=0)
self.menu_bar.add_cascade(label="File", menu=self.file_menu)
self.file_menu.add_command(label="Load Names from File", command=self.upload_file)
self.file_menu.add_command(label="Save Names to File", command=self.save_file)
self.file_menu.add_separator()
self.file_menu.add_command(label="Clear Names", command=self.clear_names)
self.file_menu.add_separator()
self.file_menu.add_command(label="Exit", command=self.root.quit)
self.name_entry = tk.Entry(self.root, width=50)
self.name_entry.pack(pady=10)
self.add_button = tk.Button(self.root, text="Add Name", command=self.add_name)
self.add_button.pack()
self.name_list_text = tk.Text(self.root, height=10, width=60)
self.name_list_text.pack(pady=10)
self.group_size_var = tk.StringVar()
self.group_size_var.trace_add("write", self.update_sort_button_text)
self.group_size_entry = tk.Entry(self.root, width=10, textvariable=self.group_size_var)
self.group_size_entry.pack(side=tk.LEFT, padx=10, pady=10)
self.sort_button = tk.Button(self.root, text="Sort into Groups", command=self.sort_into_groups)
self.sort_button.pack(side=tk.LEFT, padx=10, pady=10)
self.groups_text = tk.Text(self.root, height=10, width=60)
self.groups_text.pack(pady=10)
self.spin_button = tk.Button(self.root, text="Spin", command=self.spin_wheel)
self.spin_button.pack()
self.result_label = tk.Label(self.root, text="", font=("Arial", 24))
self.result_label.pack(pady=20)
def add_name(self):
name = self.name_entry.get().strip()
if name:
self.names.append(name)
self.update_name_list_text()
self.name_entry.delete(0, tk.END)
def upload_file(self):
file_path = filedialog.askopenfilename(filetypes=[("Text Files", "*.txt")])
if file_path:
with open(file_path, 'r') as file:
names = file.read().splitlines()
self.names.extend(names)
self.update_name_list_text()
def save_file(self):
file_path = filedialog.asksaveasfilename(defaultextension=".txt", filetypes=[("Text Files", "*.txt")])
if file_path:
with open(file_path, 'w') as file:
for name in self.names:
file.write(name + "\n")
def clear_names(self):
self.names = []
self.update_name_list_text()
def update_name_list_text(self):
self.name_list_text.delete(1.0, tk.END)
for name in self.names:
self.name_list_text.insert(tk.END, name + "\n")
def update_sort_button_text(self, *args):
group_size = self.group_size_var.get().strip()
if group_size.isdigit() and int(group_size) > 0:
self.sort_button.config(text=f"Sort into Groups of {group_size}")
else:
self.sort_button.config(text="Sort into Groups")
def sort_into_groups(self):
group_size = self.group_size_var.get().strip()
if not group_size.isdigit():
messagebox.showerror("Error", "Please enter a valid group size.")
return
group_size = int(group_size)
if group_size <= 0:
messagebox.showerror("Error", "Group size must be greater than zero.")
return
if not self.names:
messagebox.showerror("Error", "Please add or upload names first.")
return
shuffle(self.names)
groups = []
while self.names:
group = self.names[:group_size]
groups.append(group)
self.names = self.names[group_size:]
self.groups_text.delete(1.0, tk.END)
for i, group in enumerate(groups, 1):
self.groups_text.insert(tk.END, f"Group {i}: {', '.join(group)}\n")
def spin_wheel(self):
if not self.names:
self.result_label.config(text="Please add or upload names first")
return
winning_name = choice(self.names)
self.result_label.config(text=winning_name)
def run(self):
self.root.mainloop()
if __name__ == "__main__":
wheel = WheelOfNames()
wheel.run()