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