一、首先新建一个公共的文件common

为了获取token,方便调用

 二、在common文件夹下,新建文件get_header.py,(我们获取token的目的是为了给下一个接口拼接调用,我们直接拼接好,再传出去)

""" 获取登录成功的令牌,拼接到请求头,返回"""

# 定义函数
import requests

# 封装请求头
def get_header():
    url = "" # 自己抓取的url
    data = {} # 自己抓取的body
    res = requests.post(url=url, json=data)
    # 从响应体中获取 token   
    token = res.json().get("head").get("token")

    header = {"Content-Type": "application/json",
              "token": token}

    # 上面的 header 拼接用不了,可以试一下,下面的 header 拼接
    # header = {"Content-Type": "application/json",
    #           "Authorization": "Bearer " + token}  # "Bearer " 后面一定要有空格
    return header

if __name__ == '__main__':
    print(get_header())

三、 新建一个api 文件

新建一个文件 query_api.py

import requests

class QueryApi(object):
    @classmethod
    def query_hco(cls, header):
        url = "" # 下一个接口的url
        resp = requests.get(url=url, headers=header)
        return resp

 四、新建一个文件test_query.py 调用,QueryApi和get_header

import unittest

from api.query_api import QueryApi
from common.get_header import get_header

class TestQuery(unittest.TestCase):
    # 类属性
    header = None

    # 每次项目开始调用一次,获取最新的token
    @classmethod
    def setUpClass(cls) -> None:
        cls.header = get_header()


    def test_query(self):
        resp = QueryApi.query_hco(self.header)
        print(resp.json())

if __name__ == '__main__':
    print(TestQuery())

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