基於Python的接口自動化實戰-基礎篇之pymysql模塊操做數據庫

引言

        在進行功能或者接口測試時經常須要經過鏈接數據庫，操做和查看相關的數據表數據，用於構建測試數據、覈對功能、驗證數據一致性，接口的數據庫操做是否正確等。所以，在進行接口自動化測試時，咱們同樣繞不開接口和數據庫的交互，咱們須要用代碼鏈接數據庫，經過操做數據庫完成數據的準備、環境檢查以及數據庫斷言的功能。在python3中，使用python操做MySQL數據庫須要使用到第三方庫：pymysql，該模塊本質上就是一個套接字的客戶端軟件包，它提供了諸多鏈接數據庫、操做數據庫表等一系列的方法。

1、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中直接檢索該模塊，並安裝，步驟以下：

2、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()

這樣封裝後後續接口測試用例須要操做數據庫時，就能夠引入該模塊，實例化對象調用該模塊下的方法。