Python GUI 用户的登录与注册

数据以文件的形式存储

用户密码加密

import tkinter as tk
import tkinter.messagebox
import pickle
import random
from PIL import ImageTk
import hashlib


def verifycode():
    # 验证码
    code_list = ''
    # 每一位验证码都有三种可能（大写字母，小写字母，数字）
    for i in range(6):  # 控制验证码生成的位数
        state = random.randint(1, 3)
        if state == 1:
            first_kind = random.randint(65, 90)  # 大写字母
            random_uppercase = chr(first_kind)
            code_list = code_list + random_uppercase
        elif state == 2:
            second_kinds = random.randint(97, 122)  # 小写字母
            random_lowercase = chr(second_kinds)
            code_list = code_list + random_lowercase
        elif state == 3:
            third_kinds = random.randint(0, 9)
            code_list = code_list + str(third_kinds)
    return code_list


# 密码加密
def f_sha256(arg):
    sha256_pwd = hashlib.sha256()
    sha256_pwd.update(arg.encode("utf8"))
    return sha256_pwd.hexdigest()


# 登录函数
def user_log_in():
    # 输入框获取用户名、密码、验证码
    user_name = var_user_name.get()
    # 密码
    user_pwd = var_user_pwd.get()
    # 验证码
    var_verify = var_user_verify.get()
    # 从本地字典获取用户信息，如果没有则新建本地数据库
    try:
        with open('user_info.pickle', 'rb') as user_file:
            users_info = pickle.load(user_file)
    except FileNotFoundError:
        with open('user_info.pickle', 'wb') as user_file:
            users_info = {'admin': 'admin'}
            pickle.dump(users_info, user_file)

    # 判断用户名和密码是否匹配
    if user_name in users_info:
        # 判断验证码
        if var_verify == verification_Code:
            if f_sha256(user_pwd) == users_info[user_name]:
                tk.messagebox.showinfo(title='welcome', message='欢迎您：' + user_name)
            else:
                tk.messagebox.showerror(message='密码错误')
        else:
            tk.messagebox.showerror(message='验证码错误')

    # 用户名密码不能为空
    elif user_name == '' or user_pwd == '':
        tk.messagebox.showerror(message='用户名或密码不能为空')
    # 不在磁盘中弹出是否注册的对话框框
    else:
        is_signup = tk.messagebox.askyesno('欢迎', '您还没有注册，是否现在注册')
        if is_signup:
            user_sign_up()


# 注册函数
def user_sign_up():
    # 确认注册时的相应函数
    def sign_check():
        # 获取输入框内的内容
        user_name = new_name.get()
        user_pwd = new_pwd.get()
        user_check_pwd = new_pwd_confirm.get()
        # 本地加载已有用户信息,如果没有则已有用户信息为空
        try:
            with open('user_info.pickle', 'rb') as user_file:
                exist_user_info = pickle.load(user_file)
        except FileNotFoundError:
            exist_user_info = {}
            # 检查用户名已存在、密码为空、密码前后不一致
        if user_name in exist_user_info:
            tk.messagebox.showerror('错误', '用户名已存在')
        elif user_pwd == '' or user_name == '':
            tk.messagebox.showerror('错误', '用户名或密码不能为空')
        elif user_pwd != user_check_pwd:
            tk.messagebox.showerror('错误', '密码前后不一致')
        # 注册信息没有问题则将用户名、密码存入数据库
        else:
            exist_user_info[user_name] = f_sha256(user_pwd)
            with open('user_info.pickle', 'wb') as user_file:
                pickle.dump(exist_user_info, user_file)
            tk.messagebox.showinfo('欢迎', '注册成功')
            # 注册成功关闭注册框
            window_sign_up.destroy()

    # 新建注册界面
    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=sign_check)
    bt_confirm_sign_up.place(x=150, y=130)


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


# 验证码值
verification_Code = verifycode()

# 窗口
window = tk.Tk()
window.title('徐洲的用户管理系统')
window.geometry('450x300')

# 设置登录窗口大小不可拉伸
window.resizable(width=False, height=False)

# 画布放置图片
canvas = tk.Canvas(window, height=300, width=500)
image_file = ImageTk.PhotoImage(file=r'banner.jpg')
image = canvas.create_image(0, 0, anchor='nw', image=image_file)
canvas.pack(side='top')

# 创建4个标签
tk.Label(window, text='用户名').place(x=100, y=110)
tk.Label(window, text='密   码').place(x=100, y=150)
tk.Label(window, text='验证码').place(x=100, y=190)
tk.Label(window, text=verification_Code).place(x=310, y=190)

# 用户名输入框
var_user_name = tk.StringVar()
entry_user_name = tk.Entry(window, textvariable=var_user_name)
entry_user_name.place(x=160, y=110)

# 密码输入框
var_user_pwd = tk.StringVar()
entry_user_pwd = tk.Entry(window, textvariable=var_user_pwd, show='*')
entry_user_pwd.place(x=160, y=150)

# 验证码输入框
var_user_verify = tk.StringVar()
var_user_verify = tk.Entry(window, textvariable=var_user_verify)
var_user_verify.place(x=160, y=190)

# 登录、注册、退出按钮
bt_login = tk.Button(window, text='登录', command=user_log_in)
bt_login.place(x=140, y=230)
bt_logup = tk.Button(window, text='注册', command=user_sign_up)
bt_logup.place(x=210, y=230)
bt_log_quit = tk.Button(window, text='退出', command=user_sign_quit)
bt_log_quit.place(x=280, y=230)

# 主循环
window.mainloop()