LuckyDraw/main.py
2024-06-07 12:25:41 +08:00

336 lines
11 KiB
Python

import tkinter as tk
from tkinter import simpledialog, messagebox
import random
import winreg
from datetime import datetime
# 注册表路径
REGISTRY_PATH = r"Software\StudentIDDraw"
# 默认设置
DEFAULT_ALLOW_REPEAT = False
DEFAULT_MIN_ID = 1
DEFAULT_MAX_ID = 45
PASSWORD = "admin123"
selected_ids = []
# 注册表工具函数
def set_registry_value(value_name, value, value_type=winreg.REG_SZ):
try:
with winreg.CreateKey(winreg.HKEY_CURRENT_USER, REGISTRY_PATH) as reg_key:
winreg.SetValueEx(reg_key, value_name, 0, value_type, value)
except WindowsError as e:
print(f"Error setting registry value: {e}")
def get_registry_value(value_name, default_value):
try:
with winreg.OpenKey(
winreg.HKEY_CURRENT_USER, REGISTRY_PATH, 0, winreg.KEY_READ
) as reg_key:
value, _ = winreg.QueryValueEx(reg_key, value_name)
return value
except WindowsError:
return default_value
def delete_registry_value(value_name):
try:
with winreg.OpenKey(
winreg.HKEY_CURRENT_USER, REGISTRY_PATH, 0, winreg.KEY_WRITE
) as reg_key:
winreg.DeleteValue(reg_key, value_name)
except WindowsError as e:
print(f"Error deleting registry value: {e}")
# 加载已抽取的学号
def load_selected_ids():
global selected_ids
selected_ids_str = get_registry_value("SelectedIDs", "")
if selected_ids_str:
selected_ids = []
for item in selected_ids_str.split(","):
parts = item.split("_")
if len(parts) == 2:
id, timestamp = parts
timestamp = datetime.fromtimestamp(int(timestamp)).strftime(
"%Y-%m-%d %H:%M:%S"
)
selected_ids.append((id, timestamp))
else:
print(f"Error parsing item: {item}, resetting to NULL")
delete_registry_value("SelectedIDs")
selected_ids = []
else:
selected_ids = []
return selected_ids
# 保存已抽取的学号
def save_selected_ids():
global selected_ids
selected_ids_str = ",".join(
[
f"{id}_{int(datetime.strptime(timestamp, '%Y-%m-%d %H:%M:%S').timestamp())}"
for id, timestamp in selected_ids
]
)
set_registry_value("SelectedIDs", selected_ids_str)
# 加载设置
def load_settings():
allow_repeat = (
get_registry_value("AllowRepeat", str(DEFAULT_ALLOW_REPEAT)) == "True"
)
min_id = int(get_registry_value("MinID", str(DEFAULT_MIN_ID)))
max_id = int(get_registry_value("MaxID", str(DEFAULT_MAX_ID)))
return allow_repeat, min_id, max_id
# 保存设置
def save_settings(allow_repeat, min_id, max_id):
set_registry_value("AllowRepeat", str(allow_repeat))
set_registry_value("MinID", str(min_id))
set_registry_value("MaxID", str(max_id))
# 生成随机的学号
def generate_random_id(selected_ids, allow_repeat, min_id, max_id):
if not allow_repeat:
all_ids = set(range(min_id, max_id + 1))
selected_id_numbers = [int(id) for id, _ in selected_ids]
remaining_ids = list(all_ids - set(selected_id_numbers))
if not remaining_ids:
return None # 如果所有学号都已抽取完毕,则返回 None
return random.choice(remaining_ids)
else:
return random.randint(min_id, max_id)
# 更新历史记录列表
def update_history_list():
global history_list
if history_list:
history_list.delete(0, tk.END)
for id, timestamp in selected_ids:
history_list.insert(tk.END, f"{id} ({timestamp})")
# 抽取学号按钮点击事件处理函数
def draw_student_id():
global selected_ids
load_selected_ids()
allow_repeat, min_id, max_id = load_settings()
id = generate_random_id(selected_ids, allow_repeat, min_id, max_id)
if id is not None:
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
label.config(text=str(id), font=("宋体", 50))
selected_ids.append((str(id), timestamp))
save_selected_ids()
if settings_window:
update_history_list()
else:
label.config(text="已抽完所有学号", font=("宋体", 30))
# 设置按钮点击事件处理函数
def open_settings():
global settings_window
global history_list
if settings_window:
settings_window.lift()
return
settings_window = tk.Toplevel(root)
settings_window.title("设置")
allow_repeat, min_id, max_id = load_settings()
# 使用 grid 管理器来创建一个自适应大小的界面
settings_window.grid_columnconfigure(1, weight=1)
settings_window.grid_rowconfigure(4, weight=1)
tk.Label(settings_window, text="历史记录:", font=("宋体", 12)).grid(
row=0, column=0, padx=5, pady=5, sticky="w"
)
load_selected_ids()
history_frame = tk.Frame(settings_window)
history_frame.grid(row=1, column=0, columnspan=2, padx=5, pady=5, sticky="nsew")
history_frame.grid_columnconfigure(0, weight=1)
history_frame.grid_rowconfigure(0, weight=1)
history_list = tk.Listbox(history_frame, selectmode=tk.SINGLE)
history_list.grid(row=0, column=0, sticky="nsew")
scrollbar = tk.Scrollbar(
history_frame, orient="vertical", command=history_list.yview
)
scrollbar.grid(row=0, column=1, sticky="ns")
history_list.config(yscrollcommand=scrollbar.set)
update_history_list()
def add_id():
new_id = simpledialog.askstring("新增学号", "请输入新的学号:")
if new_id:
if new_id.isdigit():
new_id = int(new_id)
if min_id <= new_id <= max_id:
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
selected_ids.append((str(new_id), timestamp))
update_history_list()
else:
messagebox.showerror("错误", f"学号必须在 {min_id}{max_id} 之间")
else:
messagebox.showerror("错误", "请输入有效的学号")
def delete_id():
selected = history_list.curselection()
if selected:
index = selected[0]
selected_ids.pop(index)
update_history_list()
if index < len(selected_ids):
history_list.select_set(index)
elif selected_ids:
history_list.select_set(index - 1)
else:
messagebox.showerror("错误", "请选择一个学号进行删除")
def edit_id():
selected = history_list.curselection()
if selected:
index = selected[0]
current_id, _ = selected_ids[index]
new_id = simpledialog.askstring(
"修改学号", "请输入新的学号:", initialvalue=str(current_id)
)
if new_id:
if new_id.isdigit():
new_id = int(new_id)
if min_id <= new_id <= max_id:
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
selected_ids[index] = (str(new_id), timestamp)
update_history_list()
history_list.select_set(index)
else:
messagebox.showerror("错误", f"学号必须在 {min_id}{max_id} 之间")
else:
messagebox.showerror("错误", "请输入有效的学号")
else:
messagebox.showerror("错误", "请选择一个学号进行修改")
def clear_all_ids():
global selected_ids
selected_ids = []
update_history_list()
tk.Button(settings_window, text="新增记录", font=("宋体", 12), command=add_id).grid(
row=2, column=0, padx=5, pady=5, sticky="ew"
)
tk.Button(settings_window, text="删除记录", font=("宋体", 12), command=delete_id).grid(
row=2, column=1, padx=5, pady=5, sticky="ew"
)
tk.Button(settings_window, text="修改记录", font=("宋体", 12), command=edit_id).grid(
row=3, column=0, padx=5, pady=5, sticky="ew"
)
tk.Button(
settings_window, text="清空记录", font=("宋体", 12), command=clear_all_ids
).grid(row=3, column=1, padx=5, pady=5, sticky="ew")
allow_repeat_var = tk.BooleanVar(value=allow_repeat)
tk.Checkbutton(
settings_window, text="允许重复抽号", font=("宋体", 12), variable=allow_repeat_var
).grid(row=5, column=0, padx=5, pady=5, sticky="w")
tk.Label(settings_window, text="学号范围:", font=("宋体", 12)).grid(
row=6, column=0, padx=5, pady=5, sticky="w"
)
min_id_var = tk.StringVar(value=min_id)
max_id_var = tk.StringVar(value=max_id)
tk.Entry(settings_window, textvariable=min_id_var).grid(
row=6, column=1, padx=5, pady=5, sticky="ew"
)
tk.Entry(settings_window, textvariable=max_id_var).grid(
row=7, column=1, padx=5, pady=5, sticky="ew"
)
def save_changes():
entered_password = simpledialog.askstring("密码", "请输入密码:", show="*")
if entered_password == PASSWORD:
save_settings(
allow_repeat_var.get(), int(min_id_var.get()), int(max_id_var.get())
)
save_selected_ids()
messagebox.showinfo("提示", "设置已保存")
close_settings_window()
else:
messagebox.showerror("错误", "密码错误")
tk.Button(settings_window, text="保存", font=("宋体", 12), command=save_changes).grid(
row=8, column=0, columnspan=2, padx=5, pady=20, sticky="ew"
)
# 自动调整窗口大小
settings_window.attributes("-topmost", 1)
settings_window.update_idletasks()
settings_window.minsize(
settings_window.winfo_reqwidth(), settings_window.winfo_reqheight()
)
settings_window.resizable(False, False) # 让设置窗口不可缩放
settings_window.protocol("WM_DELETE_WINDOW", close_settings_window)
def close_settings_window():
global settings_window
settings_window.destroy()
settings_window = None
# 初始化全局变量
selected_ids = load_selected_ids()
settings_window = None
history_list = None
# 创建主窗口
root = tk.Tk()
root.title("随机抽号机")
root.attributes("-topmost", 1) # 置顶显示
# 创建标签,用于显示抽取的学号
label = tk.Label(root, text="", font=("宋体", 50))
label.pack(pady=20, padx=20)
# 创建抽取学号按钮
button = tk.Button(root, text="抽取学号", font=("宋体", 12), command=draw_student_id)
button.pack(pady=10)
# 创建设置按钮
settings_button = tk.Button(root, text="设置", font=("宋体", 12), command=open_settings)
settings_button.pack(pady=10)
# 创建多行标签,用于显示信息,内容居中显示
info_label = tk.Label(
root,
text="v1.0 2024/06/07\nCopyright 2024 earthjasonlin\n保留所有权利。\n本程序仅作为交流学习使用",
font=("宋体", 10),
justify="center",
anchor="center",
)
info_label.pack(pady=10, padx=20, fill="x")
# 自动调整窗口大小
root.update_idletasks()
root.minsize(root.winfo_reqwidth(), root.winfo_reqheight())
root.resizable(False, False)
root.mainloop()