# coding=gbk
import pymysql
import tkinter as tk
import tkinter.messagebox
import pickle
from sha import *
# 打开数据库连接
conn = pymysql.connect('localhost', 'root', 'as123', 'teacher')
cursor = conn.cursor()
# 窗口
window = tk.Tk()
window.title('欢迎进入人事管理数据库系统')
window.geometry('480x300')
# 画布放置图片
canvas = tk.Canvas(window, height=300, width=500)
imagefile = tk.PhotoImage(file='1015.png')
image = canvas.create_image(0, 0, anchor='nw', image=imagefile)
canvas.pack(side='top')
# 标签 用户名密码
tk.Label(window, text='用户名:').place(x=100, y=150)
tk.Label(window, text='密码:').place(x=100, 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)


# 登录函数
def usr_log_in():
    # 输入框获取用户名密码
    usr_name = var_usr_name.get()
    usr_pwd = var_usr_pwd.get()
    # 从本地字典获取用户信息，如果没有则新建本地数据库
    sql="SELECT ID,password FROM user WHERE ID='%s'" %(usr_name)
    cursor.execute(sql)
    results = cursor.fetchall()
    # 判断用户名和密码是否匹配
    # print(results)
    # print(usr_name)
    # if True:
    a=results[0]
    if usr_name ==a[0] and usr_pwd==a[1]:
        tk.messagebox.showinfo(title='welcome',
                               message='欢迎您：' + usr_name)
    else:
        tk.messagebox.showerror(message='密码错误')
# 用户名密码不能为空
    if usr_name == '' or usr_pwd == '':
        tk.messagebox.showerror(message='用户名或密码为空')
# 不在数据库中弹出是否注册的框
    else:
        is_signup = tk.messagebox.askyesno('欢迎', '您还没有注册，是否现在注册')
        if is_signup:
            usr_sign_up()

# 注册函数

def usr_sign_up():
    # 确认注册时的相应函数
    def signtowcg():
        # 获取输入框内的内容
        nn = new_name.get()
        np = new_pwd.get()
        npf = new_pwd_confirm.get()

        # 本地加载已有用户信息,如果没有则已有用户信息为空
        sql = "SELECT ID,password FROM user WHERE ID='%s'" % (nn)
        cursor.execute(sql)
        results = cursor.fetchall()
        # 检查用户名存在、密码为空、密码前后不一致
        print(results)
        if not results:#result为空，用户不存在，可以正常注册
            ID = nn
            password = np
            if np == '' or nn == '':
                tk.messagebox.showerror('错误', '用户名或密码为空')
                window_sign_up.destroy()
            elif np != npf:
                tk.messagebox.showerror('错误', '密码前后不一致')
                window_sign_up.destroy()
            sql = "INSERT INTO user VALUES('%s',%s,'%s')" % (ID, password, 1)
            cursor.execute(sql)
            tk.messagebox.showinfo('欢迎', '注册成功')
            # 注册成功关闭注册框
            window_sign_up.destroy()
        elif nn in results[0]:
            tk.messagebox.showerror('错误', '用户名已存在')
        # 注册信息没有问题则将用户名密码写入数据库

    # 新建注册界面
    window_sign_up = tk.Toplevel(window)
    window_sign_up.geometry('350x200')
    window_sign_up.title('注册')
    # 用户名变量及标签、输入框
    new_name = tk.StringVar()
    tk.Label(window_sign_up, text='用户名：').place(x=10, y=10)
    tk.Entry(window_sign_up, textvariable=new_name).place(x=150, y=10)
    # 密码变量及标签、输入框
    new_pwd = tk.StringVar()
    tk.Label(window_sign_up, text='请输入密码：').place(x=10, y=50)
    tk.Entry(window_sign_up, textvariable=new_pwd, show='*').place(x=150, y=50)
    # 重复密码变量及标签、输入框
    new_pwd_confirm = tk.StringVar()
    tk.Label(window_sign_up, text='请再次输入密码：').place(x=10, y=90)
    tk.Entry(window_sign_up, textvariable=new_pwd_confirm, show='*').place(x=150, y=90)
    # 确认注册按钮及位置
    bt_confirm_sign_up = tk.Button(window_sign_up, text='确认注册',
                                   command=signtowcg)
    bt_confirm_sign_up.place(x=150, y=130)


# 退出的函数
def usr_sign_quit():
    window.destroy()


# 登录 注册按钮
bt_login = tk.Button(window, text='登录', command=usr_log_in)
bt_login.place(x=140, y=230)
bt_logup = tk.Button(window, text='注册', command=usr_sign_up)
bt_logup.place(x=210, y=230)
bt_logquit = tk.Button(window, text='退出', command=usr_sign_quit)
bt_logquit.place(x=280, y=230)
# 主循环
window.mainloop()
