import tkinter as tk
from tkinter import messagebox  # import this to fix messagebox error
import pickle
username = ''
# 登錄界面
def welcome():
    def usr_login():
        usr_name = var_usr_name.get()
        usr_pwd = var_usr_pwd.get()
        try:
            with open('usrs_info.pickle', 'rb') as usr_file:
                usrs_info = pickle.load(usr_file)
                usr_file.close()
        except FileNotFoundError:
            with open('usrs_info.pickle', 'wb') as usr_file:
                usrs_info = {'admin': 'admin'}
                pickle.dump(usrs_info, usr_file)
                usr_file.close()
        if usr_name in usrs_info:
            if usr_pwd == usrs_info[usr_name]:
                tk.messagebox.showinfo(title='Welcome', message='How are you? ' + usr_name)
                # 進(jìn)入主頁
                global username
                username= usr_name
                print(username)
                window.destroy()
                index_page()
            else:
                tk.messagebox.showerror(message='Error, your password is wrong, try again.')
        else:
            is_sign_up = tk.messagebox.askyesno('Welcome', 'You have not signed up yet. Sign up today?')
            if is_sign_up:
                usr_sign_up()
    window = tk.Tk()
    window.title('Welcome to the login page!')
    window.geometry('450x300')
    # welcome image
    canvas = tk.Canvas(window, height=200, width=500)
    image_file = tk.PhotoImage(file='welcome.gif')
    image = canvas.create_image(0, 0, anchor='nw', image=image_file)
    canvas.pack(side='top')
    # user information
    tk.Label(window, text='User name: ').place(x=50, y=150)
    tk.Label(window, text='Password: ').place(x=50, y=190)
    var_usr_name = tk.StringVar()
    entry_usr_name = tk.Entry(window, textvariable=var_usr_name)
    entry_usr_name.place(x=160, y=150)
    var_usr_pwd = tk.StringVar()
    entry_usr_pwd = tk.Entry(window, textvariable=var_usr_pwd, show='*')
    entry_usr_pwd.place(x=160, y=190)
    # login and sign up button
    btn_login = tk.Button(window, text='Login', command=usr_login)
    btn_login.place(x=170, y=230)
    btn_sign_up = tk.Button(window, text='Sign up', command=usr_sign_up)
    btn_sign_up.place(x=270, y=230)
    window.mainloop()
# 個人信息頁面
def index_page():
    global username
    index_window = tk.Tk()
    index_window.title('請選擇你要進(jìn)行的操作')
    index_window.geometry('300x200')
    tk.Label(index_window,text="你好!"+username).place(x=50,y=50)
    # 修改個人信息
    def change_personal_info():
        # 新窗口
        change_info__window = tk.Toplevel()
        change_info__window.title('修改個人信息')
        change_info__window.geometry('400x300')
        # 修改個人信息點擊頁面
        def change_info_click():
            name = new_name.get()
            psw = new_password.get()
            conpsw = new_password_confirm.get()
            with open('usrs_info.pickle', 'rb') as f:
                usrs_info = pickle.load(f)
                f.close()
            if conpsw!= psw:
                tk.messagebox.showerror(title='Error', message='兩次密碼不一致請重新輸入！')
            else:
                if name in usrs_info:
                    tk.messagebox.showerror(title='Error', message='用戶名已存在，請重新?lián)Q一個新用戶名')
                else:
                    # 創(chuàng)建新
                    usrs_info.pop(username)
                    usrs_info[name] = psw
                    with open('usrs_info.pickle', 'wb') as file:
                        pickle.dump(usrs_info, file)
                        f.close()
                    tk.messagebox.showinfo(title='修改成功', message='修改成功！')
                    change_info__window.destroy()
                    tk.messagebox.showinfo(title='請重新登錄！', message='請重新登錄！')
                    index_window.destroy()
                    welcome()
        # 修改用戶名變量
        new_name = tk.StringVar()
        tk.Label(change_info__window, text="用戶名：").place(x=50, y=50)
        new_name_entry = tk.Entry(change_info__window, show=None, textvariable=new_name).place(x=120, y=50)
        # 修改新密碼變量
        new_password = tk.StringVar()
        tk.Label(change_info__window, text='密碼:').place(x=50, y=80)
        new_password_entry = tk.Entry(change_info__window, show="*", textvariable=new_password).place(x=120, y=80)
        # 修改新確認(rèn)密碼變量
        new_password_confirm = tk.StringVar()
        tk.Label(change_info__window, text="確認(rèn)密碼：").place(x=50, y=110)
        new_password_confirm_entry = tk.Entry(change_info__window, show="*", textvariable=new_password_confirm).place(x=120,
                                                                                                 y=110)
        # 修改信息按鈕綁定
        sign_window_button = tk.Button(change_info__window, text='確認(rèn)', command=change_info_click).place(x=150, y=140)
    # 修改個人信息按鈕綁定
    change_info_button = tk.Button(index_window,text='修改個人信息',command=change_personal_info).place(x=100,y=100)
# 注冊頁面
def usr_sign_up():
    sign_up_window = tk.Toplevel()
    sign_up_window.title('注冊')
    sign_up_window.geometry('400x200')
    # 注冊點擊事件
    def sign_up_hit():
        name = new_name.get()
        psw = new_password.get()
        conpsw = new_password_confirm.get()
        with open('usrs_info.pickle', 'rb') as usr_file:
            usrs_info = pickle.load(usr_file)
        if psw!=conpsw:
            tk.messagebox.showerror(title='Error', message='兩次密碼不一致請重新輸入！')
        else:
            if name in usrs_info:
                tk.messagebox.showerror(title='Error', message='用戶名已存在')
            else:
                # 創(chuàng)建新
                usrs_info[name] = psw
                with open('usrs_info.pickle', 'wb') as f:
                    pickle.dump(usrs_info,f)
                tk.messagebox.showinfo(title='注冊成功', message='注冊成功！')
                sign_up_window.destroy()
    # 注冊名變量
    new_name = tk.StringVar()
    tk.Label(sign_up_window,text="注冊名：").place(x=50,y=50)
    new_name_entry = tk.Entry(sign_up_window,show=None,textvariable=new_name).place(x=120,y=50)
    # 注冊密碼變量
    new_password = tk.StringVar()
    tk.Label(sign_up_window,text='密碼:').place(x=50,y=80)
    new_password_entry = tk.Entry(sign_up_window,show="*",textvariable=new_password).place(x=120,y=80)
    # 注冊確認(rèn)密碼變量
    new_password_confirm= tk.StringVar()
    tk.Label(sign_up_window,text="確認(rèn)密碼：").place(x=50,y=110)
    new_password_confirm_entry = tk.Entry(sign_up_window,show="*",textvariable=new_password_confirm).place(x=120,y=110)
    # 注冊按鈕和點擊事件綁定
    sign_window_button = tk.Button(sign_up_window,text='注冊',command=sign_up_hit).place(x=150,y=140)
welcome()