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