initialize_plus.pyw

import json,copy
from tkinter import *
from tkinter import messagebox
from type_defs import UserData, Setting, Course, CourseData
import ensure_package_exist
ensure_package_exist.ensure_package_exist()

def save(user_data:UserData,setting:Setting):
    with open("user_data.json",mode="w") as file:
        json.dump(user_data.to_dict(),file,indent=2)
    with open("setting.json",mode="w") as file:
        json.dump(setting.to_dict(),file,indent=2)

def window_quit():
    if old_user_data != new_user_data or old_setting_data != new_setting_data:
        if messagebox.askyesno("Exit","Do you want to save your data?"):
            save(new_user_data,new_setting_data)
    window.destroy()

def click_save_button():
    global old_user_data,old_setting_data
    try:
        save(new_user_data,new_setting_data)
        old_user_data = copy.deepcopy(new_user_data)
        old_setting_data = copy.deepcopy(new_setting_data)
        messagebox.showinfo("Message","Save data successfully")
    except:
        messagebox.showerror("Error","Failed to save data,please check your json files!")

def show_passwd():
    global passwd_show
    passwd_show = not passwd_show
    if passwd_show:
        passwd_entry_box.configure(show="")
        passwd_show_button.configure(text="Hide")
    else:
        passwd_entry_box.configure(show="*")
        passwd_show_button.configure(text="Show")

def user_ok():
    std_id,passwd = id_entry_box.get(),passwd_entry_box.get()
    if not std_id.isdigit() or len(id_entry_box.get()) != 13:
        messagebox.showerror("Invalid student ID","Your student ID should only consist of 13 numbers!")
        id_entry_box.delete(0,END)
        id_entry_box.insert(0,new_user_data.std_id)
    else:
        new_user_data.std_id = std_id
        new_user_data.password = passwd
        messagebox.showinfo("Message","Enter your user information successfully")

def plan_course():
    main_interface()
    course_name_label.configure(text="Plan courses")
    course_list_box.delete(0,END)
    for course in new_user_data.course.a:
        course_list_box.insert(END,"%s_%s"%(course.course_id,course.course_index))
    no_select()

def free_course():
    main_interface()
    course_name_label.configure(text="Free courses")
    course_list_box.delete(0,END)
    for course in new_user_data.course.b:
        course_list_box.insert(END,"%s_%s"%(course.course_id,course.course_index))
    no_select()

def course_select(event):
    course_id,course_index = event.widget.get(event.widget.curselection()).split("_")
    course_id_entry_box.delete(0,END)
    course_index_entry_box.delete(0,END)
    course_id_entry_box.insert(0,course_id)
    course_index_entry_box.insert(0,course_index)
    cancel_button.configure(state=NORMAL)
    delete_button.configure(state=NORMAL)
    manage_button.configure(text="Update",command=update_course)

def no_select(event=None):
    course_list_box.selection_clear(0, END)
    course_id_entry_box.delete(0,END)
    course_index_entry_box.delete(0,END)
    cancel_button.configure(state=DISABLED)
    delete_button.configure(state=DISABLED)
    manage_button.configure(text="Add",command=add_course)

def listbox_locate(listbox:Listbox,scrollbar:Scrollbar,index:int):
    height,size = listbox.cget("height"),listbox.size()
    first = index if index + height <= size else size - height
    last = first + height
    listbox.yview_moveto(first/size)
    scrollbar.set(first=first/size,last=last/size)

def add_course():
    course_id,course_index = course_id_entry_box.get(),course_index_entry_box.get()
    course_list = new_user_data.course.a if course_name_label.cget("text") == "Plan courses" else new_user_data.course.b
    index = -1
    for i in range(len(course_list)):
        if course_list[i].course_id == course_id:
            index = i
            break
    if not course_id or not course_index:
        messagebox.showerror("Empty course","Please input the course information first!")
    elif not course_id.isdigit() or not course_index.isdigit():
        messagebox.showerror("Invalid course","Please enter the right course data that consists of only numbers!")
        if not course_id.isdigit():
            course_id_entry_box.delete(0,END)
        if not course_index.isdigit():
            course_index_entry_box.delete(0,END)
    elif len(course_index) != 2:
        messagebox.showerror("Invalid index","Please enter the right index that consists of only 2 numbers!")
        course_index_entry_box.delete(0,END)
    elif index != -1:
        course_list_box.select_set(index)
        listbox_locate(course_list_box,course_list_box_scroll,index)
        old_course = course_list_box.get(index)
        if course_list[index].course_index != course_index and messagebox.askyesno("Update a already existing course","You already have the course %s\nWould you want to update it?(New:%s_%s)"%(old_course,course_id,course_index)):
            course_list_box.delete(index)
            course_list_box.insert(index,"%s_%s"%(course_id,course_index))
            course_list_box.select_set(index)
            course_list[index].course_index = course_index
            course_index_entry_box.delete(0,END)
            course_index_entry_box.insert(0,course_index)
        cancel_button.configure(state=NORMAL)
        delete_button.configure(state=NORMAL)
        manage_button.configure(text="Update",command=update_course)
    else:
        course_list_box.insert(END,"%s_%s"%(course_id,course_index))
        course_list.append(Course(course_id=course_id,course_index=course_index))
        course_id_entry_box.delete(0,END)
        course_index_entry_box.delete(0,END)

