Python连接数据库：宿舍管理系统的后端存储

Python连接数据库：宿舍管理系统的后端存储

在宿舍管理系统中，我们使用MySQL数据库来存储和管理宿舍与学生的数据。Python提供了多种方式来连接和操作数据库，本系统使用mysql-connector-python包来实现数据库连接和数据操作。

建立数据库连接

安装mysql-connector-python

首先，需要安装mysql-connector-python包，可以通过pip进行安装：

pip install mysql-connector-python

创建数据库连接
Python
复制
import mysql.connector
from mysql.connector import Error

class DormManager:
    def __init__(self):
        self.connection = self._create_connection()
        self._init_tables()
        self.dorms = []
        self.students = []
        self._load_data()

    def _create_connection(self):
        try:
            return mysql.connector.connect(
                host="localhost",
                user="root",
                password="123456",
                database="db4",
                auth_plugin='mysql_native_password'
            )
        except Error as e:
            print(f"数据库连接失败: {e}")
            exit(1)


def _init_tables(self):
        with self.connection.cursor() as cursor:
            cursor.execute("""
                CREATE TABLE IF NOT EXISTS tb_dorm (
                    id INT PRIMARY KEY AUTO_INCREMENT,
                    max_number INT,
                    people_number INT DEFAULT 0,
                    area DOUBLE,
                    building INT
                ) AUTO_INCREMENT=20000000
            """)
            cursor.execute("""
                CREATE TABLE IF NOT EXISTS tb_student (
                    id INT PRIMARY KEY AUTO_INCREMENT,
                    dorm_id INT,
                    name VARCHAR(20),
                    gender VARCHAR(5),
                    class VARCHAR(5),
                    FOREIGN KEY (dorm_id) REFERENCES tb_dorm(id)
                ) AUTO_INCREMENT=10000000
            """)
            self.connection.commit()