Python关于MySQL常用方法封装

import mysql.connector
class MySQL:
def __init__(self, host, user, password, database):
self.conn = mysql.connector.connect(
host=host,
user=user,
password=password,
database=database
)
self.cursor = self.conn.cursor()
def execute(self, sql, values=None):
if values:
self.cursor.execute(sql, values)
else:
self.cursor.execute(sql)
def fetchone(self):
return self.cursor.fetchone()
def fetchall(self):
return self.cursor.fetchall()
def query(self, sql, values=None):
self.execute(sql, values)
return self.fetchall()
def insert(self, table, data):
columns = ', '.join(data.keys())
placeholders = ', '.join(['%s'] * len(data))
sql = f'INSERT INTO {table} ({columns}) VALUES ({placeholders})'
self.execute(sql, tuple(data.values()))
self.conn.commit()
def update(self, table, data, condition):
set_values = ', '.join([f"{key} = '{value}'" for key, value in data.items()])
sql = f"UPDATE {table} SET {set_values} WHERE {condition}"
self.execute(sql)
self.conn.commit()
def delete(self, table, condition):
sql = f"DELETE FROM {table} WHERE {condition}"
self.execute(sql)
self.conn.commit()
def close(self):
self.cursor.close()
self.conn.close()

食用方法如下：

# 初始化MySQL类
db = MySQL('host', 'user', 'password', 'database')
# 执行SELECT语句
db.execute('SELECT * FROM users')
# 获取一个结果
print(db.fetchone())
# 获取所有结果
print(db.fetchall())
# 插入新记录
data = {'name': 'John Doe', 'email': 'johndoe@example.com'}
db.insert('users', data)
# 更新记录
data = {'name': 'Jane Doe'}
condition = "id = 1"
db.update('users', data, condition)
# 删除记录
condition = "id = 2"
db.delete('users', condition)
# 关闭连接
db.close()

最后

以上就是幸福哈密瓜为你收集整理的Python关于MySQL常用方法封装的全部内容，希望文章能够帮你解决Python关于MySQL常用方法封装所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。