-
数据库参数化的场景
部分接口,参数其他方式获取不到,可以去数据库去获取,如手机验证码 -
安装pymysql
-
配置文件
def sql_conf():
# 定义数据库的配置
host=‘127.0.0.1‘
user=‘root‘
password=‘123456‘
database=‘employees‘
port=3306
charset=‘utf8‘
return host,user,password,database,port,charset
- 定义get_mysql方法
import pymysql
from config.conf import sql_conf #导入参数配置定义的函数
def get_sql(sql):
# 建立一个连接对象
host,user,password,database,port,charset=sql_conf() #引用参数配置的值
db=pymysql.connect(host=host,user=user,database=database,port=port,charset=charset)
# 建立一个游标
cursor=db.cursor()
# 运行sql语句
cursor.execute(sql)
# 保存查询的结果
data=cursor.fetchall()
# 关闭游标
cursor.close()
# 关闭数据库连接
db.close()
# 返回数据
return data
- 调用get_mysql方法获取id
import requests
from common.get_mysql import get_sql
def test_get_mysql():
url=‘http//:127.0.0.1:8080‘
id=get_sql("SELECT dept_no FROM departments WHERE dept_name =‘Finance‘") # 从数据库获取数据
data={‘name‘:‘Finance‘,‘id‘:id}
r=requests.post(url=url,data=data)
print(r.status_code)
print(r.text)