pai2/utils/redis/rprofile.py

# -*- coding: utf-8 -*-

import json

from utils.redis.connect import r
from utils.redis.rkeys import PROFILE_INFO


# 用户相关


def set_profile_info(user):
    """ 设置用户信息 """
    r.set(PROFILE_INFO % user.user_id, json.dumps(user.data))
    return user.data


def set_profile_by_uid(user_id):
    """ 获取用户信息 """
    from account.models import UserInfo
    try:
        user = UserInfo.objects.get(user_id=user_id)
    except UserInfo.DoesNotExist:
        return {}
    return set_profile_info(user)


def get_profile_by_id(user_id):
    """ 获取用户信息 """
    return json.loads(r.get(PROFILE_INFO % user_id) or '{}') or set_profile_by_uid(user_id)