Python实现接口测试总结--PyMySql库+封装

2023-12-21 22:29:33

import pymysql
# 封装数据库工具类
class DBUtil ( object ):
# 添加类属性
????????conn = None
????????@classmethod
????????def __get_conn ( cls ):
????????# 判断 conn 是否为空，如果是，创建
????????????????if cls . conn is None :
????????????????cls . conn = pymysql . connect ( host = "" , port = , user = "" ,
????????????????password = "" , database = "" , charset = "utf8" )
????????????????# 返回 非空连接
????????????????return cls . conn
????????@classmethod
????????def __close_conn ( cls ):
????????????????# 判断，conn 不为空，需要关闭。
????????????????if cls . conn is not None :
????????????????????????cls . conn . close ()
????????????????????????cls . conn = None
????????# 常用方法：查询一条结果
????????@classmethod
????????def select_one ( cls , sql ):
????????????????cursor = None
????????????????res = None
????????????????try :
????????????????????????# 获取连接
????????????????????????cls . conn = cls . __get_conn ()
????????????????????????# 获取游标
????????????????????????cursor = cls . conn . cursor ()
????????????????????????# 执行 查询语句
????????????????????????cursor . execute ( sql )
????????????????????????# 提取一条结果
????????????????????????res = cursor . fetchone ()
????????????????except Exception as err :
????????????????????????print ( " sql 错误： " , str ( err ))
????????????????finally :
????????????????????????# 关闭游标
????????????????????????cursor . close ()
????????????????????????# 关闭连接
????????????????????????cls . __close_conn ()
????????????????# 返回查询sql执行的 结果，
????????????????return res
????????# 常用方法：增删改
????????@classmethod
????????def uid_sql ( cls , sql ):
????????????????cursor = None
????????????????try :
????????????????????????# 获取连接
????????????????????????cls . conn = cls . __get_conn ()
????????????????????????# 获取游标
????????????????????????cursor = cls . conn . cursor ()
????????????????????????# 执行 uid 语句
????????????????????????cursor . execute ( sql )
????????????????????????print ( " 影响的行数： " , cls . conn . affected_rows ())
????????????????????????# 提交事务
????????????????????????cls . conn . commit ()
????????????????except Exception as err :
????????????????????????# 回滚事务
????????????????????????cls . conn . rollback ()
????????????????????????print ( " 增删改 SQL 失败： " , str ( err ))
????????????????finally :
????????????????????????# 关闭游标
????????????????????????cursor . close ()
????????????????????????# 关闭连接
????????????????????????cls . __close_conn ()

文章来源:https://blog.csdn.net/MaoLingHao/article/details/135140657
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。