news 2026/6/21 21:31:24

【接口测试】5_PyMySQL模块 _数据库工具类封装

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
【接口测试】5_PyMySQL模块 _数据库工具类封装

文章目录

  • 一、封装的目的
  • 二、封装实现(重点)
  • 三、搭建框架
  • 四、案例(重点)
    • 4.1 设置私有方法
    • 4.2 实现类方法
      • 4.2.1 查询一条记录
      • 4.2.2 增删改数据
    • 4.3 完整代码实现
  • 五、小结

一、封装的目的

  • 将 数据库常用的操作,封装成类中的方法。 将来在使用时,只需提供 要执行的SQL语句,给这个方法,即可执行。
  • 使用者,不需要关心:建立连接、创建游标、关闭游标、关闭连接

二、封装实现(重点)

类方法,可以用类名直接调用,添加装饰器。

类方法:create_conn()//获取数据库连接 类方法:close_conn()//关闭数据库连接 类方法:query_one(sql)//查询一条数据 类方法:db_uid(sql)//执行数据库增删改 要求:定义一个DBTools工具类,对外提供以上方法

三、搭建框架

# 定义 数据库工具类classDBTools(object):# 创建连接defcreate_conn(self):pass# 查一条记录defquery_one(self,sql):pass# 创建连接# 创建游标# 关闭游标# 关闭连接# 增删改记录defdb_uid(self,sql):pass# 创建连接# 创建游标# 关闭游标# 关闭连接

四、案例(重点)

idtitlepub_datereadcommentis_delete
1射雕英雄传1960-05-0112340
2天龙八部1966-07-2437400
3笑傲江湖1995-12-2423800

4.1 设置私有方法

让创建连接方法 create_conn( ) 对用户不可见(对用户没用),但是又不影响后面查询和修改方法使用。

—>设置私有方法 __create_conn( )

# 定义 数据库工具类classDBTools(object):# 创建连接 - 类方法。可以直接使用类名调用!@classmethoddef__create_conn(cls):conn=pymysql.connect(host="172.16.28.4",port=3306,user="root",password="root",database="books",charset="utf8")# 不能遗漏(创建连接之后,要返回结果)returnconn

4.2 实现类方法

4.2.1 查询一条记录

importpymysql# 定义 数据库工具类classDBTools(object):# 创建连接 - 类方法。可以直接使用类名调用!@classmethoddef__create_conn(cls):conn=pymysql.connect(host="172.16.28.4",port=3306,user="root",password="root",database="books",charset="utf8")# 不能遗漏(创建连接之后,要返回结果)returnconn# 查一条记录 - 封装为类方法,方便调用@classmethoddefquery_one(cls,sql):my_conn=Nonemy_cursor=Noneres=Nonetry:# 创建连接。借助类名,调用 类方法 create_conn。my_conn=DBTools.__create_conn()# 接收 return conn返回的结果,给my_conn# 创建游标my_cursor=my_conn.cursor()# 执行 sql 语句,做查询my_cursor.execute(sql)# 提取一条记录res=my_cursor.fetchone()exceptExceptionaserr:print("执行查询SQL失败:",str(err))finally:# 关闭游标my_cursor.close()# 关闭连接my_conn.close()# 返回查询结果returnresif__name__=='__main__':result=DBTools.query_one("select * from t_book;")print("查询语句的结果:",result)

4.2.2 增删改数据

importpymysql# 定义 数据库工具类classDBTools(object):# 创建连接 - 类方法。可以直接使用类名调用!@classmethoddef__create_conn(cls):conn=pymysql.connect(host="172.16.28.4",port=3306,user="root",password="root",database="books",charset="utf8")# 不能遗漏returnconn# 增删改记录@classmethoddefdb_uid(cls,sql):my_conn=Nonemy_cursor=Nonetry:# 创建连接.my_conn=DBTools.__create_conn()# 创建游标my_cursor=my_conn.cursor()# 执行 增删改 语句my_cursor.execute(sql)print("Affected rows:",my_conn.affected_rows())# 提交事务my_conn.commit()exceptExceptionaserr:print("执行 增删改 SQL 失败:",str(err))# 回滚事务my_conn.rollback()finally:# 关闭游标my_cursor.close()# 关闭连接my_conn.close()if__name__=='__main__':DBTools.db_uid("update t_book set `read` = 100 where id = 3;")

4.3 完整代码实现

文件:py08_db_tools.py

