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

from utils.redis.connect import r
from utils.redis.rkeys import WXA_CODE_SCENE_PAGE_QINIUURL_MAPPING, WXA_CODE_SID_SCENE_MAPPING


def set_wxa_code_info(scene, page, qiniu_url, sid):
    p = r.pipeline()
    p.hset(WXA_CODE_SCENE_PAGE_QINIUURL_MAPPING, '{0}:{1}'.format(scene, page), qiniu_url)
    p.hset(WXA_CODE_SID_SCENE_MAPPING, sid, scene)
    p.execute()


def get_wxa_code_qiniu_url(scene, page):
    return r.hget(WXA_CODE_SCENE_PAGE_QINIUURL_MAPPING, '{0}:{1}'.format(scene, page))


def get_wxa_code_scene_(sid):
    return r.hget(WXA_CODE_SID_SCENE_MAPPING, sid)