【python接口自動化】- 對接各大數據庫

如果想從頭學起接口自動化測試，可以去看看這個系列的文章哦！

https://www.cnblogs.com/miki-peng/category/1671261.html

前言

​ 相信很多小伙伴在使用python進行自動化測試的時候，都會涉及到數據庫數據校驗的問題，在前面的隨筆中就已經有講過連接mysql的使用，今天給大家匯總一下python對接幾大常用的數據庫操作的方法！眾所周知，python有着極其豐富的第三方庫，所以不管你使用的是關系型數據庫(oracle，mysql, sqlserver等)，還是非關系型數據庫(redis，mongoDB)，python都有與之對應的第三方庫。下面給大家一一介紹。

Mysql

​ mysql應該是大家最熟悉的一種數據庫了，在python中使用的第三方庫為pymysql，這里不多介紹，前面的隨筆已單獨介紹，請移步傳送門。

Oracle

​ python 對接 Oracle 數據庫，使用的第三方庫為 cx_Oracle。安裝：pip install cx_Oracle。

import cx_Oracle                   


# 第一步 連接數據庫, 參數為'賬號/密碼/@ip:端口/庫名'
con = cx_Oracle.connect('user_name/password@127.0.0.1::8080/databases') 
# 第二步 創建游標
cur = con.cursor()                    
# 第三步 執行sql語句
sql = 'SELECT * FROM user;'
cur.execute(sql)

sql-server

​ python 對接 sql-server數據庫，使用的第三方庫為 pymssql。安裝：pip install pymssql。

import pymssql


# 第一步 連接到數據庫
con = pymssql.connect(host='xxx',   # 數據庫的地址
                    user='xxx',   # 登錄數據庫的賬號
                    password='xxxx',   # 登錄數據庫的密碼
                    database='xxx')    # 庫名
# 第二步 創建游標
cur = con.cursor()
# 第三步 執行sql語句
sql = 'SELECT * FROM user;'
cur.execute(sql)

PostgreSQL

​ python 對接 PostgreSQL數據庫，使用的第三方庫為 psycopg2。安裝：pip install psycopg2。

import psycopg2


# 第一步 連接到數據庫
con = psycopg2.connect(database="xxxxx",  # 庫名
                        user="xxxxx",	# 數據庫賬號
                        password="xxxxxx", 	# 數據庫密碼
                        host="xxxxxx", 	# 數據庫地址
                        port="5432")	# 數據庫端口號
# 第二步 創建游標
cur = con.cursor()
# 第三步 執行sql語句
sql = 'SELECT * FROM user;'
cur.execute(sql）

MongoDB

​ python 對接MongoDB數據庫，使用的第三方庫為 pymongo。安裝：pip install pymongo。

import pymongo


# 第一步：建立連接，數據庫地址，端口號
client = pymongo.MongoClient("localhost", 27017)
# 第二步：選取數據庫
db = client.test1
# 第三步：選取集合
stu = db.stu

# 第四步：執行相關操作
# 添加一條數據
data1 = {name:'test',age:18}
stu.insert_one(data1)
# 獲取一條數據
res = stu.find_one()

Redis

​ python 對接Redis數據庫，使用的第三方庫為 redis。安裝：pip install redis。

import redis 


# 連接數據庫
st = redis.StrictRedis(
                    host='localhost',	# 數據庫地址
                    port='6379',	# 端口：
                    db='testdata',	# 庫名：
)
# redis操作的命令，對應st對象的方法
# 比如在數據庫中創建一條鍵為test的數據，往里面添加3個元素
st.lpush('test',11,22,33)