❤️Python接口自动化，一文告诉你连接各大【数据库】建议收藏❤️

阅读原文时间：2021年09月28日阅读：1

前言
常见数据库
Mysql
Oracle
sql-server
PostgreSQL
MongoDB
Redis

相信很多小伙伴在使用python进行自动化测试的时候，都会涉及到数据库数据校验的问题，在前面的随笔中就已经有讲过连接mysql的使用，今天给大家汇总一下python对接几大常用的数据库操作的方法！众所周知，python有着极其丰富的第三方库，所以不管你使用的是关系型数据库(oracle，mysql, sqlserver等)，还是非关系型数据库(redis，mongoDB)，python都有与之对应的第三方库。下面给大家一一介绍。

Python 标准数据库接口为 Python DB-API，Python DB-API为开发人员提供了数据库应用编程接口。

Python 数据库接口支持非常多的数据库，你可以选择适合你项目的数据库：

GadFly

mSQL

MySQL

PostgreSQL

Microsoft SQL Server 2000

Informix

Interbase

Oracle

Sybase

MySQL是一个关系型数据库，MySQL由于性能高、成本低、可靠性好，已经成为最流行的开源数据库。最开始由瑞典的MySQL AB公司开发，后来被甲骨文公司（Oracle）收购。
安装_1：pip install mysql
安装_2:pip install PyMySQL

# -*- coding: utf-8 -*-
# @Author : 一凡

import MySQLdb

# 打开数据库连接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )

# 使用cursor()方法获取操作游标
cursor = db.cursor()

# 使用execute方法执行SQL语句
cursor.execute("SELECT VERSION()")

# 使用 fetchone() 方法获取一条数据
data = cursor.fetchone()

print "Database version : %s " % data

# 关闭数据库连接
db.close()

python 对接 Oracle 数据库，使用的第三方库为 cx_Oracle
安装： pip install cx_Oracle

-- coding: utf-8 --

@Author : 一凡

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)
python 对接 sql-server数据库，使用的第三方库为 pymssql 。
安装： pip install pymssql

-- coding: utf-8 --

@Author : 一凡

import pymssql

第一步连接到数据库

con = pymssql.connect(host='xxx', # 数据库的地址
user='xxx', # 登录数据库的账号
password='xxxx', # 登录数据库的密码
database='xxx') # 库名

第二步创建游标

cur = con.cursor()

第三步执行sql语句

sql = 'SELECT * FROM user;'
cur.execute(sql)
python 对接 PostgreSQL数据库，使用的第三方库为 psycopg2 。
安装： pip install psycopg2

# -*- coding: utf-8 -*-
# @Author : 一凡

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）

python 对接MongoDB数据库，使用的第三方库为 pymongo 。
安装： pip install pymongo

-- coding: utf-8 --

@Author : 一凡

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()
python 对接Redis数据库，使用的第三方库为 redis 。
安装： pip install redis

-- coding: utf-8 --

@Author : 一凡

import redis

连接数据库

st = redis.StrictRedis(
host='localhost', # 数据库地址
port='6379', # 端口：
db='testdata', # 库名：
)

redis操作的命令，对应st对象的方法

比如在数据库中创建一条键为test的数据，往里面添加3个元素

st.lpush('test',11,22,33)