python 对数据库的方法封装

一、做接口测试的时候，很多时候都需要查询数据库来验证结果

以下是对数据库操作的方法封装

import pymysql import json """ 通用封装查询 param sql: param args: return:返回查询结果
（（），（）） """ class MySql(object): # 链接数据库 def conn(self): # 创建新链接 self.conn =
pymysql.connect(host='www.baidu.com', user='user', password='password',
database='database', charset='utf8') # 获取 cursor 对象并以自定义格式为字典 self.cursor =
self.conn.cursor(cursor=pymysql.cursors.DictCursor) # cursor = conn.cursor() #
执行完毕返回的结果集默认以元组显示 return self.conn, self.cursor # 关闭Cursor对象 def
close_conn(self, conn, cursor): cursor.close() conn.close() def select(self,
sql): conn, cursor = MySql().conn() cursor.execute(sql) # 读取，执行 sql 后的所有数据 cds
= cursor.fetchall() # json 格式化读取到的数据 rest = json.dumps(cds, default=str,
ensure_ascii=False) self.close_conn(conn, cursor) return rest

• « 上一篇：mysql为什么使用B+树
• » 下一篇：vuex的使用