接口自动化测试学习笔记8 -------pymysql数据库封装

操作数据库的步骤：        

1、建立数据库链接：

conn = pymysql.Connect(host=None,port=0,user=None,password='',charset='',database=None)

 2、建立游标：cursor = conn.cursor()

3、执行sql语句：cursor.execute(sql)

4、提交事务：conn.commit()

5、获取游标的结果：cursor.fetchone() 

6、关闭游标：cursor.close()

7、关闭连接：conn.close()

如何将动态变化的数据写入到sql语句中：

使用cursor.execute('select * from CLS where name = %s;' args='值')

import pymysql
from pymysql.cursors import DictCursor


class DBhandler:
    def __init__(self, host=None, port=None, user=None, password=None, charset='utf-8', database=None,
                 cursorclass=DictCursor):
        self.conn = pymysql.Connect(host=host, port=port, user=user, password=password, charset=charset,
                                    database=database, cursorclass=cursorclass)  # 建立数据库连接
        self.cursor = self.conn.cursor()  # 建立游标

    def quest(self, sql, args=None, one=True):
        self.cursor.execute(sql, args)  # 执行sql语句
        self.conn.commit()  # 提交事务
        if one:  # 获取一个游标结果
            return self.cursor.fetchone()
        else:  # 获取所有的游标结果
            return self.cursor.fetchall()

    def close_db(self):
        self.cursor.close()  # 关闭游标
        self.conn.close()  # 关闭数据库连接

cursorclass=DictCursor ：游标结果由元组类型 转换为 字典类型