def get_token(self):
'''获取登录接口api_token字典'''
url = self.API_URL + 'pub/login'
data = {'username': self.c.username, 'password': self.c.password}
try:
res = HTTPClient(url=url, method='post').send(data=data)
except Exception as e:
raise GETTOKENException('获取登录响应失败: %s' % str(e))
token = res.json()['data']
data_dict = {'api_token': token}
return data_dict
"""抽取器，从响应结果中抽取部分数据"""
import json
import jmespath
import json
class JMESPathExtractor(object):
"""
用JMESPath实现的抽取器，对于json格式数据实现简单方式的抽取。
"""
def extract(self, query=None, body=None):
try:
return jmespath.search(query, json.loads(body))
except Exception as e:
raise ValueError("Invalid query: " + query + " : " + str(e))
if __name__ == '__main__':
from utils.client import HTTPClient
res = HTTPClient(
url='http://wthrcdn.etouch.cn/weather_mini?citykey=101010100').send()
print(json.dumps(res.json(), indent=2, ensure_ascii=False))
j = JMESPathExtractor()
j_1 = j.extract(query='data.forecast[0].date', body=res.text)
j_2 = j.extract(query='data.ganmao', body=res.text)
print(j_1, '\n', j_2)
