基於Python的介面自動化實戰-基礎篇之pymysql模組操作資料庫

2021 年 1 月 5 日
筆記
介面自動化測試

引言

在進行功能或者介面測試時常常需要通過連接資料庫，操作和查看相關的數據表數據，用於構建測試數據、核對功能、驗證數據一致性，介面的資料庫操作是否正確等。因此，在進行介面自動化測試時，我們一樣繞不開介面和資料庫的交互，我們需要用程式碼連接資料庫，通過操作資料庫完成數據的準備、環境檢查以及資料庫斷言的功能。在python3中，使用python操作MySQL資料庫需要使用到第三方庫：pymysql，該模組本質上就是一個套接字的客戶端軟體包，它提供了諸多連接資料庫、操作資料庫表等一系列的方法。

一、PyMySQL安裝

1.在windows環境下安裝

由於python3.6及以上版本安裝python後就自帶了pip3，python版本低於3.6的，手動安裝下pip即可，因此可以直接使用pip安裝該模組

pip3 install pymysql

2.在linux環境下安裝

下載安裝pymysql的tar包，解壓後，進入解壓的目錄下，按如下安裝即可：

[root@localhost opt]#tar -xzvf PyMySQL-0.7.11.tar.gz
[root@localhost opt]#cd PyMySQL-0.7.11
[root@localhost PyMySQL-0.7.11]#python36 setup.py install

3.在PyCharm中安裝

在PyCharm中直接檢索該模組，並安裝，步驟如下：

二、Python操作資料庫

因為方便測試，我們首先在mysql資料庫創建測試表：userinfo，表資訊如下：

有了資料庫和數據表後，我們就可以導入pymysql模組，使用該模組下封裝的方法實現資料庫操作

資料庫連接

pymysql提供的方法如下：

1. 建立資料庫連接 conn = pymysql.connect()
2. 從連接建立操作游標 cur = conn.cursor()
3. 使用游標執行sql（讀/寫） cur.execute(sql)
4. 獲取結果（讀）/ 提交更改（寫） cur.fetchall() / conn.commit()
5. 關閉游標及連接 cur.close();conn.close()

程式碼示例：

import pymysql
# 建立連接
connection = pymysql.connect(host='119.29.78.234', port=3306, user='root', password='dhcc@2020', db='test123')
cursor = connection.cursor()       # 創建游標
cursor.execute("SELECT * FROM userinfo")   #使用execute()方法執行SQL語句
data = cursor.fetchall()  #使用fetall()獲取全部數據
print(data)
cursor.close()   #關閉游標和資料庫的連接
connection.close()

#運行結果
((1, '艾佛森', '123'), (2, '科比', '123'), (3, '詹姆斯', '123'), (4, '庫里', '123'))

什麼是游標？游標類似文件句柄，可以逐條的訪問資料庫執行結果集。pymysql中只能通過游標來執行sql和獲取結果

以上程式碼執行後，默認返回的是一個嵌套元組數據類型

資料庫增刪改查

查詢操作：

使用cur.execute(), 執行資料庫查詢後無返回的是影響的行數，而非查詢結果。我們要使用cur.fetchone()/cur.fetchmany()/cur.fetchall()來獲取查詢結果
cur.fetchone(): 獲取一條數據（同時獲取的數據會從結果集刪除），返回元組
cur.fetchmany(3): 獲取多條數據，返回嵌套元組
cur.fetchall(): 獲取所有數據，返回嵌套元組

程式碼示例：

查詢單條數據：

