Python接口自动化 -RESTful请求方法封装

对 Requests 库下一些常见的请求方法进行封装，调用起来更加方便。

先来看一个例子：

比如要测试登录接口

1. 正常流测试（正确的账号、密码）
2. 异常流测试（正确的账号，错误的密码等）

import requests

# 正常流
url = 'http://192.168.28.128:8080/recruit.students/login/in?account=admin&pwd=test123'
result = requests.get(url)
print(result.text)

# 异常流，密码错误
url = 'http://192.168.28.128:8080/recruit.students/login/in?account=admin&pwd=test'
result = requests.get(url)
print(result.text)

看上面这段代码，是不是显得比较冗余，我们每个请求里都要写一遍完整的 url。
还有就是当我们需要做一些连续调用的请求，比如“先登录再把商品加入购物车“，这样的时候，需要使用 requests 的 session 功能。

就要需要登录和登录后的请求都由同一个reqeusts session对象来发出。

# -*- coding:utf-8 -*-
import requests
import json as complexjson


class RestClient():
    def __init__(self, api_root_url):
        """
        :type api_root_url: 就是url的前缀，相当于host
        """
        self.api_root_url = api_root_url
        # 用session保持会话，相当于微型浏览器
        self.session = requests.session()

    def get(self, uri, **kwargs):
        """
        :type uri: 路径，每个接口的路径不一样，所以把路径抽离出来
        :type **kwargs: 关键字参数，它将不定长参数转换为 dict 传入函数
        """
        # 返回的时候调用这个类中的 request 方法
        return self.request(uri, "GET", **kwargs)

    def post(self, uri, data=None, json=None, **kwargs):
        return self.request(uri, "POST", data, json, **kwargs)

    def put(self, uri, data=None, **kwargs):
        return self.request(uri, "PUT", data, **kwargs)

    def delete(self, uri, **kwargs):
        return self.request(uri, "DELETE", **kwargs)

    def patch(self, uri, data=None, **kwargs):
        return self.request(uri, "PATCH", data, **kwargs)

    def request(self, uri, method, data=None, json=None, **kwargs):
        url = self.api_root_url + uri
        if method == "GET":
            return self.session.get(url, **kwargs)
        if method == "POST":
            return self.session.post(url, data, json, **kwargs)
        if method == "PUT":
            if json:
                # PUT 和 PATCH 中没有提供直接使用json参数的方法，因此需要用data来传入
                data = complexjson.dumps(json)
            return self.session.put(url, data, **kwargs)
        if method == "DELETE":
            return self.session.delete(url, **kwargs)
        if method == "PATCH":
            if json:
                data = complexjson.dumps(json)
            return self.session.patch(url, data, **kwargs)

这里，这个类里我加了一些方法，包括：

1. init 方法：初始化这个类，初始化的时候需要输入 api_root_url，也就是URL的前缀host。另外还在初始化时创建了 self.session 用于保存 requests 的 session。

2. get，post 等各种 http 方法，用于让用户使用。但这里并没有真正实现这些方法，因为这些方法都是在 requests 里有实现过，我们只要把参数传给 requests 就行了。这个传递我们写在 request 方法里，所以这里的 http 方法都是调用 requests 方法。

3. request 方法：真正调用 self.session 的各种方法，这里同样是把参数传下去，只是在传之前，给所有用户输入的 url 加了一个前缀。前缀的值是用户在 init 方法里输入的。
   
   

使用封装好的方法进行登录操作：

r = RestClient("http://192.168.28.128:8080")

login = r.get(uri="/recruit.students/login/in",
              params={"account": "admin",
                       "pwd": "660B8D2D5359FF6F94F8D3345698F88C"
                       }
              )

只是要输入接口的 uri 和参数即可，不用再输入完整的 url。后续还可以对接口进行封装，接口封装完之后，测试的时候只需要输入接口入参就行。

-----分界线-------

王阳名说：知而不行，是为不知

读书、学习、看文章，总说自己知道，但不去做，其实就是不知道。说自己知道，可能只是为了好看。

学习第一个方法就是：重复

学一个知识，学 1 遍，没学会，继续学第 2 遍，第 3 遍，第 4 遍……，直到学会。

练一个技能，练 1 次，不会用，继续练 2 次，第 3 次，第 4 次……，直到会用。

书读百遍，其义自见。