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

文章目录

一、封装的目的

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

二、封装实现(重点)

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

python 复制代码
类方法:create_conn()     // 获取数据库连接

类方法:close_conn()      // 关闭数据库连接

类方法:query_one(sql)    // 查询一条数据

类方法:db_uid(sql)       // 执行数据库增删改

要求:定义一个DBTools工具类,对外提供以上方法

三、搭建框架

python 复制代码
# 定义 数据库工具类
class DBTools(object):
    
    # 创建连接
    def create_conn(self):
        pass

    # 查一条记录
    def query_one(self, sql):
        pass
        # 创建连接
        # 创建游标
        # 关闭游标
        # 关闭连接

    # 增删改记录
    def db_uid(self, sql):
        pass
        # 创建连接
        # 创建游标
        # 关闭游标
        # 关闭连接

四、案例(重点)

id title pub_date read comment is_delete
1 射雕英雄传 1960-05-01 12 34 0
2 天龙八部 1966-07-24 37 40 0
3 笑傲江湖 1995-12-24 23 80 0

4.1 设置私有方法

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

--->设置私有方法 __create_conn( )

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

4.2 实现类方法

4.2.1 查询一条记录

python 复制代码
import pymysql

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

# 查一条记录 - 封装为类方法,方便调用
    @classmethod
    def query_one(cls, sql):
        my_conn = None
        my_cursor = None
        res = None
        try:
            # 创建连接。借助类名,调用 类方法 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()
        except Exception as err:
            print("执行查询SQL失败:", str(err))
        finally:
            # 关闭游标
            my_cursor.close()
            # 关闭连接
            my_conn.close()
            # 返回查询结果
            return res
        
        
if __name__ == '__main__':
	result = DBTools.query_one("select * from t_book;")
    print("查询语句的结果:", result)

4.2.2 增删改数据

python 复制代码
import pymysql

# 定义 数据库工具类
class DBTools(object):
    # 创建连接 - 类方法。可以直接使用类名调用!
    @classmethod
    def __create_conn(cls):
        conn = pymysql.connect(host="172.16.28.4", port=3306, user="root", password="root",
                               database="books", charset="utf8")
        # 不能遗漏
        return conn

# 增删改记录
    @classmethod
    def db_uid(cls, sql):
        my_conn = None
        my_cursor = None
        try:
            # 创建连接.
            my_conn = DBTools.__create_conn()  
            # 创建游标
            my_cursor = my_conn.cursor()
            # 执行 增删改 语句
            my_cursor.execute(sql)
            print("Affected rows:", my_conn.affected_rows())
            # 提交事务
            my_conn.commit()
        except Exception as err:
            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

python 复制代码
import pymysql

# 定义 数据库工具类
class DBTools(object):
    # 创建连接 - 类方法。可以直接使用类名调用!
    @classmethod
    def __create_conn(cls):
        conn = pymysql.connect(host="172.16.28.4", port=3306, user="root", password="root",
                               database="books", charset="utf8")
        # 不能遗漏
        return conn

    # 查一条记录 - 封装为类方法,方便调用
    @classmethod
    def query_one(cls, sql):
        my_conn = None
        my_cursor = None
        res = None
        try:
            # 创建连接, 借助类名,调用 类方法 create_conn
            my_conn = DBTools.__create_conn()
            # 创建游标
            my_cursor = my_conn.cursor()
            # 执行 sql 语句,做查询
            my_cursor.execute(sql)
            # 提取一条记录
            res = my_cursor.fetchone()
        except Exception as err:
            print("执行查询SQL失败:", str(err))
        finally:
            # 关闭游标
            my_cursor.close()
            # 关闭连接
            my_conn.close()
            # 返回查询结果
            return res

    # 增删改记录
    @classmethod
    def db_uid(cls, sql):
        my_conn = None
        my_cursor = None
        try:
            # 创建连接
            my_conn = DBTools.__create_conn()
            # 创建游标
            my_cursor = my_conn.cursor()
            # 执行 增删改 语句
            my_cursor.execute(sql)
            print("Affected rows:", my_conn.affected_rows())
            # 提交事务
            my_conn.commit()
        except Exception as err:
            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;")

五、小结

相关推荐
小光学长2 小时前
基于ssm旅游管理系统的开发与设计z050cft7(程序+源码+数据库+调试部署+开发环境)带论文文档1万字以上,文末可获取,系统界面在最后面。
java·数据库·旅游
剑之所向2 小时前
MCU开机按键,怎么避免抖动造成的开机
数据库·单片机·mongodb
lightningyang3 小时前
渗透入门之SQL 注入(二)
数据库·sql·渗透·sql注入
四谎真好看3 小时前
MySQL 学习笔记(运维篇1)
运维·数据库·笔记·学习·mysql·学习笔记
ZLZQ_Yuan4 小时前
MyBatis、MyBatis-Plus
数据库·mybatis
、BeYourself4 小时前
PostgreSQL 向量扩展插件pgvector安装
数据库·postgresql
RPA机器人就选八爪鱼5 小时前
RPA在银行IT运维领域的应用场景与价值分析
大数据·运维·数据库·人工智能·机器人·rpa
廋到被风吹走5 小时前
【数据库】【Oracle】函数整理
数据库·oracle
冉冰学姐5 小时前
SSM校园二手物品交易系统051x4(程序+源码+数据库+调试部署+开发环境)带论文文档1万字以上,文末可获取,系统界面在最后面
数据库·ssm 框架·商品全流程管理