import pymysql
db_config = {
      "host":"119.29.78.234",
       "port":3306,
       "user":"root",
       "password":"dhcc@2020",
       "db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchone()    # fetchone()第一次只能查詢表中的首行數據
print(res)
res = cursor.fetchone()    # 第二次查詢下一行數據
print(res)
cursor.close()
db.close()

# 返回結果

((1, ‘艾佛森’, ‘123’))

（(2, ‘科比’, ‘123’)）

查詢多條數據：

import pymysql
db_config = {
      "host":"119.29.78.234",
       "port":3306,
       "user":"root",
       "password":"dhcc@2020",
       "db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchmany(3)   # 第一次查詢表中的前3行數據
print(res)
res = cursor.fetchmany(3)    # 第二次查詢下一個3行的數據
print(res)
cursor.close()
db.close()

#返回結果
((1, '艾佛森', '123'), (2, '科比', '123'), (3, '詹姆斯', '123'))
((4, '庫里', '123'),)

查詢所有數據：

import pymysql
db_config = {
      "host":"119.29.78.234",
       "port":3306,
       "user":"root",
       "password":"dhcc@2020",
       "db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchall()   # 第一次查詢表中的所有數據
print(res)
res = cursor.fetchall()    # 第二次查詢無數據
print(res)
cursor.close()
db.close()

#返回結果
((1, '艾佛森', '123'), (2, '科比', '123'), (3, '詹姆斯', '123'), (4, '庫里', '123'))
()

　默認都是返回元組的數據類型，看起來不太直觀，因此，在實例化時可以將游標設置成如下這樣，就可以返回字典類型的數據

cursor = db.cursor(cursor=pymysql.cursors.DictCursor)

#返回結果

[{‘username’: ‘艾佛森’, ‘id’: 1, ‘passwd’: ‘123’}, {‘username’: ‘科比’, ‘id’: 2, ‘passwd’: ‘123’}, {‘username’: ‘詹姆斯’, ‘id’: 3, ‘passwd’: ‘123’}, {‘username’: ‘庫里’, ‘id’: 4, ‘passwd’: ‘123’}]

增刪改操作：

在進行增刪改，執行修改資料庫的操作後不立即生效，使用連接conn.commit()提交後才生效，支援事物及回滾

程式碼示例：

import pymysql
db_config = {
      "host":"119.29.78.234",
       "port":3306,
       "user":"root",
       "password":"dhcc@2020",
       "db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES('克萊','123')"
#sql = "UPDATE userinfo SET username = '奧尼爾' WHERE username = '科比'"  # 修改數據
#sql = "DELETE FROM username WHERE username ='奧尼爾'"                    # 刪除數據
try:
    cursor.execute(sql)
    db.commit()
except Exception as e :   # 執行異常回滾
    db.rollback()
cursor.close()
db.close()
#或者在execute提供需要插入的數據
import pymysql
db_config = {
      "host":"119.29.78.234",
       "port":3306,
       "user":"root",
       "password":"dhcc@2020",
       "db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES(%s,%s)"
try:
    cursor.execute(sql,("克萊","123"))
    db.commit()
except Exception as e :
    db.rollback()
cursor.close()
db.close()
#批量插入數據
import pymysql
db_config = {
      "host":"119.29.78.234",
       "port":3306,
       "user":"root",
       "password":"dhcc@2020",
       "db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES(%s,%s)"
try:
    cursor.executemany(sql,[("韋德","123"),("字母哥","123")])
    db.commit()
except Exception as e :
    db.rollback()
cursor.close()
db.close()

封裝資料庫操作

由於經常要使用到資料庫操作，建議將所有資料庫操作封裝成公用的資料庫模組

封裝的程式碼示例如下：

import pymysql.cursors

class Operation_mysql(object):
    def __init__(self):
        # 建立連接
        db_config = {
            "host": "119.29.78.234",
            "port": 3306,
            "user": "root",
            "password": "dhcc@2020",
            "db": "test123"
        }
        self.connection = pymysql.connect(**db_config)
        # 創建游標
        self.cursor = self.connection.cursor()

    def execute_sql(self, sql):
        try:
            self.cursor.execute(sql)
            self.connection.commit()
        except Exception as e:  # 執行異常回滾
            db.rollback()

    def get_data(self):
        data = self.cursor.fetchone()
        #data = self.cursor.fetchall() # 查詢所有數據
        return data

    def close_mysql(self):
        # 關閉游標
        self.cursor.close()
        # 關閉資料庫連接
        self.connection.close()

這樣封裝後後續介面測試用例需要操作資料庫時，就可以引入該模組，實例化對象調用該模組下的方法。

Tags: 介面自動化測試