python接口自动化-9. token 的处理

  • Post author:
  • Post category:python



一、提取token


1、json解析提取

# -*- coding:utf-8 -*-
import requests
host = ""
user = ""
pwd = ""

url = host + "/pub/api/v1/web/web_login"
body = {
    "phone": user,
    "pwd": pwd
}
r = requests.post(url=url, data=body).json()  # .json() 就是 json 解析,把json格式转为字典
token = r["data"]["token"]  # 字典取值
print(token)


2、正则提取json

# -*- coding:utf-8 -*-
import requests, re
host = ""
user = ""
pwd = ""

url = host + "/pub/api/v1/web/web_login"
body = {
    "phone": user,
    "pwd": pwd
}
r = requests.post(url=url, data=body).text
print(r)
token = re.findall('"token":"(.+?)"', r)
token = token[0]  # 正则取出来的值是 列表 类型,所以要进行列表取值
print(token)


二、结合框架使用 token


python里面有个全局变量global,但这个只是针对于在同一个.py里才有效,跨脚本就不起作用了。


整体的思路:


1、登录后返回 token,把 token 写入 yaml 文件中,yaml 文件放在公共模块 commen 中


2、需要 token 的时候(一般是调用写用例的时候),在初始化中读取 yaml 文件中最新的 token 值


3、每个用例的 package 执行的时候去调用获取 token


4、最后执行所有用例,生成报告,发送邮件等


所以先把读写 yaml 的方法封装好


headle.py

# -*- coding:utf-8 -*-import os, yaml


defwrite_yaml(token):
    cur = os.path.dirname(os.path.realpath(__file__))  # 获取当前路径
    yaml_path = os.path.join(cur, "token.yaml")  # 获取yaml文件的路径print(yaml_path)
    t = {"token": token}  # 写入的内容withopen(yaml_path, 'w', encoding='utf-8') as f:
       yaml.dump(t, stream=f, allow_unicode=True)


defget_yaml(yaml_file):
    # yaml_file = "D:\\code\\api_test\\commen\\token.yaml"
    f = open(yaml_file, 'r', encoding='utf-8')
    cfg = f.read()
    d = yaml.load(cfg, Loader=yaml.FullLoader)
    """
    用load转字典
    yaml5.1版本后弃用了yaml.load(file)这个用法,因为觉得很不安全,5.1版本之后就修改了需要指定Loader,通过默认加载器(FullLoader)禁止执行任意函数
    Loader=yaml.FullLoader 加上这行代码,告警就没了
    """
    d = d["token"]
    return d


if __name__ == '__main__':
    r = write_yaml("token的值")


封装接口的时候,把 token 设置成变量


user.py

# -*- coding:utf-8 -*-import requests


defuser_info(host, token):
    url = host + '/pub/api/v1/web/user_info'
    headers = {"token": token}  # token 放在请求头
    r = requests.get(url=url, headers=headers)
    return r


编写用例的时候先获取 yaml 文件中 token 的值


test_user_info.py

# -*- coding:utf-8 -*-from interface.user import user_info
from commen.headle_token import *
import unittest, os
current_path = os.path.dirname(os.path.realpath(__file__))  # 获取当前路径# 获取 token 的路径
token_path = os.path.join(current_path, "commen")
token_path = os.path.join(token_path, "token.yaml")


classTestUserInfo(unittest.TestCase):
    @classmethoddefsetUpClass(cls):
        cls.token = get_yaml(token_path)
        print("获取token")
        print(cls.token)

    deftest_user_info_success(self):
        r = user_info("https://api.xdclass.net", self.token)
        print(r.text)


if __name__ == '__main__':
    unittest.main()


主程序


1、登录返回token


2、token 写入 yaml 文件


3、执行用例初始化的时候先获取 token


4、执行用例生成报告


run_case_report.py

# -*- coding:utf-8 -*-
import unittest, os
from commen import HTMLTestRunner
from interface.login import login
from commen.headle_token import *

current_path = os.path.dirname(os.path.realpath(__file__))
case_path = os.path.join(current_path, "case")

report_path = os.path.join(current_path, "report")
report = os.path.join(report_path, "report.html")


def all_case():
    discover = unittest.defaultTestLoader.discover(case_path,
                                                   pattern='test*.py')
    print(discover)
    return discover


def run_case_report():
    fb = open(report, "wb")
    runner = HTMLTestRunner.HTMLTestRunner(
        stream=fb,
        title="xx项目测试报告",
        description="xx项目的测试报告"
    )
    runner.run(all_case())
    fb.close()


if __name__ == '__main__':
    # 调用登录获取token
    token = login("host", "登录的账号", "密码")
    # 把token写入 yaml 文件
    write_yaml(token)
    # 执行用例的时候会读取 yaml 中的token,case文件下 test_user_info.py 的
    run_case_report()



版权声明:本文为weixin_45246215原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。