10 python学习笔记-操作数据库

在功能、接口测试中，常常需要通过数据库的操作，来准备数据、检测环境及核对功能、接口的数据库操作是否正确。 在自动化测试中，就需要我们用代码连接数据库自动完成数据准备、

环境检查及数据库断言的功能。数据库有很多类型，关系型数据库和非关系数据库，这里主要介绍python怎么操作mysql、redis。

python操作数据库流程：

• 导入 API 模块，如pymysql。
• 建立数据库的连接：conn=pymysql.connect()
• 从连接建立游标（有游标才能操作数据库）：cur = conn.cursor()
• 使用游标执行sql语句(读/写)：cur.execute(sql)
• 获取结果（读）/提交更改（写）：cur.fetchall()/conn.commit()
• 关闭游标及数据库连接。cur.close()/conn.close()

一、python操作mysql数据库

　　使用Python操作mysql数据库这里我们需要用到三方库pymysql，可直接使用pip install pymysql安装，在python2中是MySQLdb模块，在python3中没有MySQLdb模块了，所以使用pymysql。

具体使用方法如下：

import pymysql

ip ='127.0.0.1'
port = 3306  #端口类型是数字int
username ='root'
password = '123456'
db ='test'
charset ='utf8'
# sql = input("请输入要执行的sql：")
sql ='select * from c;'

#创建连接，指定数据库的ip地址，账号、密码、端口号、要操作的数据库、字符集,autocommit
conn = pymysql.connect(host=ip,port=port,user=username,password=password,db = db,charset=charset,autocommit=True)
# conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', password='123456', db='test',charset='utf8',autocommit=True)
#创建游标，结果返回的是元组
# cusor = conn.cursor()
#创建游标，指定游标类型为字典类型，结果返回的是字典
cusor = conn.cursor(pymysql.cursors.DictCursor)
#执行sql,create insert update select
cusor.execute(sql)
#手动提交,conn 配置autocommit则不需要该行
conn.commit()
# 获取最新自增ID
new_id = cusor.lastrowid
print(new_id)
#关闭游标
cusor.close()
#关闭连接
conn.close()
#获取查询结果的第一条数据
print(cusor.fetchone())
#获取前n行数据
print(cusor.fetchmany())
#获取所有数据
print(cusor.fetchall())

二、python操作redis

redis是一个非关系型数据库，数据都存在内存中，有很快的读写速度，python操作redis要使用redis模块，可直接使用pip install redis安装。

import redis  #导入redis模块

#指定连接redis的ip、端口、数据库
r= redis.Redis(host='127.0.0.1',password='123456',port=6379,db =0,decode_responses=True)
r1= redis.Redis(host='127.0.0.1',password='123456',port=6379,db =1,decode_responses=True)
#======================字符串类型========================
r.set('key',"{'key1':'value'}")  #设置值
r.mset({'addr':'beijing','phone':15421114455}) #批量设置值
print(r.get('key')) #获取值
print(r.mget('key1','key2')) #批量获取值
r.expire('key',30) #设置有效时间
r.delete('name') # 删除值
r.delete('key1','key2') #批量删除
r.setnx('name2', 'value')  # 设置的name的值，如果name不存在的时候才会设置
r.setex('name3', 'value', 3)  # 设置的name的值，和超时时间，过了时间key就会自动失效
#=======================哈希类型============================
#set 哈希类型的值
r1.hset('hname','key','value')
r1.hset('students','myj',"{'money':50000,'addr':'北京','phone':13562141562}")
#批量设置哈希类型的key和value
r1.hmset('hname',{'key1':'value','key2':'value'})
r1.hmset('students',{'ym':"{'money':50004,'addr‘:'广州','phone':17144445555}",
                     'tm':"{'money':50005,'addr':'杭州',phone:13211111111}"})
print(r1.hget("students","libai")) #获取值
print(r1.hget('hname','key'))
print(r1.hgetall('hname')) #获取hname所有的值
r1.hdel('hname') # 删除哈希类型hname所有的值
print(r.keys()) #获取redis连接库的所有key