【python介面自動化】- 對接各大資料庫

miki_peng發表於2021-02-18
Python大資料資料庫

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

Mysql

image-20201211171351461

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

Oracle

image-20201211171351461

​ 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

image-20201211171351461

​ 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

image-20201211171351461

​ 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

image-20201211171351461

​ 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

image-20201211171351461

​ 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)

相關文章