356 lines
12 KiB
Python
356 lines
12 KiB
Python
import tkinter as tk
|
|
from tkinter import simpledialog, messagebox
|
|
import random
|
|
import winreg
|
|
from datetime import datetime
|
|
import os
|
|
import pkg_resources
|
|
import base64
|
|
|
|
# 注册表路径
|
|
REGISTRY_PATH = r"Software\StudentIDDraw"
|
|
|
|
# 默认设置
|
|
DEFAULT_ALLOW_REPEAT = False
|
|
DEFAULT_MIN_ID = 1
|
|
DEFAULT_MAX_ID = 45
|
|
PASSWORD = "admin000"
|
|
|
|
selected_ids = []
|
|
temp_selected_ids = []
|
|
is_running = False
|
|
job = None
|
|
|
|
# 注册表工具函数
|
|
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, temp_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 = []
|
|
temp_selected_ids = list(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 temp_selected_ids:
|
|
history_list.insert(tk.END, f"{id} ({timestamp})")
|
|
|
|
# 每隔 5 毫秒更新一次显示的随机学号
|
|
def update_number(allow_repeat, min_id, max_id):
|
|
global job
|
|
if is_running:
|
|
random_id = generate_random_id(temp_selected_ids, allow_repeat, min_id, max_id)
|
|
label.config(text=str(random_id))
|
|
job = root.after(5, update_number, allow_repeat, min_id, max_id)
|
|
|
|
# 抽取学号按钮点击事件处理函数
|
|
def draw_student_id():
|
|
global is_running, job, selected_ids, temp_selected_ids
|
|
allow_repeat, min_id, max_id = load_settings()
|
|
|
|
if is_running:
|
|
# 停止跳动，保存最终学号
|
|
root.after_cancel(job)
|
|
id = label.cget("text")
|
|
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
selected_ids.append((str(id), timestamp))
|
|
temp_selected_ids.append((str(id), timestamp))
|
|
save_selected_ids()
|
|
if settings_window:
|
|
update_history_list()
|
|
button.config(text="开始")
|
|
is_running = False
|
|
else:
|
|
# 开始跳动
|
|
is_running = True
|
|
button.config(text="停止")
|
|
update_number(allow_repeat, min_id, max_id)
|
|
|
|
# 设置按钮点击事件处理函数
|
|
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)
|
|
|
|
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("新增学号", "请输入新的学号:", parent=settings_window)
|
|
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")
|
|
temp_selected_ids.append((str(new_id), timestamp))
|
|
update_history_list()
|
|
else:
|
|
messagebox.showerror("错误", f"学号必须在 {min_id} 和 {max_id} 之间")
|
|
else:
|
|
messagebox.showerror("错误", "请输入有效的学号")
|
|
settings_window.focus_force()
|
|
|
|
def delete_id():
|
|
selected = history_list.curselection()
|
|
if selected:
|
|
index = selected[0]
|
|
temp_selected_ids.pop(index)
|
|
update_history_list()
|
|
if index < len(temp_selected_ids):
|
|
history_list.select_set(index)
|
|
elif temp_selected_ids:
|
|
history_list.select_set(index - 1)
|
|
else:
|
|
messagebox.showerror("错误", "请选择一个学号进行删除")
|
|
settings_window.focus_force()
|
|
|
|
def edit_id():
|
|
selected = history_list.curselection()
|
|
if selected:
|
|
index = selected[0]
|
|
current_id, _ = temp_selected_ids[index]
|
|
new_id = simpledialog.askstring(
|
|
"修改学号", "请输入新的学号:", initialvalue=str(current_id), parent=settings_window
|
|
)
|
|
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")
|
|
temp_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("错误", "请选择一个学号进行修改")
|
|
settings_window.focus_force()
|
|
|
|
def clear_all_ids():
|
|
global temp_selected_ids
|
|
temp_selected_ids = []
|
|
update_history_list()
|
|
|
|
tk.Button(settings_window, text="新增记录", font=("宋体", 10), command=add_id).grid(
|
|
row=2, column=0, padx=5, pady=5, sticky="ew"
|
|
)
|
|
tk.Button(settings_window, text="删除记录", font=("宋体", 10), command=delete_id).grid(
|
|
row=2, column=1, padx=5, pady=5, sticky="ew"
|
|
)
|
|
tk.Button(settings_window, text="修改记录", font=("宋体", 10), command=edit_id).grid(
|
|
row=3, column=0, padx=5, pady=5, sticky="ew"
|
|
)
|
|
tk.Button(
|
|
settings_window, text="清空记录", font=("宋体", 10), 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=("宋体", 10), variable=allow_repeat_var
|
|
).grid(row=5, column=0, padx=5, pady=5, sticky="w")
|
|
|
|
tk.Label(settings_window, text="学号范围:", font=("宋体", 10)).grid(
|
|
row=6, column=0, padx=5, pady=5, sticky="e"
|
|
)
|
|
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("密码", "请输入密码:", parent=settings_window, show="*")
|
|
if entered_password == PASSWORD:
|
|
global selected_ids
|
|
selected_ids = list(temp_selected_ids)
|
|
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("错误", "密码错误")
|
|
settings_window.focus_force()
|
|
|
|
tk.Button(settings_window, text="保存", font=("宋体", 10), 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
|
|
|
|
def get_icon_data():
|
|
icon_path = pkg_resources.resource_filename(__name__, 'icon.png')
|
|
with open(icon_path, 'rb') as icon_file:
|
|
return base64.b64encode(icon_file.read())
|
|
|
|
# 初始化全局变量
|
|
selected_ids = load_selected_ids()
|
|
settings_window = None
|
|
history_list = None
|
|
|
|
icon_data = get_icon_data()
|
|
|
|
# 创建主窗口
|
|
root = tk.Tk()
|
|
root.title("随机抽号机")
|
|
root.attributes("-topmost", 1)
|
|
icon_image = tk.PhotoImage(data=icon_data)
|
|
root.iconphoto(True, icon_image)
|
|
|
|
# 创建标签，用于显示抽取的学号
|
|
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="LuckyDraw v1.2 2024/06/22\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()
|