- 安装
mysql-connector-python
- 导入代码并修改数据库连接信息
import mysql.connector
from mysql.connector import Error
class MySQLDatabase:
def __init__(self, host, user, password, db, port=3306):
self.connection = None
self.cursor = None
try:
self.connection = mysql.connector.connect(
host=host,
user=user,
password=password,
database=db,
port=port
)
self.cursor = self.connection.cursor()
except Error as e:
print(f"连接到MySQL数据库时出错:{e}")
def insert(self, query, values):
if self.cursor:
self.cursor.execute(query, values)
self.connection.commit()
return self.cursor.lastrowid
else:
print("操作失败:没有数据库连接。")
return None
def delete(self, query, values):
if self.cursor:
self.cursor.execute(query, values)
self.connection.commit()
return self.cursor.rowcount
else:
print("操作失败:没有数据库连接。")
return None
def update(self, query, values):
if self.cursor:
self.cursor.execute(query, values)
self.connection.commit()
return self.cursor.rowcount
else:
print("操作失败:没有数据库连接。")
return None
def select(self, query, values=None):
if self.cursor:
self.cursor.execute(query, values or ())
return self.cursor.fetchall()
else:
print("操作失败:没有数据库连接。")
return []
def __del__(self):
if self.cursor:
try:
self.cursor.close()
except Error as e:
print(f"关闭游标时出错:{e}")
if self.connection:
try:
self.connection.close()
except Error as e:
print(f"关闭数据库连接时出错:{e}")