python接口自动化测试中请求工具类封装

前面博主大致的写了一个http与requests库方面的知识，下面我们就结合这两方面的知识来完成一下接口自动化请求工具类封装

这也是我们即将进行接口自动化测试的第一步。

废话不多说直接上代码吧。

import requests


class Requestuil:
    def __init__(self):
        pass

    def request(self,url,method,headers=None,param=None,content_type=None):
        '''请求工具类封装'''
        try:
            if method == "get":
                result = requests.get(url=url,params=param,headers =headers).json()
                return result
            elif method == "post":
                if content_type == "application/json":
                    result = requests.post(url=url,json=param,headers = headers).json()
                    return result
                else:
                    result = requests.post(url = url,data=param,headers=headers).json()
                    return result
            else:
                print('只做了post and get请求')
        except Exception as e:
            print("http请求报错：{0}".format(e))


if __name__ == '__main__':
    r = Requestuil()

在post请求时请求头中Content_type字段表明该请求格式为json格式时我们在使用post请求时必须表明，所以在post请求这里博主做了一个判断，这里有不理解的地方可以看博主前面写的关于requests库博客。

这些代码我相信不难理解，大家直接直接复制下面试一试啦。