Tkinter制作登录界面以及登陆后页面切换--用户数据从数据库获取并进行正当 ...

续接上集，废话不多说，开搞!
新增功能

• 数据库管理
  
• 用户表创建（用户信息增删改查操作）
  
• 完善登录校验
  

1. 数据库管理（SqlLite）

   下面代码中 from data import Sql, BcryptUtils 引入的是本身写的包，Sql用来记载需要初始化的信息（创建基础表，添加管理用户等）， BcryptUtils 重要是对密码进行加密哈希和校验密码是否正确

1. import json
2. import sqlite3
4. from data import Sql, BcryptUtils
7. class SqlUtils:
8.     def __init__(self):
9.         # 不再在初始化时直接建立连接，而是在需要时建立
10.         self.database_name = 'platform.db'
12.     def init(self):
13.         """ 初始化数据库 """
14.         # 创建基础数据表和初始化表单数据时建立连接并在完成后关闭
15.         with sqlite3.connect(self.database_name) as conn:
16.             cursor = conn.cursor()
17.             for sql in Sql.create_sqls:
18.                 cursor.execute(sql)
19.             for sql in Sql.insert_sqls:
20.                 cursor.execute(sql)
21.             conn.commit()
23.     def query(self, sql):
24.         """
25.         查询数据，返回格式:{'字段名':'数值'}
26.         :param sql: 需要查询的sql
27.         :return: 返回查询到的数据，没有数据时返回None
28.         """
29.         with sqlite3.connect(self.database_name) as conn:
30.             cursor = conn.cursor()
31.             cursor.execute(sql)
32.             _rows = cursor.fetchall()
33.             columns = [description[0] for description in cursor.description]
34.             return [dict(zip(columns, _row)) for _row in _rows]
36.     def insert(self, sql: str):
37.         """
38.         插入数据方法
39.         :param sql: 插入数据的sql
40.         :return: 成功返回True，否则False
41.         """
42.         with sqlite3.connect(self.database_name) as conn:
43.             cursor = conn.cursor()
44.             cursor.execute(sql)
45.             conn.commit()
46.             rows_affected = cursor.rowcount
47.             return rows_affected > 0
49.     def insert_json(self, table_name: str, data_dict: json) -> object:
50.         # 解析 JSON 获取键值对
51.         columns = ', '.join(data_dict.keys())
52.         placeholders = ', '.join(['?'] * len(data_dict))
53.         values = tuple(data_dict.values())
55.         with sqlite3.connect(self.database_name) as conn:
56.             cursor = conn.cursor()
57.             # 拼接 SQL 语句
58.             sql = f"INSERT INTO {table_name} ({columns}) VALUES ({placeholders})"
59.             try:
60.                 cursor.execute(sql, values)
61.                 conn.commit()
62.                 return True
63.             except Exception as e:
64.                 print(f"插入失败：{e}")
65.                 conn.rollback()
66.                 return False
68.     def delete(self, table_name: str, data_dict: json):
69.         if not data_dict or len(data_dict) == 0:
70.             return False, "条件为空，不允许删除。"
71.         conditions = []
72.         values = []
73.         for key, value in data_dict.items():
74.             conditions.append(f"{key}=?")
75.             values.append(value)
76.         sql = f"DELETE FROM {table_name} WHERE {' AND '.join(conditions)}"
77.         with sqlite3.connect(self.database_name) as conn:
78.             try:
79.                 cursor = conn.cursor()
80.                 cursor.execute(sql, values)
81.                 conn.commit()
82.                 rows_affected = cursor.rowcount
83.                 return rows_affected > 0
84.             except Exception as e:
85.                 print(f"删除失败：{e}")
86.                 conn.rollback()
87.                 return False
89.     def login_user(self, username, password):
90.         """
91.         用户登录判断接口,返回True代表允许登录
92.         :param username: 用户账号
93.         :param password: 用户密码
94.         :return: 成功返回True,用户数据，失败则返回False,原因
95.         """
96.         user_data = self.query(f"SELECT * FROM sys_user WHERE username='{username}'")
97.         if user_data:
98.             row = user_data[0]
99.             if BcryptUtils.check_password(password, row['password']):
100.                 return True, row
101.             else:
102.                 return False, '账号或密码错误'
103.         else:
104.             return False, "没有当前登录的账号信息。"
107. if __name__ == '__main__':
108.     sql_utils = SqlUtils()
109.     sql_utils.init()
110.     sql_utils.insert_json("sys_user", {
111.         "name": "测试", "username": "admin",
112.         "password": BcryptUtils.hash_password('123456')
113.     })
114.     input_username = "admin"
115.     input_password = "123456"
116.     res, data = sql_utils.login_user(input_username, input_password)
117.     if res is True:
118.         extend_str = data.get('extend')
119.         extend_dict = json.loads(extend_str) if extend_str else {}
120.         print(f"欢迎您，{data.get('name')}，您的账号是：{data.get('username')}，头像：{extend_dict.get('img')}")
121.     else:
122.         print(f'{data}')
123.     # 删除用户
124.     sql_utils.delete("sys_user", {
125.         "username": "admin"
126.     })