def update_course():
    index = course_list_box.curselection()[0]
    old_id,old_index = course_list_box.get(index).split("_")
    new_id,new_index = course_id_entry_box.get(),course_index_entry_box.get()
    course_list = new_user_data.course.a if course_name_label.cget("text") == "Plan courses" else new_user_data.course.b
    if not new_id or not new_index:
        messagebox.showerror("Empty course","Please input the course information first!")
        if not new_id:
            course_id_entry_box.insert(0,old_id)
        if not new_index:
            course_index_entry_box.insert(0,old_index)
    elif not new_id.isdigit() or not new_index.isdigit():
        messagebox.showerror("Invalid course","Please enter the right course data that consists of only numbers!")
        if not new_id.isdigit():
            course_id_entry_box.delete(0,END)
            course_id_entry_box.insert(0,old_id)
        if not new_index.isdigit():
            course_index_entry_box.delete(0,END)
            course_index_entry_box.insert(0,old_index)
    elif len(new_index) != 2:
        messagebox.showerror("Invalid index","Please enter the right index that consists of only 2 numbers!")
        course_index_entry_box.delete(0,END)
        course_index_entry_box.insert(0,old_index)
    else:
        course_list_box.delete(index)
        course_list_box.insert(index,"%s_%s"%(new_id,new_index))
        course = course_list[index]
        course.course_id,course.course_index = new_id,new_index

def delete_course():
    index = course_list_box.curselection()[0]
    course_id,course_index = course_list_box.get(index).split("_")
    if messagebox.askyesno("Confirm","Are you sure you want to delete the course %s_%s?"%(course_id,course_index)):
        course_list_box.delete(index)
        course_id_entry_box.delete(0,END)
        course_index_entry_box.delete(0,END)
        course_list = new_user_data.course.a if course_name_label.cget("text") == "Plan courses" else new_user_data.course.b
        course_list.remove(Course(course_id=course_id,course_index=course_index))

def manual_login_click():
    if manual_login_button.cget("text") == "Disable":
        if messagebox.askyesno("Disable manual login","It\'ll significantly accelerate the login, but it also usually leads to the incapability of the login when the server is busy (especially during peak course selection periods).\nAre you sure to disable?"):
            try:
                new_setting_data.manual_login = False
                manual_login_button.configure(text="Enable")
                manual_login_state.configure(text="Disabled",fg="red")
                messagebox.showinfo("Message","Disable manual login successfully")
            except:
                messagebox.showwarning("Error","Failed to disable manual login!")
    else:
        if messagebox.askyesno("Enable manual login","It\'ll significantly stabilize the login (especially during peak course selection periods), but it\'ll also significantly slow the login down.\nAre you sure to enable?"):
            try:
                new_setting_data.manual_login = True
                manual_login_button.configure(text="Disable")
                manual_login_state.configure(text="Enabled",fg="green")
                messagebox.showinfo("Message","Enable manual login successfully")
            except:
                messagebox.showwarning("Error","Failed to enable manual login!")

def browser_click():
    if new_setting_data.browser == "edge":
        new_setting_data.browser = "chrome"
        browser_state.configure(text="Chrome",fg="blue")
        browser_button.configure(text="Switch to Edge")
    else:
        new_setting_data.browser = "edge"
        browser_state.configure(text="Edge",fg="green")
        browser_button.configure(text="Switch to Chrome")

def main_interface():
    for widget in window.winfo_children():
        widget.place_forget()

    id_label.place(x=size[0]-290,y=50)
    id_entry_box.place(x=size[0]-220,y=50)
    passwd_label.place(x=size[0]-285,y=85)
    passwd_entry_box.place(x=size[0]-220,y=85)
    passwd_show_button.place(x=size[0]-70,y=80)
    user_ok_button.place(x=size[0]-175,y=125)
    save_button.place(x=size[0]-225,y=size[1]-110)
    course_name_label.place(x=50,y=5)
    course_list_box.place(x=50,y=30)
    course_list_box_scroll.place(x=295,y=30,height=380)
    course_id_label.place(x=size[0]-290,y=200)
    course_index_label.place(x=size[0]-310,y=235)
    course_id_entry_box.place(x=size[0]-220,y=200)
    course_index_entry_box.place(x=size[0]-220,y=235)
    cancel_button.place(x=size[0]-230,y=280)
    manage_button.place(x=size[0]-180,y=325)
    delete_button.place(x=size[0]-130,y=280)

