python接口測試—mysql數據庫操做

python操做mysql數據庫

一、安裝pymysql庫

在python中安裝pymysql第三方庫，經過pip install pymysql 命令進行安裝。

二、python操做mysql數據庫的步驟

• 創建數據庫鏈接，鏈接到數據庫 connect
• 建立一個遊標 cursor
• 執行對應的sql語句
• 獲取查詢到的數據

三、獲取查詢數據的兩個方式

• 方式一：獲取一條查詢數據

　　data_one = cur.fetchone() # 遊標名.fetchone()

 

• 方式二：獲取全部的查詢數據

 

　　data = cur.fetchall() # 遊標名.fetchall()

 

四、python操做mysql數據庫的具體代碼

import pymysql # 一、鏈接到數據庫
con = pymysql.connect(host='ip', port=3306, user='test', password='test', database='future', charset='utf8') # 二、建立一個遊標 cursor
cur = con.cursor() # 三、執行對應的sql語句 execute()
sql = 'select * from member '
# 查詢到的數據條數
res = cur.execute(sql) print(res) # 四、獲取查詢到的數據 # 方法一:獲取一條查詢數據
data_one = cur.fetchone() print(data_one) # 方法二：獲取全部的查詢數據
data = cur.fetchall() print(data)

五、注意

pymysql操做數據庫，默認開啓了事務，須要使用 con.commit() 進行提交事務，未提交的話，所作的操做不能生效。

六、封裝一個本身的mysql類

目的：

• 方便讀取數據
• 將數據庫配置提取到配置文件中，方便管理測試環境

import pymysql from common.config import conf class ReadMySQLData(object): def __init__(self): # 鏈接到數據庫
        self.con = pymysql.connect(host=conf.get('mysql','host'), port=conf.getint('mysql','port'), user=conf.get('mysql','user'), password=conf.get('mysql','password'), database=conf.get('mysql','database'), charset='utf8') # 建立遊標
        self.cur= self.con.cursor() def find_one(self,sql): ''' 查找並返回找到的第一條數據,返回的數據類型是元組 :param sql: :return: ''' self.cur.execute(sql) self.con.commit() return self.cur.fetchone() def find_all(self,sql): ''' 查找並返回找到的全部數據,返回的數據類型是元組 :param sql: :return: ''' self.cur.execute(sql) self.con.commit() return self.cur.fetchall() def close(self): ''' 關閉遊標，斷開數據庫鏈接 :return: ''' self.cur.close() self.con.close()