python连接mysql数据库通用类

在 Python 中创建一个通用的数据库连接类，用于连接 MySQL 数据库并执行常见的数据库操作，可以提高代码的复用性和可维护性。下面是一个示例：

```python

import mysql.connector

class MySQLDatabase:

    def __init__(self, host, user, password, database):

        self.host = host

        self.user = user

        self.password = password

        self.database = database

        self.connection = None

        self.cursor = None

    def connect(self):

        try:

            self.connection = mysql.connector.connect(

                host=self.host,

                user=self.user,

                password=self.password,

                database=self.database

            )

            self.cursor = self.connection.cursor()

            print("Connected to MySQL database")

        except mysql.connector.Error as error:

            print(f"Error connecting to MySQL database: {error}")

    def execute_query(self, query):

        try:

            self.cursor.execute(query)

            self.connection.commit()

            print("Query executed successfully")

        except mysql.connector.Error as error:

            print(f"Error executing query: {error}")

    def fetch_query(self, query):

        try:

            self.cursor.execute(query)

            rows = self.cursor.fetchall()

            return rows

        except mysql.connector.Error as error:

            print(f"Error fetching data: {error}")

            return None

    def close_connection(self):

        if self.connection.is_connected():

            self.cursor.close()

            self.connection.close()

            print("MySQL connection closed")

# Example of usage:

if __name__ == "__main__":

    # Initialize MySQLDatabase object

    db = MySQLDatabase(host="your_host",

                       user="your_username",

                       password="your_password",

                       database="your_database_name")

    # Connect to the database

    db.connect()

    # Example query execution

    create_table_query = """

        CREATE TABLE IF NOT EXISTS users (

            id INT AUTO_INCREMENT PRIMARY KEY,

            username VARCHAR(50) NOT NULL,

            email VARCHAR(100) NOT NULL

        )

    """

    db.execute_query(create_table_query)

    # Example data insertion

    insert_query = """

        INSERT INTO users (username, email)

        VALUES ('john_doe', 'john@example.com')

    """

    db.execute_query(insert_query)

    # Example data retrieval

    select_query = "SELECT * FROM users"

    rows = db.fetch_query(select_query)

    if rows:

        for row in rows:

            print(row)

    # Close the database connection

    db.close_connection()

```

在这个示例中：

- `MySQLDatabase` 类封装了 MySQL 数据库的连接和常用操作，包括连接、执行查询、执行插入操作以及关闭连接。

- `__init__` 方法用于初始化数据库连接参数。

- `connect` 方法用于建立数据库连接。

- `execute_query` 方法用于执行 SQL 查询或更新操作。

- `fetch_query` 方法用于执行查询并获取结果集。

- `close_connection` 方法用于关闭数据库连接。

根据自己的需求扩展这个类，添加更多的数据库操作方法或错误处理机制。记得替换示例中的 `your_host`、`your_username`、`your_password` 和 `your_database_name` 为实际的数据库连接信息。