给出数据库连接的封装类,实现连接数据库、执行sql命令的函数
时间: 2024-05-02 13:19:34 浏览: 11
好的,以下是一个简单的数据库连接封装类:
```python
import pymysql
class DB:
def __init__(self, host, user, password, database):
self.host = host
self.user = user
self.password = password
self.database = database
self.connection = pymysql.connect(
host=self.host,
user=self.user,
password=self.password,
database=self.database,
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor
)
def execute_query(self, query):
with self.connection.cursor() as cursor:
cursor.execute(query)
result = cursor.fetchall()
return result
def execute_update(self, query):
with self.connection.cursor() as cursor:
cursor.execute(query)
self.connection.commit()
```
通过调用 `DB` 类的构造函数来连接数据库,然后可以使用 `execute_query` 函数执行查询命令,使用 `execute_update` 函数执行更新、插入、删除等命令。这个封装类使用的是 PyMySQL 库,可以连接 MySQL 数据库。