You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.8 KiB
39 lines
1.8 KiB
from random import random |
|
|
|
from django_redis import get_redis_connection |
|
import json |
|
import random |
|
import xmltodict |
|
|
|
|
|
# 封装方法处理从redis中获取接口数据 |
|
def get_data_from_redis(redis_key): |
|
# 初始化redis |
|
redis_conn = get_redis_connection() |
|
# 如果有用户数据则处理用户数据 |
|
if redis_conn.exists(redis_key + '_user_data'): |
|
redis_data = json.loads(redis_conn.get(redis_key + '_user_data')) |
|
user_data = redis_data['user_data'] |
|
print('user_data', json.dumps(user_data)) |
|
return True, redis_data['format'], user_data[0], redis_data['timestamp'] |
|
# 如果没用户数据时优先获取随机数据 |
|
if redis_conn.exists(redis_key + '_random'): |
|
redis_data = json.loads(redis_conn.get(redis_key + '_random')) |
|
user_data = redis_data['user_data'] |
|
print('user_data', json.dumps(user_data)) |
|
return True, redis_data['format'], user_data[0], redis_data['timestamp'] |
|
# 没有用户数据时，随机选择数据 |
|
if redis_conn.exists(redis_key): |
|
redis_data = json.loads(redis_conn.get(redis_key)) |
|
resp_data = [] |
|
resp = redis_data['handled_data'] |
|
print('handled_data', json.dumps(redis_data['handled_data'])) |
|
resp_result = resp['res']['status'] |
|
if str(resp_result) == '1': |
|
resp_data = resp['res']['data'] if isinstance(resp['res']['data'], list) else [resp['res']['data']] |
|
if str(resp_result) != '1': |
|
return False, redis_data['format'], '请检查接口返回值或手动输入参数：', 0 |
|
if len(resp_data) == 0: |
|
return False, redis_data['format'], '接口返回数据为空，请手动输入参数：', 0 |
|
return True, redis_data['format'], random.choice(resp_data), redis_data['timestamp'] |
|
return False, 'json', '请手动输入参数，或先请求接口：', 0
|
|
|