JsonPath | 说明 |
---|---|
$ | 文档根元素 |
@ | 当前元素 |
.或[] | 匹配下级元素 |
.. | 递归匹配所有子元素 |
* | 通配符,匹配下级元素 |
[] | 下标运算符,根据索引获取元素,JsonPath索引从0开始 |
[,] | 连接操作符,将多个结果拼接成数组返回,可以使用索引或别名 |
[start:end:step] | 数据切片操作 |
?() | 过滤表达式 |
这么多语法规范,大家会觉得一脸懵逼吧,细心的小伙伴会发现语法中 .. 表示递归匹配所有子元素,简单粗暴,用..获取所有匹配的子元素,通过索引拿到想要的值,以下以实战进行演练。
def jsonpath(obj, expr, result_type='VALUE', debug=0, use_eval=True): """traverse JSON object using jsonpath expr, returning values or paths"""
obj
: 需要解析的对象,比如接口的返回值。
expr
: 匹配的表达式。
以下编写代码进行演示。
from jsonpath import jsonpath # json格式结果 res = { "code": 0, "msg": "OK", "data": { "token_info": { "token_type": "Bearer", "expires_in": "2020-05-28 13:07:29", "token": "eyJhbGciOiJIUzUxAiJ9.eyJtZW1iZZJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNjQyNDQ5fQ.s6A7pzLILf9tqpEDAU9wIPGGLGkgKEQ6EBHq26l-eUM5seCb48DWalkE7u16iRZv3uzD5hIDFbw41Jmi9V0T_Q" } } } # 获取res中的token token = jsonpath(res,'$..token')[0] print(token)
响应结果为:
eyJhbGciOiJIUzUxBiJ9.eyJtZW1iZZJfaWQiOpEwMDA1MTMxNSwiZXhwIjoxNTkwNjQyNDQ5fQ.
s6A7pzLILf9tqpEDAU9wIPGGLGkgKEQ6EBHq26leUM5seCb48SWalkE8u16iRZv3uzD5hIDFbw41Jmi9V0T_Q
class DevConfig(): # 项目的域名或IP host='http://127.0.0.1:8888' config = DevConfig()
config.yaml
user: mobile_phone: '155********' pwd: '12345678'
helper.py
from common.requests_handler import RequestsHandler from config.setting import config from common.yaml_handler import yaml_data def login(): """登录,接口返回token""" req = RequestsHandler() res = req.visit('post', config.host+'/login', json=yaml_data['user']) return res print(login())
运行结果如下:
{ 'code': 0, 'msg': 'OK', 'data': { 'token_info': { 'token_type': 'Bearer', 'expires_in': '2020-05-26 13:31:41', 'token': 'eyJhbGciOiJIUpUxMiJ9.eyJtZW1iZXJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNDcxMTAxfQ.4QDNdQ-Wyecvpv7yRnMlCx1ZXpAlb8vpohxHht5OfNcjGn_yTaISyhmn7OMzjMGlz68sJ_uFbCh2NhIaO2p_Lg' } } }
获取登录接口返回值并提取token和token_type进行拼接 。这里要使用jsonpath模块提取返回的token和token_type。
helper.py
def save_token(): """保存token信息""" res = login() token = jsonpath(res,'$..token')[0] token_type = jsonpath(res,'$..token_type')[0] token = " ".join([token_type, token]) return token print(save_token())
运行结果如下:
Bearer eyJhbGciOJIPzUxMiJ9.eyJtZW1iZXJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwN
将token作为类属性,整体代码如下:
helper.py
from common.requests_handler import RequestsHandler from config.setting import config from common.yaml_handler import yaml_data from jsonpath import jsonpath def login(): """登录,返回token信息""" req = RequestsHandler() res = req.visit('post', config.host+'/login', json=yaml_data['user']) return res def save_token(): """保存token信息""" res = login() token = jsonpath(res,'$..token')[0] token_type = jsonpath(res,'$..token_type')[0] token = " ".join([token_type, token]) Context.token = token return token class Context: """将token作为类属性""" token = '' if __name__ == '__main__': print(save_token())
运行结果如下:
Bearer eyJhbGciOiJIUzUxMiJ9.eyJtZW1iZXJfaWQiOjEwMDA1MTMxNSwiZXhwIjoxNTkwNDcz
import unittest import requests from middleware.helper import Context, save_token class TestRecharge(unittest.TestCase): def setUp(self): pass def tearDown(self): pass def test_recharge(self): save_token() token = Context.token url = 'http://127.0.0.1:88888/recharge' payload = { "member_id": 100051315, "amount": 80 } res = requests.post(url,json=payload) print(res.json()) if __name__ == '__main__': unittest.main()
运行结果为:
{ 'code': 0, 'msg': 'OK', 'data': { 'leave_amount': 240.0, 'mobile_phone': '155********', } }
以上就是如何提取token、将token作为类属性全局调用及充值接口如何携带token进行请求的内容了,希望大家以后多多支持脚本之家!