requests请求封装思路

一:发送HTTP请求模块封装

1. 使用requests库访问接口

import requests

url = "https://bimdc.bzlrobot.com/bsp/test/user/ugs/ibs/api/ibs-material/material/jobRequire/detail?t=&isEdit=&requireId=&projectId="

payload={}
headers = {
  'BSP_TOKEN': '{{bspToken}}'
}

response = requests.request("GET", url, headers=headers, data=payload)

print(response.json()
     )
{
    "code": 400,
    "success": false,
    "data": null,
    "msg": "未找到请购单"
}

2. 功能分析

可以根据用例数据动态的

  1. 向不同url发送http请求
  2. 发送各种方法的http请求
  3. 可以携带各种参数

3. 封装成函数

封装思路:

  1. 输入参数为url,method,和需要携带的各种参数以及请求头等,与requests的原生形参一致
  2. 可以使用关键字动态参数解决参数过多的问题
  3. 根据传入的method,发送对应的请求

common文件夹下创建req.py模块,在其中编写如下函数:

# -*- coding: utf-8 -*-
# @Time : 2019/11/18 10:17
# @Author : kira
# @Email : 262667641@qq.com
# @File : req.py
# @Project : risk_api_project

import requests


def req(url, method='GET', **kwargs):
    """
    发送http请求
    @param url: 接口url
    @param method:http 请求方法
    @param kwargs:接受request原生的关键字参数
    @return 响应对象
    """
    # 将方法统一转为小写
    method = method.lower()
    # 根据传入的方法名发送对应的请求
    if method == 'get':
        res = requests.get(url, **kwargs)
    if method == 'post':
        res = requests.post(url, **kwargs)
    elif method == 'put':
        res = requests.put(url, **kwargs)
    elif method == 'delete':
        res = requests.delete(url, **kwargs)
    elif method == 'patch':
        res = requests.patch(url, **kwargs)
    else:
        raise ValueError('请输入正确的请求方法')
    return res
    # 代码逻辑请求,但是冗余,菜鸟级的代码。面向对象里的反射,改写上面的代码


if __name__ == '__main__':
    url = 'https://bimdc.com/bsp/test/api/material/jobRequire/pages?t=1681118956000'
    method = 'POST'
    request_data = {
        "json":{"userId": "2019", "projectId": "104999"},
        "headers":{"BSP_TOKEN": "fc8fc6626920j8a8c729c6e00323fc4f"}
    }
    data = send_http_request(url, method, **request_data)  # 在传递实参时,可以通过 ** 对字典对象进行解包。解包成多个关键字参数
    print(data.json())
代码逻辑请求,但是冗余,菜鸟级的代码。面向对象里的反射,改写上面的代码
getattr(requests, 'get')() # 返回一个对象给定名称的属性
<function requests.api.get(url, params=None, **kwargs)>

同样:

requests.get()
<function requests.api.get(url, params=None, **kwargs)>

所以可以将上述复杂的逻辑使用反射简化

# -*- coding: utf-8 -*-
# @Time : 2019/11/18 10:17
# @Author : kira
# @Email : 262667641@qq.com
# @File : req.py
# @Project : risk_api_project
import requests

def req(url, method='GET', **kwargs):
    """
    发送http请求
    @param url: 接口url
    @param method:http 请求方法
    @param kwargs:接受request原生的关键字参数
    @return 响应对象
    """
    # 将方法统一转为小写
    method = method.lower()
    # # 根据传入的方法名发送对应的请求
    # if method == 'get':
    #     res = requests.get(url, **kwargs)
    # if method == 'post':
    #     res = requests.post(url, **kwargs)
    # elif method == 'put':
    #     res = requests.put(url, **kwargs)
    # elif method == 'delete':
    #     res = requests.delete(url, **kwargs)
    # elif method == 'patch':
    #     res = requests.patch(url, **kwargs)
    # else:
    #     raise ValueError('请输入正确的请求方法')
    # return res.json()
    # # 代码逻辑请求,但是冗余,菜鸟级的代码。面向对象里的反射,改写上面的代码
    return getattr(requests, method)(url, **kwargs)


if __name__ == '__main__':
   url = 'https://bimdc.com/bsp/test/api/material/jobRequire/pages?t=1681118956000'
    method = 'POST'
    request_data = {
        "json":{"userId": "2019", "projectId": "104999"},
        "headers":{"BSP_TOKEN": "fc8fc6626920j8a8c729c6e00323fc4f"}
    }
    data = send_http_request(url, method, **request_data)  # 在传递实参时,可以通过 ** 对字典对象进行解包。解包成多个关键字参数
    print(data.json())

 注意:使用反射修改上述代码存在一些问题,比如:反射会对性能有一定的影响,但是对于自动化来说,不差这几秒时间

posted @ 2023-04-12 15:04  测试玩家勇哥  阅读(76)  评论(0编辑  收藏  举报