earthjasonlin/LuckyDraw
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases 3 Wiki Activity

fix saving list without password when drawing with unsaved modified list in settings window

Browse Source
This commit is contained in:
Zichao Lin 2024-06-22 23:19:15 +08:00
parent 36bf0fcf33
commit 291db1828c
Signed by: earthjasonlin
GPG Key ID: 406D9913DE2E42FB
1 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
main.py
View File

@ -17,10 +17,10 @@ DEFAULT_MAX_ID = 45
PASSWORD = "Admin@123" PASSWORD = "Admin@123"
selected_ids = [] selected_ids = []
temp_selected_ids = []
is_running = False is_running = False
job = None job = None
# 注册表工具函数 # 注册表工具函数
def set_registry_value(value_name, value, value_type=winreg.REG_SZ): def set_registry_value(value_name, value, value_type=winreg.REG_SZ):
try: try:
@ -29,7 +29,6 @@ def set_registry_value(value_name, value, value_type=winreg.REG_SZ):
except WindowsError as e: except WindowsError as e:
print(f"Error setting registry value: {e}") print(f"Error setting registry value: {e}")
def get_registry_value(value_name, default_value): def get_registry_value(value_name, default_value):
try: try:
with winreg.OpenKey( with winreg.OpenKey(
@ -40,7 +39,6 @@ def get_registry_value(value_name, default_value):
except WindowsError: except WindowsError:
return default_value return default_value
def delete_registry_value(value_name): def delete_registry_value(value_name):
try: try:
with winreg.OpenKey( with winreg.OpenKey(
@ -50,10 +48,9 @@ def delete_registry_value(value_name):
except WindowsError as e: except WindowsError as e:
print(f"Error deleting registry value: {e}") print(f"Error deleting registry value: {e}")
# 加载已抽取的学号 # 加载已抽取的学号
def load_selected_ids(): def load_selected_ids():
global selected_ids global selected_ids, temp_selected_ids
selected_ids_str = get_registry_value("SelectedIDs", "") selected_ids_str = get_registry_value("SelectedIDs", "")
if selected_ids_str: if selected_ids_str:
selected_ids = [] selected_ids = []
@ -71,9 +68,9 @@ def load_selected_ids():
selected_ids = [] selected_ids = []
else: else:
selected_ids = [] selected_ids = []
temp_selected_ids = list(selected_ids)
return selected_ids return selected_ids
# 保存已抽取的学号 # 保存已抽取的学号
def save_selected_ids(): def save_selected_ids():
global selected_ids global selected_ids
@ -85,7 +82,6 @@ def save_selected_ids():
) )
set_registry_value("SelectedIDs", selected_ids_str) set_registry_value("SelectedIDs", selected_ids_str)
# 加载设置 # 加载设置
def load_settings(): def load_settings():
allow_repeat = ( allow_repeat = (
@ -95,14 +91,12 @@ def load_settings():
max_id = int(get_registry_value("MaxID", str(DEFAULT_MAX_ID))) max_id = int(get_registry_value("MaxID", str(DEFAULT_MAX_ID)))
return allow_repeat, min_id, max_id return allow_repeat, min_id, max_id
# 保存设置 # 保存设置
def save_settings(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("AllowRepeat", str(allow_repeat))
set_registry_value("MinID", str(min_id)) set_registry_value("MinID", str(min_id))
set_registry_value("MaxID", str(max_id)) set_registry_value("MaxID", str(max_id))
# 生成随机的学号 # 生成随机的学号
def generate_random_id(selected_ids, allow_repeat, min_id, max_id): def generate_random_id(selected_ids, allow_repeat, min_id, max_id):
if not allow_repeat: if not allow_repeat:
@ -120,20 +114,20 @@ def update_history_list():
global history_list global history_list
if history_list: if history_list:
history_list.delete(0, tk.END) history_list.delete(0, tk.END)
for id, timestamp in selected_ids: for id, timestamp in temp_selected_ids:
history_list.insert(tk.END, f"{id} ({timestamp})") history_list.insert(tk.END, f"{id} ({timestamp})")
# 每隔 5 毫秒更新一次显示的随机学号 # 每隔 5 毫秒更新一次显示的随机学号
def update_number(allow_repeat, min_id, max_id): def update_number(allow_repeat, min_id, max_id):
global job global job
if is_running: if is_running:
random_id = generate_random_id(selected_ids, allow_repeat, min_id, max_id) random_id = generate_random_id(temp_selected_ids, allow_repeat, min_id, max_id)
label.config(text=str(random_id)) label.config(text=str(random_id))
job = root.after(5, update_number, allow_repeat, min_id, max_id) job = root.after(5, update_number, allow_repeat, min_id, max_id)
# 抽取学号按钮点击事件处理函数 # 抽取学号按钮点击事件处理函数
def draw_student_id(): def draw_student_id():
global is_running, job, selected_ids global is_running, job, selected_ids, temp_selected_ids
allow_repeat, min_id, max_id = load_settings() allow_repeat, min_id, max_id = load_settings()
if is_running: if is_running:
@ -142,6 +136,7 @@ def draw_student_id():
id = label.cget("text") id = label.cget("text")
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
selected_ids.append((str(id), timestamp)) selected_ids.append((str(id), timestamp))
temp_selected_ids.append((str(id), timestamp))
save_selected_ids() save_selected_ids()
if settings_window: if settings_window:
update_history_list() update_history_list()
@ -153,7 +148,6 @@ def draw_student_id():
button.config(text="停止") button.config(text="停止")
update_number(allow_repeat, min_id, max_id) update_number(allow_repeat, min_id, max_id)
# 设置按钮点击事件处理函数 # 设置按钮点击事件处理函数
def open_settings(): def open_settings():
global settings_window global settings_window
@ -197,7 +191,7 @@ def open_settings():
new_id = int(new_id) new_id = int(new_id)
if min_id <= new_id <= max_id: if min_id <= new_id <= max_id:
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
selected_ids.append((str(new_id), timestamp)) temp_selected_ids.append((str(new_id), timestamp))
update_history_list() update_history_list()
else: else:
messagebox.showerror("错误", f"学号必须在 {min_id} 和 {max_id} 之间") messagebox.showerror("错误", f"学号必须在 {min_id} 和 {max_id} 之间")
@ -208,11 +202,11 @@ def open_settings():
selected = history_list.curselection() selected = history_list.curselection()
if selected: if selected:
index = selected[0] index = selected[0]
selected_ids.pop(index) temp_selected_ids.pop(index)
update_history_list() update_history_list()
if index < len(selected_ids): if index < len(temp_selected_ids):
history_list.select_set(index) history_list.select_set(index)
elif selected_ids: elif temp_selected_ids:
history_list.select_set(index - 1) history_list.select_set(index - 1)
else: else:
messagebox.showerror("错误", "请选择一个学号进行删除") messagebox.showerror("错误", "请选择一个学号进行删除")
@ -221,7 +215,7 @@ def open_settings():
selected = history_list.curselection() selected = history_list.curselection()
if selected: if selected:
index = selected[0] index = selected[0]
current_id, _ = selected_ids[index] current_id, _ = temp_selected_ids[index]
new_id = simpledialog.askstring( new_id = simpledialog.askstring(
"修改学号", "请输入新的学号:", initialvalue=str(current_id) "修改学号", "请输入新的学号:", initialvalue=str(current_id)
) )
@ -230,7 +224,7 @@ def open_settings():
new_id = int(new_id) new_id = int(new_id)
if min_id <= new_id <= max_id: if min_id <= new_id <= max_id:
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
selected_ids[index] = (str(new_id), timestamp) temp_selected_ids[index] = (str(new_id), timestamp)
update_history_list() update_history_list()
history_list.select_set(index) history_list.select_set(index)
else: else:
@ -241,8 +235,8 @@ def open_settings():
messagebox.showerror("错误", "请选择一个学号进行修改") messagebox.showerror("错误", "请选择一个学号进行修改")
def clear_all_ids(): def clear_all_ids():
global selected_ids global temp_selected_ids
selected_ids = [] temp_selected_ids = []
update_history_list() update_history_list()
tk.Button(settings_window, text="新增记录", font=("宋体", 10), command=add_id).grid( tk.Button(settings_window, text="新增记录", font=("宋体", 10), command=add_id).grid(
@ -278,6 +272,8 @@ def open_settings():
def save_changes(): def save_changes():
entered_password = simpledialog.askstring("密码", "请输入密码:", parent=settings_window, show="*") entered_password = simpledialog.askstring("密码", "请输入密码:", parent=settings_window, show="*")
if entered_password == PASSWORD: if entered_password == PASSWORD:
global selected_ids
selected_ids = list(temp_selected_ids)
save_settings( save_settings(
allow_repeat_var.get(), int(min_id_var.get()), int(max_id_var.get()) allow_repeat_var.get(), int(min_id_var.get()), int(max_id_var.get())
) )
@ -302,7 +298,6 @@ def open_settings():
settings_window.protocol("WM_DELETE_WINDOW", close_settings_window) settings_window.protocol("WM_DELETE_WINDOW", close_settings_window)
def close_settings_window(): def close_settings_window():
global settings_window global settings_window
settings_window.destroy() settings_window.destroy()