importpymysql# 定义 数据库工具类classDBTools(object):# 创建连接 - 类方法。可以直接使用类名调用!@classmethoddef__create_conn(cls):conn=pymysql.connect(host="172.16.28.4",port=3306,user="root",password="root",database="books",charset="utf8")# 不能遗漏returnconn# 查一条记录 - 封装为类方法,方便调用@classmethoddefquery_one(cls,sql):my_conn=Nonemy_cursor=Noneres=Nonetry:# 创建连接, 借助类名,调用 类方法 create_connmy_conn=DBTools.__create_conn()# 创建游标my_cursor=my_conn.cursor()# 执行 sql 语句,做查询my_cursor.execute(sql)# 提取一条记录res=my_cursor.fetchone()exceptExceptionaserr:print("执行查询SQL失败:",str(err))finally:# 关闭游标my_cursor.close()# 关闭连接my_conn.close()# 返回查询结果returnres# 增删改记录@classmethoddefdb_uid(cls,sql):my_conn=Nonemy_cursor=Nonetry:# 创建连接my_conn=DBTools.__create_conn()# 创建游标my_cursor=my_conn.cursor()# 执行 增删改 语句my_cursor.execute(sql)print("Affected rows:",my_conn.affected_rows())# 提交事务my_conn.commit()exceptExceptionaserr:print("执行 增删改 SQL 失败:",str(err))# 回滚事务my_conn.rollback()finally:# 关闭游标my_cursor.close()# 关闭连接my_conn.close()if__name__=='__main__':result=DBTools.query_one("select * from t_hero;")print("查询语句的结果:",result)DBTools.db_uid("update t_book set `read` = 100 where id = 3;")

五、小结

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/12 15:56:15

GitHub Actions自动化测试PyTorch-CUDA镜像兼容性

GitHub Actions自动化测试PyTorch-CUDA镜像兼容性 在深度学习项目日益复杂的今天,一个常见的工程痛点浮出水面:为什么代码在我本地能跑,在CI里却报错?更典型的情况是——模型训练脚本在开发者机器上运行顺畅,但一旦进入…

作者头像 李华
网站建设 2026/6/18 13:30:14

电子电器架构 --- 主机厂的安全理念(上)

我是穿拖鞋的汉子,魔都中坚持长期主义的汽车电子工程师。 老规矩,分享一段喜欢的文字,避免自己成为高知识低文化的工程师: 做到欲望极简,了解自己的真实欲望,不受外在潮流的影响,不盲从,不跟风。把自己的精力全部用在自己。一是去掉多余,凡事找规律,基础是诚信;二是…

作者头像 李华
网站建设 2026/6/16 6:43:02

GitHub项目README模板:突出PyTorch-CUDA环境优势

GitHub项目README优化:构建高效的PyTorch-CUDA开发环境 在深度学习项目中,最让人头疼的往往不是模型设计,而是环境配置。你是否也经历过这样的场景:论文复现代码跑不通,排查半天才发现是CUDA版本和PyTorch不匹配&#…

作者头像 李华
网站建设 2026/6/18 23:41:41

【毕业设计】基于springBoot的高校毕业生公职资讯系统的设计与实现(源码+文档+远程调试,全bao定制等)

博主介绍:✌️码农一枚 ,专注于大学生项目实战开发、讲解和毕业🚢文撰写修改等。全栈领域优质创作者,博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围:&am…

作者头像 李华
网站建设 2026/6/15 12:03:09

【计算机毕业设计案例】基于Springboot的克州旅游网站的设计与实现精品路线推荐、行程规划、价格查询(程序+文档+讲解+定制)

博主介绍:✌️码农一枚 ,专注于大学生项目实战开发、讲解和毕业🚢文撰写修改等。全栈领域优质创作者,博客之星、掘金/华为云/阿里云/InfoQ等平台优质作者、专注于Java、小程序技术领域和毕业项目实战 ✌️技术范围:&am…

作者头像 李华
网站建设 2026/6/15 14:59:52

Git Hooks自动化检查PyTorch代码提交规范

Git Hooks自动化检查PyTorch代码提交规范 在深度学习项目日益复杂的今天,一个常见的场景是:团队成员提交的代码在本地运行正常,却在CI流水线或生产环境中因环境差异、风格不一致甚至低级错误而失败。尤其是使用 PyTorch 进行 GPU 加速训练时…

作者头像 李华