你可以使用 Python 的 pymysql
库来连接和操作 MySQL 数据库。你可以使用如下的代码来实现一个类,用来动态切换 MySQL 数据源:
import pymysql
class MySQLClient:
def __init__(self, host, user, password, database):
self.host = host
self.user = user
self.password = password
self.database = database
def connect(self):
self.connection = pymysql.connect(
host=self.host,
user=self.user,
password=self.password,
database=self.database
)
self.cursor = self.connection.cursor()
def close(self):
self.cursor.close()
self.connection.close()
def change_database(self, database):
self.database = database
self.connect()
使用这个类时,你可以先初始化一个 MySQLClient 对象,然后使用 connect
方法来连接到 MySQL 数据库。当你想要切换数据库时,可以使用 change_database
方法来更新数据库名称并重新连接。
例如:
mysql_client = MySQLClient('localhost', 'root', 'password', 'database1')
mysql_client.connect()
# 在 database1 中执行 SQL 语句
mysql_client.cursor.execute('SELECT * FROM table1')
# 切换到 database2
mysql_client.change_database('database2')
# 在 database2 中执行 SQL 语句
mysql_client.cursor.execute('SELECT * FROM table2')
mysql_client.close()
注意:在使用完 MySQLClient 对象之后,应该调用 close
方法来关闭数据库连接。