pymysql如何连接多个数据库

原创

mob64ca12f3496a 2024-10-07 03:41:45 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12f3496a的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用 PyMySQL 连接多个数据库的方案

在许多项目中，我们可能需要操作多个数据库，例如，一个应用可能需要从用户数据库读取用户信息，同时从订单数据库获取订单细节。使用 PyMySQL，我们可以方便地连接到多个数据库，以满足这样的需求。本文将详细介绍如何使用 PyMySQL 连接多个数据库，并提供具体的代码示例。

1. 安装 PyMySQL

首先，我们需要安装 PyMySQL。你可以使用以下命令来安装它：

pip install pymysql

2. 数据库连接基本结构

在连接多个数据库之前，我们需要确定每个连接的基本配置，包括主机、用户名、密码和数据库名。

3. 实现示例

我们将创建一个名为 DatabaseManager 的类，用于管理多个数据库连接。代码如下：

import pymysql

class DatabaseManager:
    def __init__(self, configs):
        self.connections = {}
        for db_name, config in configs.items():
            self.connections[db_name] = pymysql.connect(
                host=config['host'],
                user=config['user'],
                password=config['password'],
                database=config['database'],
                charset='utf8mb4',
                cursorclass=pymysql.cursors.DictCursor
            )

    def query(self, db_name, sql):
        with self.connections[db_name].cursor() as cursor:
            cursor.execute(sql)
            return cursor.fetchall()

    def close(self):
        for conn in self.connections.values():
            conn.close()

3.1 类图

下面是 DatabaseManager 类的类图，帮助理解类之间的关系。

classDiagram
    class DatabaseManager {
        +__init__(configs)
        +query(db_name, sql)
        +close()
    }

4. 使用示例

下面是如何使用 DatabaseManager 类来连接多个数据库并执行查询的示例：

# 数据库配置
configs = {
    'user_db': {
        'host': 'localhost',
        'user': 'root',
        'password': 'password',
        'database': 'user_database'
    },
    'order_db': {
        'host': 'localhost',
        'user': 'root',
        'password': 'password',
        'database': 'order_database'
    }
}

# 创建数据库管理器实例
db_manager = DatabaseManager(configs)

# 查询用户表
user_query = "SELECT * FROM users;"
users = db_manager.query('user_db', user_query)
print("Users:", users)

# 查询订单表
order_query = "SELECT * FROM orders;"
orders = db_manager.query('order_db', order_query)
print("Orders:", orders)

# 关闭所有连接
db_manager.close()