将代码封装成一个模块,命名为db.py
时间: 2023-06-14 11:04:58 浏览: 51
好的,以下是一个可能的实现,可以用于连接 MySQL 数据库并执行 SQL 查询操作:
```python
import mysql.connector
class Database:
def __init__(self, host, port, user, password, database):
self.host = host
self.port = port
self.user = user
self.password = password
self.database = database
def connect(self):
self.connection = mysql.connector.connect(
host=self.host,
port=self.port,
user=self.user,
password=self.password,
database=self.database
)
self.cursor = self.connection.cursor()
def disconnect(self):
self.connection.close()
def execute_query(self, query):
self.cursor.execute(query)
return self.cursor.fetchall()
```
使用示例:
```python
from db import Database
db = Database(host="localhost", port=3306, user="root", password="password", database="mydb")
db.connect()
result = db.execute_query("SELECT * FROM users")
print(result)
db.disconnect()
```
注意:以上代码仅为示例,实际使用时需要根据具体情况进行修改和优化。