复制代码

2. 用户表创建（用户信息增删改查操作）

   data/Sql.py

1. create_sqls = []
3. create_drop_user = "DROP TABLE IF EXISTS sys_user;"
4. create_sqls.append(create_drop_user)
6. create_user = '''
7.         CREATE TABLE IF NOT EXISTS sys_user
8.         (
9.             id        INTEGER PRIMARY KEY,
10.             name      varchar(255)       not null,
11.             username  varchar(32)        not null,
12.             password  varchar(255)       not null,
13.             is_active blob default false not null,
14.             extend    json,
15.             unique (username)
16.         );
17.         '''
18. create_sqls.append(create_user)
20. insert_sqls = []
22. insert_user = """
23. insert into sys_user(name, username, password, extend)
24.         values ('超级管理员', 'admin',
25.         '$2b$12$v/PNEi9kMoAafg17SneZ8eiX2/CC/BGWjCI4l.ynkfLcRhXm4TPdW',
26.         '{"img":"http:127.0.0.1/statics/super/image/super.png"}');
27. """
28. insert_sqls.append(insert_user)

复制代码

3. 完善登录校验

   data/BcryptUtils.py ,引入依靠下令： pip install bcrypt

1. import bcrypt
4. def check_password(plain_password, hashed_password):
5.     """
6.     验证密码
7.     :param plain_password: 明文密码
8.     :param hashed_password: 哈希密码
9.     :return: 匹配则返回True，否则返回False
10.     """
11.     plain_password = plain_password.encode('utf-8')
12.     hashed_password = hashed_password.encode('utf-8')
13.     return bcrypt.checkpw(plain_password, hashed_password)
16. def hash_password(password):
17.     """
18.     对密码进行加密
19.     :param password: 用户密码
20.     :return: 返回加密的密文
21.     """
22.     password = password.encode('utf-8')
23.     hashed = bcrypt.hashpw(password, bcrypt.gensalt())
24.     return hashed.decode('utf-8')

复制代码

  login/ui.py 不清晰这个去看一下一中的ui.py全代码，此处只展示修改的信息

1. from data.SqlLiteUtils import SqlUtils
4. class Controller:
5.     # 导入UI类后，替换以下的 object 类型，将获得 IDE 属性提示功能
6.     ui: WinGUI
8.     def __init__(self):
9.         ....保持原有代码....下方是新增代码
10.         self.sql_utils = SqlUtils()
11.    
12.     def login_submit(self, evt):
13.         u, p = self.ui.login()
14.         if u is None or len(u) == 0:
15.             tkinter.messagebox.showinfo("登录提示", "请输入登录用户名！")
16.         elif p is None or len(p) == 0:
17.             tkinter.messagebox.showinfo("登录提示", "请输入登录密码！")
18.         print(
19.             f"触发了登录操作，地址：{self.url},账号：{u},密码：{p}")
20.         res, user = self.sql_utils.login_user(u, p)
21.         if res is True:
22.             print(f"欢迎您，{user.get('name')}")
23.             self.ui.destroy()
24.             from home.control import Controller as HomeUIController
25.             from home.ui import Win as MainWin
26.             app = MainWin(HomeUIController())
27.             app.set_user(user)
28.             app.mainloop()
29.         else:
30.             tkinter.messagebox.showinfo("登录提示", user)

复制代码

  home/ui.py 此处一样只展示心增代码

1. class WinGUI(Tk):
2.     def __init__(self):
3.         super().__init__()
4.                 self.user: json = {}
5.                
6.         def set_user(self, user: json):
7.         self.user = user

复制代码

好了，到这里结束了，后续会开始做注册用户操作，需要团体源代码的可以留言或SL俺！

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。