封装一个链接数据库的类
时间: 2024-05-01 17:22:21 浏览: 57
好的,以下是一个简单的链接数据库的类的Python代码:
```python
import pymysql
class Database:
def __init__(self, host, user, password, db):
self.host = host
self.user = user
self.password = password
self.db = db
self.connection = None
self.cursor = None
def connect(self):
try:
self.connection = pymysql.connect(
host=self.host,
user=self.user,
password=self.password,
db=self.db,
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor
)
self.cursor = self.connection.cursor()
print("Database connected successfully!")
except pymysql.DatabaseError as e:
print(f"Error connecting to database: {e}")
def disconnect(self):
if self.connection:
self.connection.close()
print("Database disconnected successfully!")
else:
print("No database connection found.")
def execute_query(self, query, params=None):
try:
self.cursor.execute(query, params)
result = self.cursor.fetchall()
return result
except pymysql.DatabaseError as e:
print(f"Error executing query: {e}")
```
使用方法:
1. 实例化类,传入数据库的主机名、用户名、密码和数据库名。
```python
db = Database('localhost', 'root', 'password', 'mydatabase')
```
2. 连接到数据库。
```python
db.connect()
```
3. 执行查询语句,可以选择传入参数。
```python
result = db.execute_query("SELECT * FROM users WHERE age > %s", (18,))
```
4. 断开与数据库的连接。
```python
db.disconnect()
```
注意:在使用该类时,需要先安装pymysql模块。