Python介面自動化之pymysql資料庫操作

在自動化過程中，我們需要查詢資料庫，校驗結果是否正確，比如充值完成之後，需要查詢資料庫，檢視充值是否成功。

以下主要介紹，pymysql安裝、操作流程、語法基礎及封裝運算元據庫類。

一pymysql介紹及安裝

mysql應該說是如今使用最為普遍的資料庫了，沒有之一，而python作為最為流行的語言之一，自然少不了與mysql打交道，其中pymysql就是使用最多的工具庫。

02 pymysql安裝

方式一：使用命令安裝

pip install pymysql

import pymysql

建立資料庫連線：使用pymysql的connect()方法連線資料，返回連線物件；

使用連線物件建立游標物件（用於操作sql）;

準備寫sql語句（select * from student）;

使用游標物件執行sql;

查詢資料使用游標獲取;

關閉游標（先）和資料庫連線（後）。

02 pymysql模組說明

▌connection物件

表示：conn=connect(引數列表)

作用：用於建立與資料庫的連線；

建立物件：呼叫connect()方法；

引數列表：

▌物件的方法

物件方法如下：

▌cursor物件

游標（cursor）就是游動的標識,通俗的說，一條sql取出對應n條結果資源的介面/控制代碼，就是游標，沿著游標可以一次取出一行。

▌物件的屬性

三pymysql語法基礎

import pymysql

# 連線資料庫
conn = pymysql.connect(host='127.0.0.1', user='itester', password='123456',
database='itester', charset='utf8')
# 建立游標
cursor = conn.cursor()
# 執行sql語句
sql = 'select * from user limit 3;'
res = cursor.execute(sql)
# 獲取查詢結果的1條資料
data = cursor.fetchone()
print(data)
# 關閉游標連線
cursor.close()
# 關閉資料庫連線
conn.close()

2.運算元據庫先建立游標；

3.執行指定的sql語句，如果涉及到增、刪、改資料庫必須要conn.commit()，提交事務

4.查詢獲取資料條數有三種方法fetchone、fetchmany、fetchall。

5.需要注意的是，fetch獲取的資料預設是元組，如果想要字典型別，

cursor=pymysql.cursors.dictcursor；

6.先關閉游標，後關閉資料庫連線；

四封裝資料庫類

在實際專案中，很多地方都有用到資料庫的操作，所以需要將資料庫相關操作進行封裝，方便其他模組呼叫。

如下，在common目錄下，新建檔案db_handler.py 用於封裝資料庫操作。

# 連線資料庫伺服器

self.conn = pymysql.connect(host=host, port=port, user=user,password=password,

database=database,cursorclass=pymysql.cursors.dictcursor,

charset=charset,**kwargs)

# 獲取游標

self.cursor = self.conn.cursor()

def query(self, sql, args=none,one=true):

self.cursor.execute(sql, args)

# 提交事務

self.conn.commit()

if one:

return self.cursor.fetchone()

else:

return self.cursor.fetchall()

def close(self):

self.cursor.close()

self.conn.close()

if __name__ == "__main__":

db = dbhandler(host='127.0.0.1', port=3306,

user='itester', password='123456',

database='itester', charset='utf8')

sql = 'select * from user limit 1;'

data = db.query(sql)

print(data)

總結：本文主要介紹pymysql安裝、操作流程、語法基礎及封裝運算元據庫類。

python之requests介面自動化

在進行介面自動化測試時，有好多介面都基於登陸介面的響應值來關聯進行操作的，在次之前試了很多方法，都沒有成功，其實很簡單用session來做。coding utf 8 import requests 在登陸模組建立乙個全域性session，在其他介面操作時帶入登陸時的session，保持session...

python介面自動化之token

python介面自動化token 匯入request包 import requests 匯入json包 import json 設定訊息頭資訊 header 設定引數資訊 params 將設定的引數資訊轉換為json格式的資料 data json.dumps params 設定url資料 url 設...

Python 介面自動化

python環境搭建 第三方庫 requests安裝 pip install requests 網爛時可能安裝失敗，嗯 pip安裝第三方庫提示版本較低時需要更新pip的版本 命令 python m pip install upgrade pip 檢視安裝的第三方庫 pip list requests...