kodo/page/screen_views.py

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

import shortuuid
from django.conf import settings
from django_logit import logit
from django_query import get_query_value
from django_response import response
from json_render import json_render

from guideline.models import ScreenAdminInfo
from utils.error.errno_utils import PermissionStatusCode, ScreenStatusCode
from utils.redis.connect import r
from utils.redis.rkeys import SCREEN_ADMIN_LOGIN


@logit
def screen_admin_oauthqr(request):
    brand_id = request.GET.get('brand_id') or settings.KODO_DEFAULT_BRAND_ID

    return json_render(request, 'page/screen_admin_oauth_qrcode.html', unjsondumpsdict={
        'qr': settings.KODO_SCREEN_AUTH_URL.format(brand_id)
    })


@logit
def screen_admin_oauth(request):
    brand_id = request.GET.get('brand_id') or settings.KODO_DEFAULT_BRAND_ID
    unionid = request.GET.get('unionid', '')
    openid = request.GET.get('openid', '')

    ScreenAdminInfo.objects.update_or_create(brand_id=brand_id, unionid=unionid, openid=openid, defaults={
        'nickname': request.GET.get('nickname', ''),
        'avatar': request.GET.get('avatar', '') or request.GET.get('headimgurl', ''),
        'user_status': ScreenAdminInfo.ACTIVATED,
    })

    return json_render(request, 'page/screen_admin_oauth_success.html', unjsondumpsdict={
    })


@logit
def screen_admin_loginqr(request):
    brand_id = get_query_value(request, 'brand_id', settings.KODO_DEFAULT_BRAND_ID)

    token = shortuuid.uuid()

    return response(200, data={
        'qr': settings.KODO_SCREEN_LOGIN_URL.format(brand_id, token),
        'token': token,
    })


@logit
def screen_admin_login(request):
    brand_id = request.GET.get('brand_id') or settings.KODO_DEFAULT_BRAND_ID
    unionid = request.GET.get('unionid', '')
    openid = request.GET.get('openid', '')
    token = request.GET.get('token', '')

    try:
        admin = ScreenAdminInfo.objects.get(unionid=unionid, user_status=ScreenAdminInfo.ACTIVATED)
    except ScreenAdminInfo.DoesNotExist:
        return json_render(request, 'page/screen_admin_login_fail.html', unjsondumpsdict={
        })

    r.setex(SCREEN_ADMIN_LOGIN % (brand_id, token), r.REDIS_EXPIRED_HALF_HOUR, unionid)

    return json_render(request, 'page/screen_admin_login_success.html', unjsondumpsdict={
    })


@logit
def screen_admin_loginrst(request):
    brand_id = get_query_value(request, 'brand_id', settings.KODO_DEFAULT_BRAND_ID)
    token = get_query_value(request, 'token', '')

    admin_login_key = SCREEN_ADMIN_LOGIN % (brand_id, token)

    if not r.exists(admin_login_key):
        return response(ScreenStatusCode.QRCODE_NOT_SCAN)

    unionid, _ = r.getdelete(admin_login_key)

    try:
        admin = ScreenAdminInfo.objects.get(unionid=unionid, user_status=ScreenAdminInfo.ACTIVATED)
    except ScreenAdminInfo.DoesNotExist:
        return response(PermissionStatusCode.PERMISSION_DENIED)

    return response(200, data={
        'info': admin.data,
    })