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