python，requests试下简单的查询，新增，修改

import requests
import json

#传参的body和header都是字典，所以先建一个字典格式
body = {}
header = {}

#通用地址头部配置
base_url = '公用url'
header['Content-Type'] = 'application/json'



#获取token的方法
def get_token():
    url = base_url + '登录接口'
    body['grant_type'] = 'password'
    body['username'] = 'admin'
    body['password'] = '123456'
    body['client_secret'] = 'secret'


#requests方法就是http请求
    with requests.request('post',url=url,params=body) as rsp:
        rst = rsp.json()
        token = rst.get('data').get('token')
    return token

#将获取到的token放在请求头的配置中，作为全局变量去使用
header['Authorization'] = f'Bearer {get_token()}'


#查询
def get_select():

    url = base_url +'查询'
    body = {"pageIndex":1,"pageNum":10,"supplyName":None,"materialName":None,"factoryName":None,"startTime":None,"endTime":None}
    #要将body转译下再传入
    body = json.dumps(body)

    with requests.request('post',url=url,data=body,headers=header) as rsp:
        #将respon转译下
        rst = rsp.json()
        rst = rst.get('msg')


    if rst == '查询成功':
        print('测试用例执行成功')
    else:
        print('测试用例执行失败')


#新增
def get_add():

    header['Content-Type'] = 'application/json'

    url = base_url +'新增接口'
    body = {"deptIds":[1,74],"enabled":False,"headImgUrl":None,"mobile":"13000000000","nickname":"测试自动化4","username":"testapi4","roleIds":[39]}
    body = json.dumps(body)

    with requests.request('post',url=url,data=body,headers=header) as rsp:
        rst = rsp.json()
        print(rst)
        rst = rst.get('msg')

#可以做个简单的响应断言判断
    if rst == '新增成功':
        print('成功')
    else:
        print('失败')

def get_modify():

    url = base_url + '修改接口'
    body = {"deptIds":[1,74],"enabled":True,"headImgUrl":"","id":4970,"mobile":"13000000000","nickname":"测试自动化修改","username":"testapi","roleIds":[39]}
    body = json.dumps(body)

    with requests.request('post',url=url,data =body,headers=header) as rsp:
        rst = rsp.json()
        rst = rst.get('msg')


    if rst == '更新成功':
        print('成功')
    else:
        print('失败')