pymysql ETL项目定义一个类下面数据库工具开发中切换数据库的代码
时间: 2024-05-15 15:19:19 浏览: 83
以下是一个基于pymysql的数据库工具类,其中包含了切换数据库的方法:
```python
import pymysql
class DatabaseTool:
def __init__(self, host, port, user, password, database):
self.conn = pymysql.connect(host=host, port=port, user=user, password=password, database=database)
self.cursor = self.conn.cursor()
def switch_database(self, database):
self.conn.select_db(database)
def execute_query(self, query):
self.cursor.execute(query)
result = self.cursor.fetchall()
return result
def execute_update(self, query):
self.cursor.execute(query)
self.conn.commit()
def close_connection(self):
self.cursor.close()
self.conn.close()
```
在上述代码中,`switch_database`方法接受一个数据库名作为参数,然后使用`select_db`方法切换到对应的数据库。其他方法可以用来执行查询和更新操作,并且在使用完后需要调用`close_connection`方法关闭连接。
阅读全文