import tkinter as tk
import tkinter.messagebox
import pickle
from bazi_invasion import run_game
import  pygame
from pygame.locals import *

window = tk.Tk()
window.title('login')
window.geometry('300x300')

pygame.init()
pygame.mixer.music.load("sound/game_music.ogg")
pygame.mixer.music.set_volume(0.2)
pygame.mixer.music.play(-1)

#welcome image
#创建一个200X500的画布
canvas =  tk.Canvas(window,height = 100,width = 300)
#logo的路径
image_file = tk.PhotoImage(file = 'images/logo.png')
#什么位置插入logo图片
image = canvas.create_image(0,0,anchor = 'nw',image = image_file)
canvas.pack(side = 'top')

# 登陆界面
tk.Label(window, text='账户：').place(x=60, y=100)
tk.Label(window, text='密码：').place(x=60, y=150)

var_usr_name = tk.StringVar()
enter_usr_name = tk.Entry(window, textvariable=var_usr_name)
enter_usr_name.place(x=100, y=100)

var_usr_pwd = tk.StringVar()
enter_usr_pwd = tk.Entry(window, textvariable=var_usr_pwd, show='*')
enter_usr_pwd.place(x=100, y=150)


# 登陆
def usr_log_in():
    # 输入框内容
    usr_name = var_usr_name.get()
    usr_pwd = var_usr_pwd.get()
    try:
        with open('usr_info.pickle', 'rb') as usr_file:
            users_info = pickle.load(usr_file)
    except:
        with open('usr_info.pickle', 'wb') as usr_file:
            users_info = {'admin': 'admin'}
            pickle.dump(users_info, usr_file)

    # 判断
    if usr_name in users_info:
        if usr_pwd == users_info[usr_name]:
            # 登陆成功
            window.destroy()
        else:
            tk.messagebox.showerror(message='ERROR!')
    # 用户名密码不能为空
    elif usr_name == '' or usr_pwd == '':
        tk.messagebox.showerror(message='用户名不能为空！')


def usr_sign_quit():
    window.destroy()


def usr_sign_up():
    def signtowcg():
        NewName = new_name.get()
        NewPwd = new_pwd.get()
        ConfirPwd = pwd_comfirm.get()
        try:
            with open('usr_info.pickle', 'rb') as usr_file:
                exist_usr_info = pickle.load(usr_file)
        except FileNotFoundError:
            exist_usr_info = {}
        if NewName in exist_usr_info:
            tk.messagebox.showerror(message='用户名存在！')
        elif NewName == '' and NewPwd == '':
            tk.messagebox.showerror(message='用户名和密码不能为空！')
        elif NewPwd != ConfirPwd:
            tk.messagebox.showerror(message='密码前后不一致！')
        else:
            exist_usr_info[NewName] = NewPwd
            with open('usr_info.pickle', 'wb') as usr_file:
                pickle.dump(exist_usr_info, usr_file)
                tk.messagebox.showinfo(message='注册成功！')
                window_sign_up.destroy()

    # 新建注册窗口
    window_sign_up = tk.Toplevel(window)
    window_sign_up.geometry('350x250')
    window_sign_up.title('sign_up')

    # 注册编辑框
    new_name = tk.StringVar()
    new_pwd = tk.StringVar()
    pwd_comfirm = tk.StringVar()

    tk.Label(window_sign_up, text='账户名：').place(x=70, y=40)
    tk.Entry(window_sign_up, textvariable=new_name).place(x=140, y=40)

    tk.Label(window_sign_up, text='密码：').place(x=70, y=90)
    tk.Entry(window_sign_up, textvariable=new_pwd, show='*').place(x=140, y=90)

    tk.Label(window_sign_up, text='确认密码：').place(x=70, y=140)
    tk.Entry(window_sign_up, textvariable=pwd_comfirm, show='*').place(x=140, y=140)
    # 确认注册
    bt_confirm = tk.Button(window_sign_up, text='确定', command=signtowcg).place(x=160, y=180)


# 登录 注册按钮
bt_login = tk.Button(window, text='登录', command=usr_log_in)
bt_login.place(x=110, y=200)

bt_signup = tk.Button(window, text='注册', command=usr_sign_up)
bt_signup.place(x=160, y=200)

window.wait_window(window=window)
run_game()

window.mainloop()