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

一、封装的目的

• 将 数据库常用的操作，封装成类中的方法。 将来在使用时，只需提供 要执行的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# 创建连接# 创建游标# 关闭游标# 关闭连接

四、案例（重点）

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;")

五、小结