python结合QT实现简单的图书管理系统

class Role:
    ADMIN = "admin"
    TEACHER = "teacher"
    STUDENT = "student"

class Permission:
    ADD_BOOK = "add_book"
    DELETE_BOOK = "delete_book"
    IMPORT_BOOK = "import_book"
    EXPORT_BOOK = "export_book"

class User:
    def __init__(self, id, username, role):
        self.id = id
        self.username = username
        self.role = role

class Book:
    def __init__(self, id, title, author, isbn, category):
        self.id = id
        self.title = title
        self.author = author
        self.isbn = isbn
        self.category = category

import sqlite3

class Database:
    def __init__(self, db_name="library.db"):
        self.conn = sqlite3.connect(db_name)
        self.conn.row_factory = sqlite3.Row  # ← 关键！让结果像字典一样访问
        self.init_tables()

    def init_tables(self):
        # 创建 users 表
        self.conn.execute("""
            CREATE TABLE IF NOT EXISTS users(
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                username TEXT UNIQUE,
                password TEXT,
                role TEXT
            )
        """)
        # 创建 books 表
        self.conn.execute("""
            CREATE TABLE IF NOT EXISTS books(
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                title TEXT NOT NULL,
                author TEXT,
                isbn TEXT,
                category TEXT
            )
        """)
        self.conn.commit()

    def execute(self, sql, params=()):
        cursor = self.conn.cursor()
        cursor.execute(sql, params)
        self.conn.commit()
        return cursor

import hashlib
from dao.database import Database
from model.user import User
from model.role import Role

class UserController:
    def __init__(self):
        self.db = Database()
        self.ensure_admin()  # 确保 admin 用户存在

    def hash_password(self, password):
        return hashlib.sha256(password.encode()).hexdigest()

    def ensure_admin(self):
        cursor = self.db.execute("SELECT * FROM users WHERE username=?", ("admin",))
        if cursor.fetchone() is None:
            self.register("admin", "123456", Role.ADMIN)

    def register(self, username, password, role):
        if not username or not password:
            raise Exception("用户名或密码不能为空")
        self.db.execute("""
            INSERT INTO users(username, password, role)
            VALUES (?, ?, ?)
        """, (username, self.hash_password(password), role))

    def login(self, username, password):
        cursor = self.db.execute("SELECT * FROM users WHERE username=?", (username,))
        user = cursor.fetchone()
        if user and user["password"] == self.hash_password(password):
            return User(user["id"], user["username"], user["role"])
        return None

from model.role import Role
from model.permission import Permission

class PermissionManager:
    ROLE_PERMISSIONS = {
        Role.ADMIN: [
            Permission.ADD_BOOK,
            Permission.DELETE_BOOK,
            Permission.IMPORT_BOOK,
            Permission.EXPORT_BOOK
        ],
        Role.TEACHER: [Permission.ADD_BOOK],
        Role.STUDENT: []
    }

    @classmethod
    def has_permission(cls, role, permission):
        return permission in cls.ROLE_PERMISSIONS.get(role, [])

from dao.database import Database
from controller.permission_manager import PermissionManager
from model.permission import Permission

class BookController:
    def __init__(self, user):
        self.db = Database()
        self.user = user

    def add_book(self, title, author, isbn, category):
        if not PermissionManager.has_permission(self.user.role, Permission.ADD_BOOK):
            raise Exception("没有权限")
        self.db.execute("""
            INSERT INTO books(title, author, isbn, category)
            VALUES (?, ?, ?, ?)
        """, (title, author, isbn, category))

    def delete_book(self, book_id):
        if not PermissionManager.has_permission(self.user.role, Permission.DELETE_BOOK):
            raise Exception("没有权限")
        self.db.execute("DELETE FROM books WHERE id=?", (book_id,))

    def list_books(self):
        cursor = self.db.execute("SELECT * FROM books")
        return cursor.fetchall()

from PyQt5.QtWidgets import *
from controller.user_controller import UserController
from view.register_window import RegisterWindow

class LoginWindow(QDialog):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("登录")
        self.controller = UserController()
        # ... 布局代码 ...
        self.user = None

    def login(self):
        user = self.controller.login(self.edit_user.text(), self.edit_pass.text())
        if user:
            self.user = user
            self.accept()  # 关闭对话框并返回 True
        else:
            QMessageBox.warning(self, "错误", "登录失败")

    def open_register(self):
        RegisterWindow().exec_()  # 模态弹出注册窗口

class RegisterWindow(QDialog):
    def __init__(self):
        super().__init__()
        self.combo_role.addItems([Role.TEACHER, Role.STUDENT])  # 只能选教师或学生

    def register(self):
        self.controller.register(
            self.edit_user.text(),
            self.edit_pass.text(),
            self.combo_role.currentText()
        )
        QMessageBox.information(self, "成功", "注册成功")
        self.accept()

import sys
from PyQt5.QtWidgets import QApplication
from view.login_window import LoginWindow
# ... 导入各角色窗口 ...

if __name__ == "__main__":
    app = QApplication(sys.argv)
    login = LoginWindow()
    if login.exec_():  # 如果登录成功
        user = login.user
        if user.role == Role.ADMIN:
            window = AdminWindow(user)
        elif user.role == Role.TEACHER:
            window = TeacherWindow(user)
        else:
            window = StudentWindow(user)
        window.show()
        sys.exit(app.exec_())

# 在 TeacherWindow.__init__ 中
if PermissionManager.has_permission(user.role, Permission.DELETE_BOOK):
    btn_layout.addWidget(self.btn_delete)

pip install PyQt5

python main.py