接口测试基础—pymysql模块

测试自动化顾问 2020-05-29

import pymysql  #导入pymysql模块from scripts.handle_yaml import HandleYaml #系统配置模块,关于数据库的配置都放在mysql区域名里面do_yaml = HandleYaml()class hander_sql:     def __init__(self):         # 第一步:创建连接对象         self.conn = pymysql.connect(host=do_yaml.get_data(‘mysql‘, ‘host‘),                                     user=do_yaml.get_data(‘mysql‘, ‘user‘),                                     password=do_yaml.get_data(‘mysql‘, ‘password‘),                                     db=do_yaml.get_data(‘mysql‘, ‘db‘),                                     port=do_yaml.get_data(‘mysql‘, ‘port‘),                                     charset=‘utf8‘,  # 这里只能写为utf8                                     #cursorclass设置返回的数据类型                                     # a.指定每一条记录为字典类型                                     # b.默认每一条记录为元祖类型                                     cursorclass=pymysql.cursors.DictCursor)         # 第二步:创建游标对象         self.one_a=self.conn.cursor()         pass     #创建查询一条数据的方法     def one_select(self,sql,arge=None):         #第三步:执行SQL语句         self.one_a.execute(sql,arge=None)#args=None 是占位符,当sql语句里面有 %s占位符的时候,就要传递         #第四步:获取值并提交         self.conn.connect()        #5 第五步:fetchone获取一条数据,如果游标类为DictCursor,则结果为字典类型         return self.one_a.fetchone()     # 创建查询一条数据的方法     def moer_select(self,sql):         self.one_a.execute(sql)         self.conn.connect()         # fetchall获取多条数据,如果游标类为DictCursor,则结果为嵌套字典的列表类型         return self.one_a.fetchall()     # 关闭的方法     def colc(self):         #  第六步:关闭连接         # a.必须先关游标         # b.再关连接         self.one_a.close()         self.conn.close()

相关推荐