Python接口自动化之request请求封装源码分析

前言：

我们在做自动化测试的时候，大家都是希望自己写的代码越简洁越好，代码重复量越少越好。那么，我们可以考虑将request的请求类型（如：Get、Post、Delect请求）都封装起来。这样，我们在编写用例的时候就可以直接进行请求了。

1. 源码分析

我们先来看一下Get、Post、Delect等请求的源码，看一下它们都有什么特点。

（1）Get请求源码

1 2 3 4 5 6 7 8 9

def get(self, url, **kwargs):     r"""Sends a GET request. Returns :class:`Response` object.     :param url: URL for the new :class:`Request` object.     :param \*\*kwargs: Optional arguments that ``request`` takes.     :rtype: requests.Response      """           kwargs.setdefault('allow_redirects', True)     return self.request('GET', url, **kwargs)

（2）Post请求源码

1 2 3 4 5 6 7 8 9 10

def post(self, url, data=None, json=None, **kwargs):     r"""Sends a POST request. Returns :class:`Response` object.     :param url: URL for the new :class:`Request` object.     :param data: (optional) Dictionary, list of tuples, bytes, or file-like     object to send in the body of the :class:`Request`.     :param json: (optional) json to send in the body of the :class:`Request`.     :param \*\*kwargs: Optional arguments that ``request`` takes.     :rtype: requests.Response     """     return self.request('POST', url, data=data, json=json, **kwargs)

（3）Delect请求源码

1 2 3 4 5 6 7

def delete(self, url, **kwargs):     r"""Sends a DELETE request. Returns :class:`Response` object.     :param url: URL for the new :class:`Request` object.     :param \*\*kwargs: Optional arguments that ``request`` takes.     :rtype: requests.Response     """     return self.request('DELETE', url, **kwargs)

（4）分析结果

我们发现，不管是Get请求、还是Post请求或者是Delect请求，它们到最后返回的都是request函数。那么，我们再去看一看request函数的源码。

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66

def request(self, method, url,         params=None, data=None, headers=None, cookies=None, files=None,         auth=None, timeout=None, allow_redirects=True, proxies=None,         hooks=None, stream=None, verify=None, cert=None, json=None):     """Constructs a :class:`Request <Request>`, prepares it and sends it.     Returns :class:`Response <Response>` object.     :param method: method for the new :class:`Request` object.     :param url: URL for the new :class:`Request` object.     :param params: (optional) Dictionary or bytes to be sent in the query         string for the :class:`Request`.     :param data: (optional) Dictionary, list of tuples, bytes, or file-like         object to send in the body of the :class:`Request`.     :param json: (optional) json to send in the body of the         :class:`Request`.     :param headers: (optional) Dictionary of HTTP Headers to send with the         :class:`Request`.     :param cookies: (optional) Dict or CookieJar object to send with the         :class:`Request`.     :param files: (optional) Dictionary of ``'filename': file-like-objects``         for multipart encoding upload.     :param auth: (optional) Auth tuple or callable to enable         Basic/Digest/Custom HTTP Auth.     :param timeout: (optional) How long to wait for the server to send         data before giving up, as a float, or a :ref:`(connect timeout,         read timeout) <timeouts>` tuple.     :type timeout: float or tuple     :param allow_redirects: (optional) Set to True by default.     :type allow_redirects: bool     :param proxies: (optional) Dictionary mapping protocol or protocol and         hostname to the URL of the proxy.     :param stream: (optional) whether to immediately download the response         content. Defaults to ``False``.     :param verify: (optional) Either a boolean, in which case it controls whether we verify         the server's TLS certificate, or a string, in which case it must be a path         to a CA bundle to use. Defaults to ``True``.     :param cert: (optional) if String, path to ssl client cert file (.pem).         If Tuple, ('cert', 'key') pair.     :rtype: requests.Response     """     # Create the Request.     req = Request(         method=method.upper(),         url=url,         headers=headers,         files=files,         data=data or {},         json=json,         params=params or {},         auth=auth,         cookies=cookies,         hooks=hooks,     )     prep = self.prepare_request(req)     proxies = proxies or {}     settings = self.merge_environment_settings(         prep.url, proxies, stream, verify, cert     )     # Send the request.     send_kwargs = {         'timeout': timeout,         'allow_redirects': allow_redirects,     }     send_kwargs.update(settings)     resp = self.send(prep, **send_kwargs)     return resp

从request源码可以看出，它先创建一个Request，然后将传过来的所有参数放在里面，再接着调用self.send()，并将Request传过去。这里我们将不在分析后面的send等方法的源码了，有兴趣的同学可以自行了解。

分析完源码之后发现，我们可以不需要单独在一个类中去定义Get、Post等其他方法，然后在单独调用request。其实，我们直接调用request即可。

2. requests请求封装

代码示例：

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48

import requests class RequestMain:     def __init__(self):         """         session管理器         requests.session(): 维持会话,跨请求的时候保存参数         """         # 实例化session         self.session = requests.session()     def request_main(self, method, url, params=None, data=None, json=None, headers=None, **kwargs):         """         :param method: 请求方式         :param url: 请求地址         :param params: 字典或bytes，作为参数增加到url中                 :param data: data类型传参，字典、字节序列或文件对象，作为Request的内容         :param json: json传参，作为Request的内容         :param headers: 请求头，字典         :param kwargs: 若还有其他的参数，使用可变参数字典形式进行传递         :return:         """         # 对异常进行捕获         try:             """                           封装request请求，将请求方法、请求地址，请求参数、请求头等信息入参。             注 ：verify: True/False，默认为True，认证SSL证书开关；cert: 本地SSL证书。如果不需要ssl认证，可将这两个入参去掉             """             re_data = self.session.request(method, url, params=params, data=data, json=json, headers=headers, cert=(client_crt, client_key), verify=False, **kwargs)         # 异常处理 报错显示具体信息         except Exception as e:             # 打印异常             print("请求失败：{0}".format(e))         # 返回响应结果         return re_data if __name__ == '__main__':     # 请求地址     url = '请求地址'     # 请求参数     payload = {"请求参数"}     # 请求头     header = {"headers"}     # 实例化 RequestMain()     re = RequestMain()     # 调用request_main，并将参数传过去     request_data = re.request_main("请求方式", url, json=payload, headers=header)     # 打印响应结果     print(request_data.text)

注 ：如果你调的接口不需要SSL认证，可将cert与verify两个参数去掉。

3. 总结

本文简单的介绍了Python接口自动化之request请求封装

​现在我也找了很多测试的朋友，做了一个分享技术的交流群，共享了很多我们收集的技术文档和视频教程。
如果你不想再体验自学时找不到资源，没人解答问题，坚持几天便放弃的感受
qq群号：485187702【暗号：csdn11】
可以加入我们一起交流。而且还有很多在自动化，性能，安全，测试开发等等方面有一定建树的技术大牛
分享他们的经验，还会分享很多直播讲座和技术沙龙
可以免费学习！划重点！开源的！！！
视频+文档+PDF+面试题可以关注公众号：【软件测试小dao】

最后感谢每一个认真阅读我文章的人，看着粉丝一路的上涨和关注，礼尚往来总是要有的，虽然不是什么很值钱的东西，如果你用得到的话可以直接拿走！ 希望能帮助到你！【100%无套路免费领取】