pymysql封装
pymysql封装
-
- 安装PyMySQL库
- 基本操作-增删改查
-
- 查询操作
- 更新操作
- 插入数据操作
- 删除操作
- 简单封装数据库操作
-
- 代码示例：
安装PyMySQL库
本机环境：
- Python 3.7.0
- MySQL 5.7
使用Python来操作MySQL，需要用到 PyMySQL库
# 安装PyMySQL库
pip install PyMySQL
我这里安装的版本是 0.9.3。
基本操作-增删改查
Python操作MySQL的步骤如下：
- 建立数据库连接
- 通过 cursor() 创建游标对象（pymysql通过游标来执行sql和获取结果）
- 使用 execute() 执行sql
- 获取结果（查询）/提交事务（增删改）
- 关闭游标
- 关闭连接
**注意：**用Python进行增删改查操作之前，最好先检查下sql语句是否正确，确保没有错误之后，再放到Python代码中。
查询操作
import pymysql
# 查询操作
def select_db(select_sql):
"""查询"""
# 建立数据库连接
# root:[email protected]:3306/ebook_management_system?charset=utf8', pool_recycle=3600
db = pymysql.connect(
host="120.79.30.173",
port=3306,
user="root",
passwd="Mv2eISCe",
db="ebook_management_system"
)
# 通过 cursor() 创建游标对象，并让查询结果以字典格式输出
cur = db.cursor(cursor=pymysql.cursors.DictCursor)
# 使用 execute() 执行sql
cur.execute(select_sql)
# 使用 fetchall() 获取所有查询结果
data = cur.fetchall()
# 关闭游标
cur.close()
# 关闭数据库连接
db.close()
# 返回查询数据
return data
if __name__ == '__main__':
select_sql = 'SELECT * FROM user WHERE username="张三"'
print(select_db(select_sql))
在上面，创建游标时使用了 cursor=pymysql.cursors.DictCursor ，目的是为了让查询结果以 列表嵌套字典 的格式输出，以方便后续操作。如果不使用该参数，那么返回的查询结果将是 元组嵌套元组 的形式。
更新操作
import pymysql
def update_db(update_sql):
"""更新"""
# 建立数据库连接
db = pymysql.connect(
host="120.79.30.173",
port=3306,
user="root",
passwd="Mv2eISCe",
db="ebook_management_system"
)
# 通过 cursor() 创建游标对象
cur = db.cursor()
try:
# 使用 execute() 执行sql
cur.execute(update_sql)
# 提交事务
db.commit()
# 如果执行出错，捕获异常
except Exception as e:
print("操作出现错误：{}".format(e))
# 回滚所有更改
db.rollback()
# 无论是否成功提交事务，都关闭游标和数据库
finally:
# 关闭游标
cur.close()
# 关闭数据库连接
db.close()
在执行更新操作sql时，需要使用事务处理，即使用创建的连接进行提交事务 db.commit() ，使当前更改生效。当执行sql语句出现错误时，还应该支持事务回滚操作 db.rollback() 。
插入数据操作
# 插入数据
def insert_db(insert_sql):
"""插入"""
# 建立数据库连接
db = pymysql.connect(
host="120.79.30.173",
port=3306,
user="root",
passwd="Mv2eISCe",
db="ebook_management_system"
)
# 通过 cursor() 创建游标对象
cur = db.cursor()
try:
# 使用 execute() 执行sql
cur.execute(insert_sql)
# 提交事务
db.commit()
except Exception as e:
print("操作出现错误：{}".format(e))
# 回滚所有更改
db.rollback()
finally:
# 关闭游标
cur.close()
# 关闭数据库连接
db.close()
在执行新增操作sql时，需要像 更新操作 一样，使用事务来处理。
删除操作
def delete_db(delete_sql):
"""删除"""
# 建立数据库连接
db = pymysql.connect(
host="120.79.30.173",
port=3306,
user="root",
passwd="Mv2eISCe",
db="ebook_management_system"
)
# 通过 cursor() 创建游标对象
cur = db.cursor()
try:
# 使用 execute() 执行sql
cur.execute(delete_sql)
# 提交事务
db.commit()
except Exception as e:
print("操作出现错误：{}".format(e))
# 回滚所有更改
db.rollback()
finally:
# 关闭游标
cur.close()
# 关闭数据库连接
db.close()
在执行删除操作sql时，需要像 更新操作、插入操作 一样，使用事务来处理。
简单封装数据库操作
从上面的代码中，我们可以发现存在不少问题，比如增删改操作的代码除了执行的sql不同外，其他代码都是一样的；同时每个操作时都写了一次建立数据库连接的代码，也就是说每次操作都会建立和删除连接的操作。
针对上面的问题，我们完全可以对上面的增删改查操作进行简单的封装，把Python操作MySQL的代码封装到一个类中，只在创建对象实例的时候建立数据库连接，只在对象删除时关闭连接。
代码示例：
参考：https://www.cnblogs.com/wintest/p/12152687.html
import pymysql
class MysqlDb():
def __init__(self, host, port, user, passwd, db):
"""
建立数据库连接
:param host: 传入数据库主机地址
:param port: 传入数据库端口
:param user: 传入数据库用户名
:param passwd: 传入数据库登陆密码
:param db: 传入数据库名称
"""
# 建立数据库连接
self.conn = pymysql.connect(
host=host,
port=port,
user=user,
passwd=passwd,
db=db
)
# 通过 cursor() 创建游标对象，并让查询结果以字典格式输出
self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)
def __del__(self): # 对象资源被释放时触发，在对象即将被删除时的最后操作
# 关闭游标
self.cur.close()
# 关闭数据库连接
self.conn.close()
def select_db(self, sql):
"""
查询操作
:param sql: 传入查询sql语句
:return: 返回查询结果
"""
# 使用 execute() 执行sql
self.cur.execute(sql)
# 使用 fetchall() 获取查询结果
data = self.cur.fetchall()
# 返回查询数据
return data
def execute_db(self, sql):
"""
更新/插入/删除
:param sql: 传入数据库语句
:return: 没有返回值
"""
try:
# 使用 execute() 执行sql
self.cur.execute(sql)
# 提交事务
self.conn.commit()
except Exception as e:
print("操作出现错误：{}".format(e))
# 回滚所有更改
self.conn.rollback()