| 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進行請求的內容了,希望大家以后多多支持腳本之家！
