Trae+DeepSeek学习Python开发MVC框架程序笔记（四）:使用sqlite存储查询并验证用户名和密码

继续通过Trae向DeepSeek发问并修改程序，实现程序运行时生成数据库，用户在系统登录页面输入用户名和密码后，控制器通过模型查询用户数据库表来验证用户名和密码，验证通过后显示登录成功页面，验证失败则显示登录失败页面。

目录结果如下图所示：

mvctest2/
│── model.py        # 数据模型(SQLite数据库操作)
│── view.py         # 视图界面
│── controller.py   # 控制器
│── main.py         # 程序入口
│── database.py     # 数据库初始化脚本
│── user.db         # 数据库初始化脚本首次运行后自动生成

1. 数据库初始化脚本 (database.py)

#database.py
# 数据库模块 (Model) - 处理数据存储和查询import sqlite3
from pathlib import PathDB_PATH = Path(__file__).parent / "users.db"def init_db():"""初始化数据库和用户表"""conn = sqlite3.connect(DB_PATH)cursor = conn.cursor()# 创建用户表cursor.execute("""CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY AUTOINCREMENT,username TEXT UNIQUE NOT NULL,password TEXT NOT NULL)""")# 插入测试用户数据test_users = [('admin', '123456'),('user1', 'password1'),('user2', 'password2')]cursor.executemany("INSERT OR IGNORE INTO users (username, password) VALUES (?, ?)",test_users)conn.commit()conn.close()if __name__ == "__main__":init_db()print(f"数据库已初始化，路径: {DB_PATH}")

2. 模型文件 model.py

#model.py
# 模型 (Model)
import sqlite3
from pathlib import Path
from database import DB_PATH
from PyQt5.QtCore import QObject, pyqtSignalclass UserModel(QObject):#login_success = pyqtSignal(str)  # 登录成功信号，带用户名参数#login_failed = pyqtSignal()      # 登录失败信号def __init__(self):super().__init__()self.conn = sqlite3.connect(DB_PATH)# 模型方法：验证用户凭据def query(self, username, password):"""验证用户凭据"""cursor = self.conn.cursor()sql = "SELECT username FROM users WHERE username=? AND password=?"cursor.execute(sql,(username, password))return cursor.fetchone() is not Nonedef __del__(self):"""关闭数据库连接"""self.conn.close()

3.视图文件view.py

# view.py
# 
from PyQt5.QtWidgets import (QWidget, QLabel, QLineEdit, QPushButton, QVBoxLayout, QMessageBox)
from PyQt5.QtCore import Qtclass LoginView(QWidget):def __init__(self, controller=None):super().__init__()self.controller = controllerself.init_ui()# 初始化UI界面        def init_ui(self):self.setWindowTitle('系统登录')self.resize(300, 200)layout = QVBoxLayout()# 用户名输入self.lbl_user = QLabel('用户名:')self.txt_user = QLineEdit()layout.addWidget(self.lbl_user)layout.addWidget(self.txt_user)# 密码输入self.lbl_pass = QLabel('密码:')self.txt_pass = QLineEdit()self.txt_pass.setEchoMode(QLineEdit.Password)layout.addWidget(self.lbl_pass)layout.addWidget(self.txt_pass)# 登录按钮self.btn_login = QPushButton('登录')self.btn_login.clicked.connect(self.on_login)layout.addWidget(self.btn_login)# 登录状态显示self.lbl_status = QLabel()self.lbl_status.setAlignment(Qt.AlignCenter)layout.addWidget(self.lbl_status)self.setLayout(layout)# 登录按钮点击事件处理    def on_login(self):username = self.txt_user.text()password = self.txt_pass.text()if self.controller:self.controller.handle_login(username, password)# 显示登录成功或失败消息       def show_success(self, username):self.lbl_status.setText(f"欢迎, {username}!")self.lbl_status.setStyleSheet("color: green")# 显示登录失败消息    def show_failure(self):self.lbl_status.setText("用户名或密码错误!")self.lbl_status.setStyleSheet("color: red")# 显示登录页面    def show(self):super().show()self.txt_user.setFocus()

4.控制器文件 controller.py

# controller.py
# 控制器 (Controller) 
class LoginController:def __init__(self, model, view):self.model = modelself.view = viewself.view.controller = self  # 设置view的controller引用# 处理登录逻辑    def run(self):self.view.show()  # 直接调用view的show方法# 处理登录按钮点击事件   def handle_login(self, username, password):if self.model.query(username, password):self.view.show_success(username)else:self.view.show_failure()

5.程序运行入口文件mail.py

# main.py
# 主程序入口
import sys
from PyQt5.QtWidgets import QApplication
from model import UserModel
from view import LoginView
from controller import LoginController
from database import init_dbdef main():# 初始化数据库init_db()#创建QT应用    app = QApplication(sys.argv)  # 创建QApplication实例# 初始化MVC组件model = UserModel()view = LoginView()controller = LoginController(model, view)# 启动应用controller.run()sys.exit(app.exec_())if __name__ == "__main__":main()

运行可以成功。

由于上述代码80%是DeepSeep帮忙生成的，为便于理解和学习，我通过Trae请DeepSeek根据上述项目代码，分析 MVC架构的类方法调用关系，系统成生了时序图代码如下：

sequenceDiagramparticipant Mainparticipant Controller as LoginControllerparticipant Model as UserModelparticipant View as LoginViewMain->>Controller: 创建实例(model, view)Controller->>View: 设置controller引用Main->>Controller: run()Controller->>View: show()View->>Controller: on_login()点击事件Controller->>Model: query(username, password)alt 验证成功Model-->>Controller: 返回TrueController->>View: show_success(username)else 验证失败Model-->>Controller: 返回FalseController->>View: show_failure()end

上述代码可以在https://develop.git.mermaid.live/edit

在线生成时序图如下：