接口测试---python对数据库的操作

背景：在实际做项目中，哪些场景需要用到数据库呢？

场景一：在注册时，同一个手机号不能重复注册，否则脚本会报错！解决方案：连接数据库，把注册字段的最大手机号查询出来，然后+1再次去做请求；

场景二：充值前和充值后的数据验证；

场景三：做权限测试时，需要登录数据库把用户身份进行修改；

等等。。。

Python里有个可以操作mysql的模块，叫pymysql，安装方式：

pip install pymysql

1、 新建表

import pymysql

#1.连接数据库（connect-连接）
host="****"                #数据库的IP地址
user="test"               #用户名
password="test"           #密码
mysql=pymysql.connect( host=host, user=user, password=password,port=3306)

#2.新建查询页面(cursor-光标)
cursor=mysql.cursor()

#3.如果EMPLOYEE表存在，就删除它
sql="DROP TABLE IF EXISTS EMPLOYEE"

#4.执行删除表sql语句
cursor.execute(sql)

#5.新建employee表的sql
sql="""CREATE TABLE EMPLOYEE (
         FIRST_NAME  CHAR(20) NOT NULL,
         LAST_NAME  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         INCOME FLOAT )"""
         
#6.执行新建表的sql语句
cursor.execute(sql)
mysql.commit()   #记得提交

#7.关闭光标（查询）
cursor.close()

#8.关闭数据库
mysql.close()

2、 增

import pymysql

#1.连接数据库（connect-连接）
host="****"                #数据库的IP地址
user="test"               #用户名
password="test"           #密码
mysql=pymysql.connect( host=host, user=user, password=password,port=3306)

#2.新建查询页面(cursor-光标)
cursor=mysql.cursor()

#3、插入语句SQL
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""

#4、执行sql语句
cursor.execute(sql)
mysql.commit()   #记得提交

#5.关闭光标（查询）
cursor.close()

#6.关闭数据库
mysql.close()

3、 删

import pymysql

#1.连接数据库（connect-连接）
host="****"                #数据库的IP地址
user="test"               #用户名
password="test"           #密码
mysql=pymysql.connect( host=host, user=user, password=password,port=3306)

#2.新建查询页面(cursor-光标)
cursor=mysql.cursor()

#3、删除语句SQL
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)

#4、执行sql语句
cursor.execute(sql)
mysql.commit()   #记得提交

#5.关闭光标（查询）
cursor.close()

#6.关闭数据库
mysql.close()

4、 改

import pymysql

#1.连接数据库（connect-连接）
host="****"                #数据库的IP地址
user="test"               #用户名
password="test"           #密码
mysql=pymysql.connect( host=host, user=user, password=password,port=3306)

#2.新建查询页面(cursor-光标)
cursor=mysql.cursor()

#3、更改语句SQL
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = '%c'" % ('M')

#4、执行sql语句
cursor.execute(sql)
mysql.commit()   #记得提交

#5.关闭光标（查询）
cursor.close()

#6.关闭数据库
mysql.close()

5、 查

import pymysql

#1.连接数据库（connect-连接）
host="****" #数据库的IP地址
user="test"               #用户名
password="test"           #密码
mysql=pymysql.connect( host=host, user=user, password=password,port=3306)

#2.新建查询页面(cursor-光标)
cursor=mysql.cursor()

#3.写入sql语句
sql="select MAX(MobilePhone)from future.member"

#4.执行sql语句
cursor.execute(sql)

#5.查看结果
result=cursor.fetchone()    #元组类型的数据   fetchall()是查全部，fetchone只查一条
print(result[0])

#6.关闭光标（查询）
cursor.close()

#7.关闭数据库
mysql.close()

6、封装一个对mysql进行操作的类

import pymysql

class MysqlMethod:
    '''对数据库进行操作的类'''
    def __init__(self):
  		host="****" #数据库的IP地址
    	user="test"               #用户名
    	password="test"           #密码
    	self.mysql=pymysql.connect( host=host, user=user, password=password,port=3306)
    	self.cursor = self.mysql.cursor()

	def select(self,sql):
		 """查询函数"""
		self.cursor.execute(sql)
    	result = self.cursor.fetchone()
	   	return result
	   	
	def update(self, sql):
		"""增，删，改的函数"""
        self.cursor.execute(sql)
        self.mysql.commit()
        
	def close(self):
		self.cursor.close()
		self.mysql.close()        

未完待续。。。