【Python接口测试】09_unit-MySQL数据库操作

Python与MySQL数据库连接及数据处理python3代码。如有疑问则可以私信或留言。

#!/bin/usr/env python3
# -*- coding: utf-8 -*-
# connect_db.py

import pymysql
import json

class ConnectDB:

    def __init__(self):
        # 数据库建立连接，字段分别时数据库IP、用户名、密码、库
        self.conn = pymysql.connect("127.0.0.1", "root", "159357", "InterfaceData", cursorclass=pymysql.cursors.DictCursor)
        # 创建游标操作数据库
        self.cur = self.conn.cursor()

    def select_data(self, sql):
        # 执行SQL语句 
        self.cur.execute(sql)
        # 获取执行SQL语句获得的第一条数据
        result = self.cur.fetchone()
        # 将返回结果转换为str格式
        result = json.dumps(result)
        return result


if __name__ == "__main__":
    conn = ConnectDB()
    res = conn.select_data('''SELECT * FROM test_conn''')
    print(type(res))