login.py

import tkinter as tk
import tkinter.messagebox
import pickle
import os

from PIL import Image, ImageTk


class Login:
    window = None
    right = 0

    # 登录函数
    def usr_log_in(self):
        # 输入框获取用户名密码
        usr_name = self.var_usr_name.get()
        usr_pwd = self.var_usr_pwd.get()
        # 从本地字典获取用户信息，如果没有则新建本地数据库
        try:
            with open('usr_info.pickle', 'rb') as usr_file:
                usrs_info = pickle.load(usr_file)
        except FileNotFoundError:
            with open('usr_info.pickle', 'wb') as usr_file:
                usrs_info = {'admin': 'admin'}
                pickle.dump(usrs_info, usr_file)
        # 判断用户名和密码是否匹配
        if usr_name in usrs_info:
            if usr_pwd == usrs_info[usr_name]:
                tk.messagebox.showinfo(title='welcome', message='欢迎您：' + usr_name)
                self.window.destroy()
                self.right = True
            else:
                tk.messagebox.showerror(message='密码错误')
        # 用户名密码不能为空
        elif usr_name == '' or usr_pwd == '':
            tk.messagebox.showerror(message='用户名或密码为空')
        # 不在数据库中弹出是否注册的框
        else:
            is_signup = tk.messagebox.askyesno('欢迎', '您还没有注册，是否现在注册')
            if is_signup:
                self.usr_sign_up()

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

            # 本地加载已有用户信息,如果没有则已有用户信息为空
            try:
                with open('usr_info.pickle', 'rb') as usr_file:
                    exist_usr_info = pickle.load(usr_file)
            except FileNotFoundError:
                exist_usr_info = {}

                # 检查用户名存在、密码为空、密码前后不一致
            if nn in exist_usr_info:
                tk.messagebox.showerror('错误', '用户名已存在')
            elif np == '' or nn == '':
                tk.messagebox.showerror('错误', '用户名或密码为空')
            elif np != npf:
                tk.messagebox.showerror('错误', '密码前后不一致')
            # 注册信息没有问题则将用户名密码写入数据库
            else:
                exist_usr_info[nn] = np
                with open('usr_info.pickle', 'wb') as usr_file:
                    pickle.dump(exist_usr_info, usr_file)
                tk.messagebox.showinfo('欢迎', '注册成功')
                # 注册成功关闭注册框
                window_sign_up.destroy()

        # 新建注册界面
        window_sign_up = tk.Toplevel(self.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(self):
        self.window.destroy()
        self.right = False

    def __init__(self):
        # 窗口
        self.window = tk.Tk()
        self.window.title('花卉识别系统')
        self.window.geometry('450x300')
        self.topWidth = 450
        self.topheight = 300

        # 窗口居中
        screenwidth = self.window.winfo_screenwidth()
        screenheight = self.window.winfo_screenheight()
        alignstr = '%dx%d+%d+%d' % (
            self.topWidth, self.topheight, (screenwidth - self.topWidth) / 2, (screenheight - self.topheight) / 2)
        self.window.geometry(alignstr)
        ima = Image.open(os.path.dirname(os.path.realpath(__file__)) + '/bg.jpg').resize((500, 300))
        ima = ImageTk.PhotoImage(ima)

        # 画布放置图片
        canvas = tk.Canvas(self.window, height=300, width=500, )
        image = canvas.create_image(0, 0, anchor='nw', image=ima)
        canvas.pack(side='top')

        # 标签 用户名密码
        tk.Label(self.window, text='用户名:').place(x=100, y=150)
        tk.Label(self.window, text='密码:').place(x=100, y=190)

        # 用户名输入框
        self.var_usr_name = tk.StringVar()
        self.entry_usr_name = tk.Entry(self.window, textvariable=self.var_usr_name)
        self.entry_usr_name.place(x=160, y=150)
        # self.entry_usr_name.place(x=140, y=230)

        # 密码输入框
        self.var_usr_pwd = tk.StringVar()
        self.entry_usr_pwd = tk.Entry(self.window, textvariable=self.var_usr_pwd, show='*')
        self.entry_usr_pwd.place(x=160, y=190)

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