0
点赞
收藏
分享

微信扫一扫

用pymysql封装连接mysql数据库的工具类【接口自动化框架设计系列】【多测师】

#coding=utf-8
"""
===========================
Author:多测师_王sir
Time:2020/5/20 17:24
Company:上海多测师信息有限公司
===========================
"""

"""
查询
"""

import pymysql
from common.handleconfig import conf


class DB:
def __init__(self):
# 创建一个连接对象
self.conn = pymysql.connect(host=conf.get("db", "host"),
port=conf.getint("db", "port"),
user=conf.get("db", "user"),
password=conf.get("db", "pwd"),
charset=conf.get("db", "charset"),
cursorclass=pymysql.cursors.DictCursor
)
# 创建一个游标
self.cur = self.conn.cursor()

def find_one(self, sql):
"""获取查询出来的第一条数据"""
# 执行查询语句
self.conn.commit()
self.cur.execute(sql)
data = self.cur.fetchone()
return data

def find_all(self, sql):
"""获取查询出来的所有数据"""
self.conn.commit()
self.cur.execute(sql)
data = self.cur.fetchall()
return data

def close(self):
"""关闭游标,断开连接"""
self.cur.close()
self.conn.close()

 



举报

相关推荐

0 条评论