这次封装数据库，就算不是为了自动化测试，数据库操作也必须要会，为什么？因为在很多时候，修复现网用户数据都是通过脚本实现的，如果不用其他语言，那么python在这一块的优势就十分明显了。
'''
Created on 2019年10月10日
@author: qguan
'''
import pymysql
from utils.HandleLogging import log as logging
mysql_info = {"host": '127.0.0.1',
"port": 3306,
"user": 'root',
"passwd": 'root',
"db": 'mysql',
"charset": 'utf8'}
class HandleMySQL(object):
'''
classdocs
'''
def __init__(self):
'''
初始化mysql数据库信息，创建连接游标
'''
self.db_info = mysql_info
self.logger = logging
try:
self.conn = pymysql.connect(host=self.db_info['host'],
port=self.db_info['port'],
user=self.db_info['user'],
passwd=self.db_info['passwd'],
db=self.db_info['db'],
charset=self.db_info['charset'],
cursorclass=pymysql.cursors.DictCursor)
self.logger.info("success!")
except Exception as a:
self.logger.info("failture:%s" % a)
raise a
else:
self.logger.info("create cursor!")
self.cursor = self.conn.cursor()
def __call__(self, sql, arg=None, is_more=False):
'''
调用数据库执行方法
:param sql:执行的sql语句，str类型
:param arg:参数必须是序列类型：list、str、tuple
:param is_more:是否查询显示更多数据，通常根据sql的limit来确定
:return:
'''
self.mysql_execute(sql, arg)
if is_more:
result = self.cursor.fetchall()
else:
result = self.cursor.fetchone()
# 不合适，如果还有sql在循环执行，在这里关闭，会造成sql执行失败
# self.mysql_close()
return result
def mysql_execute(self, sql, arg=None):
'''郑州人流医院哪家好 http://mobile.zhongyuan120.com/
封装自己执行sql语句的方法
'''
try:
self.cursor.execute(sql)
except Exception as a:
self.conn.rollback() # sql回滚处理
self.logger.info("回滚sql,异常如下：{}".format(a))
raise a
else:
self.conn.commit() # sql提交
def mysql_getrows(self, sql):
'''
返回查询结果集
:return: 返回的是一个嵌套字典的元组
'''
try:
self.cursor.execute(sql) # arg必须是序列类型
except Exception as a:
self.logger.info("执行SQL语句出现异常：%s" % a)
raise a # 直接抛出异常，不在执行后面的语句
else:
rows = self.cursor.fetchall()
return rows
def mysql_getrow(self, sql):
'''
返回查询结果集
return:结果是个字典
'''
try:
self.cursor.execute(sql) # arg必须是序列类型
except Exception as a:
self.logger.info("执行SQL语句出现异常：%s" % a)
raise a
else:
row = self.cursor.fetchone()
return row
def mysql_close(self):
'''
关闭数据库连接，关闭游标
'''
try:
self.conn.close()
self.cursor.close()
self.logger.info("正常关闭mysql连接!")
except Exception as a:
self.logger.info("关闭mysql连接池失败，异常如下：{}".format(a))
raise a
if __name__ == '__main__':
mq = HandleMySQL()
sql = 'SELECT *from user limit 3'
# print(mq(sql, is_more=True))
mq.mysql_close()