def setting_interface():
    for widget in window.winfo_children():
        widget.place_forget()

    save_button.place(x=size[0]-225,y=size[1]-110)
    manual_login_label.place(x=50,y=20)
    manual_login_state.place(x=280,y=23)
    manual_login_button.place(x=500,y=20)
    browser_label.place(x=50,y=60)
    browser_state.place(x=280,y=63)
    browser_button.place(x=500,y=60)

#load the data,if it exists
try:
    old_user_data = UserData.from_file("user_data.json")
    old_setting_data = Setting.from_file("setting.json")
except Exception as e:
    print(e)
    old_user_data = UserData(std_id="", password="", course=CourseData())
    old_setting_data = Setting(manual_login=False)

new_user_data = copy.deepcopy(old_user_data)
new_setting_data = copy.deepcopy(old_setting_data)
passwd_show = False
size = [640,480]

#Create a gui
window = Tk()
window.title("Initialize")
window.iconbitmap("favicon.ico")
window.geometry("%dx%d"%(size[0],size[1]))
window.protocol("WM_DELETE_WINDOW",window_quit)

#Labels
id_label = Label(window,text="Student ID")
passwd_label = Label(window,text="Password")
course_id_label = Label(window,text="Course ID")
course_index_label = Label(window,text="Course Index")
course_name_label = Label(window,font=("consolas",15))

manual_login_label = Label(window,text="Manual login")
manual_login_state = Label(window)

browser_label = Label(window,text="Browser")
browser_state = Label(window)

#initialize the state
if new_setting_data.manual_login:
    manual_login_state.configure(text="Enabled",fg="green")
else:
    manual_login_state.configure(text="Disabled",fg="red")

#Entries
id_entry_box = Entry(window,width=20)
passwd_entry_box = Entry(window,width=20,show="*")
course_id_entry_box = Entry(window,width=20)
course_index_entry_box = Entry(window,width=20)

#Buttons
passwd_show_button = Button(window,text="Show",command=show_passwd)
save_button = Button(window,text="Save",width=20,height=2,command=click_save_button)
user_ok_button = Button(window,text="OK",width=7,height=1,command=user_ok)
cancel_button = Button(window,text="Cancel",width=7,height=1,state=DISABLED,command=no_select)
manage_button = Button(window,text="Add",width=7,height=1,command=add_course)
delete_button = Button(window,text="Delete",width=7,height=1,state=DISABLED,command=delete_course)

manual_login_button = Button(window,width=7,height=1,command=manual_login_click)

browser_button = Button(window,width=15,height=1,command=browser_click)

#initialize the setting button
if new_setting_data.manual_login:
    manual_login_button.configure(text="Disable")
else:
    manual_login_button.configure(text="Enable")

#initialize the browser state
if new_setting_data.browser == "edge":
    browser_state.configure(text="Edge",fg="green")
    browser_button.configure(text="Switch to Chrome")
else:
    browser_state.configure(text="Chrome",fg="blue")
    browser_button.configure(text="Switch to Edge")

#Listbox
course_list_box_scroll = Scrollbar(window,width=20)
course_list_box = Listbox(window,height=15,font=("consolas",16),yscrollcommand=course_list_box_scroll.set,exportselection=False)
course_list_box.bind("<Button-3>",no_select)
course_list_box.bind("<<ListboxSelect>>",course_select)
course_list_box_scroll.config(command=course_list_box.yview)

#Menus
window_menu = Menu(window)
window_menu_list = Menu(window_menu,tearoff=False)
window_menu.add_cascade(label="Course type",menu=window_menu_list)
window_menu_list.add_command(label="Plan course",command=plan_course)
window_menu_list.add_command(label="Free course",command=free_course)
window_menu.add_command(label="Settings",command=setting_interface)

#Insert
id_entry_box.insert(0,new_user_data.std_id)
passwd_entry_box.insert(0,new_user_data.password)

"""
#Place
id_label.place(x=size[0]-290,y=50)
id_entry_box.place(x=size[0]-220,y=50)
passwd_label.place(x=size[0]-285,y=85)
passwd_entry_box.place(x=size[0]-220,y=85)
passwd_show_button.place(x=size[0]-70,y=80)
user_ok_button.place(x=size[0]-175,y=125)
save_button.place(x=size[0]-225,y=size[1]-110)
course_name_label.place(x=50,y=5)
course_list_box.place(x=50,y=30)
course_list_box_scroll.place(x=295,y=30,height=380)
course_id_label.place(x=size[0]-290,y=200)
course_index_label.place(x=size[0]-310,y=235)
course_id_entry_box.place(x=size[0]-220,y=200)
course_index_entry_box.place(x=size[0]-220,y=235)
cancel_button.place(x=size[0]-230,y=280)
manage_button.place(x=size[0]-180,y=325)
delete_button.place(x=size[0]-130,y=280)
"""
window.config(menu=window_menu)
plan_course()

#keep the gui active
window.mainloop()