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

相信不少小夥伴在使用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)