kodo/api/encrypt_views.py

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

from __future__ import division

import random
import re

from django.conf import settings
from django.db import transaction
from django_logit import logit
from django_response import response
from pywe_marketcode import tickettocode
from pywe_storage import RedisStorage

from account.models import UserInfo
from logs.models import MchInfoDecryptLogInfo, MchInfoEncryptLogInfo
from marketcode.models import MarketCodeInfo
from mch.models import ActivityInfo, BrandInfo, ModelInfo, ConsumeShotUnbindingInfo
from utils.algorithm.b64 import b64_decrypt, b64_encrypt
from utils.algorithm.caesar import caesar_decrypt, caesar_encrypt
from utils.algorithm.rsalg import rsa_decrypt, rsa_encrypt
from utils.error.errno_utils import MarketCodeStatusCode, UserStatusCode
from utils.redis.connect import r


WECHAT = settings.WECHAT

# CIPHER_ALGORITHM = ('CAESAR', 'B64', 'RSA')
CIPHER_ALGORITHM = ('CAESAR', )

CIPHER_PREFIX = {
    'CAESAR': '0',
    'B64': '1',
    'RSA': '2',
}


def get_ciphertext(plaintext, optor_id, marketcode):
    # brand_id#model_id#distributor_id#sn#time
    # AAAA#AAAAAA#AAAAA#AAAAAAAAAAAAAA#180224
    brand_pk, model_pk, distributor_pk, sn, time = plaintext.split('#')

    mieli, created_at = MchInfoEncryptLogInfo.objects.get_or_create(plaintext=plaintext)

    if marketcode:
        if created_at or not mieli.code:
            with transaction.atomic():
                marketcode = MarketCodeInfo.objects.select_for_update().filter(has_used=False).first()

                if not marketcode:
                    return ''

                marketcode.has_used = True
                marketcode.save()

                mieli.application_id = marketcode.application_id
                mieli.code = marketcode.code
                mieli.code_url = marketcode.code_url
                mieli.brand_pk = brand_pk
                mieli.model_pk = model_pk
                mieli.distributor_pk = distributor_pk
                mieli.sn = sn
                mieli.operator_id = optor_id
                mieli.save()

        return mieli.code_url

    if created_at:
        alg = random.choice(CIPHER_ALGORITHM)

        if alg == 'CAESAR':
            ciphertext = caesar_encrypt(plaintext)
        elif alg == 'B64':
            ciphertext = b64_encrypt(plaintext)
        elif alg == 'RSA':
            ciphertext = rsa_encrypt(plaintext)
        else:
            ciphertext = plaintext

        mieli.alg = alg
        mieli.ciphertext = ciphertext
        mieli.brand_pk = brand_pk
        mieli.model_pk = model_pk
        mieli.distributor_pk = distributor_pk
        mieli.sn = sn
        mieli.operator_id = optor_id
        mieli.save()

    return u'{prefix}+{cipherlen}+{ciphertext}'.format(
        prefix=CIPHER_PREFIX.get(mieli.alg, ''),
        cipherlen=len(mieli.ciphertext),
        ciphertext=mieli.ciphertext,
    )


@logit(res=True)
def encrypt(request):
    plaintext = request.POST.get('plaintext', '')
    optor_id = request.POST.get('optor_id', '')
    marketcode = request.POST.get('marketcode', '')

    return response(200, data={
        'ciphertext': get_ciphertext(plaintext, optor_id, marketcode),
    })


@logit(res=True)
def decrypt(request, v='v2'):
    ciphertext = request.POST.get('ciphertext', '')

    prefix, cipherlen, ciphertext = ciphertext.split('+', 2)

    ciphertext = ciphertext[:int(cipherlen)]

    if prefix == CIPHER_PREFIX['CAESAR']:
        plaintext = caesar_decrypt(ciphertext)
    elif prefix == CIPHER_PREFIX['B64']:
        plaintext = b64_decrypt(ciphertext)
    elif prefix == CIPHER_PREFIX['RSA']:
        plaintext = rsa_decrypt(ciphertext)
    else:
        plaintext = ciphertext

    # brand_id#model_id#distributor_id#sn#time
    # AAAA#AAAAAA#AAAAA#AAAAAAAAAAAAAA#180224
    brand_pk, model_pk, distributor_pk, sn, time = plaintext.split('#')

    try:
        brand = BrandInfo.objects.get(pk=brand_pk)
    except BrandInfo.DoesNotExist:
        brand = None

    try:
        model = ModelInfo.objects.get(pk=model_pk)
    except ModelInfo.DoesNotExist:
        model = None

    mdli, created_at = MchInfoDecryptLogInfo.objects.get_or_create(ciphertext=ciphertext, defaults={
        'brand_pk': brand_pk,
        'model_pk': model_pk,
        'distributor_pk': distributor_pk,
        'sn': sn,
        'decrypt_count': 1,
    })

    if not created_at:
        mdli.decrypt_count += 1
        mdli.save()

    unbound = ConsumeShotUnbindingInfo.objects.filter(model_id=model.model_id, sn=sn, status=True).exists()

    # 弃用老版本的劵形式，和会员系统统一
    if v == 'v1':
        has_unexpired_activity = False
        coupon_infos = {}

    else:
        activities = ActivityInfo.objects.filter(status=True).order_by('-pk')
        coupon_infos = [act.coupon_info3 for act in activities if act and act.has_unexpired_activity(model.model_uni_name)]

    return response(200, data={
        'plaintext': plaintext,
        'logo_url': brand.brand_logo_url if brand else '',
        'model_imgs': model.images if model else [],
        'code_version': 1,
        'goodsInfo': {
            'BrandID': brand_pk,
            'Brand': brand.brand_name if brand else '',
            'ModelID': model_pk,
            'Model': (model.model_full_name or model.model_name) if model else '',
            'DistributorID': distributor_pk,
            'SerialNo': sn,
            'img': model.imgdata1 if model else '',
            'img2': model.imgdata if model else {},
        },
        'has_unexpired_activity': False,
        'coupon_infos': coupon_infos,
        'unbound': unbound,
    })


@logit(res=True)
def decrypt2(request, v='v2'):
    code_ticket = request.POST.get('code_ticket', '')
    code = request.POST.get('code', '')
    user_id = request.POST.get('user_id', '')

    if code_ticket and user_id:
        try:
            user = UserInfo.objects.get(user_id=user_id)
        except UserInfo.DoesNotExist:
            return response(UserStatusCode.USER_NOT_FOUND)

        wxcfg = WECHAT.get('JSAPI', {})

        appid = wxcfg.get('appID')
        secret = wxcfg.get('appsecret')

        code_info = tickettocode(code_ticket=code_ticket, openid=user.openid_miniapp, appid=appid, secret=secret, token=None, storage=RedisStorage(r))

        code = code_info.get('code', '')

        try:
            mieli = MchInfoEncryptLogInfo.objects.get(code=code)
        except MchInfoEncryptLogInfo.DoesNotExist:
            return response()
    else:
        code = re.sub(r'http://|https://', '', code)
        try:
            mieli = MchInfoEncryptLogInfo.objects.get(code_url=code)
        except MchInfoEncryptLogInfo.DoesNotExist:
            return response()

    plaintext = mieli.plaintext

    # brand_id#model_id#distributor_id#sn#time
    # AAAA#AAAAAA#AAAAA#AAAAAAAAAAAAAA#180224
    brand_pk, model_pk, distributor_pk, sn, time = plaintext.split('#')

    try:
        brand = BrandInfo.objects.get(pk=brand_pk)
    except BrandInfo.DoesNotExist:
        brand = None

    try:
        model = ModelInfo.objects.get(pk=model_pk)
    except ModelInfo.DoesNotExist:
        model = None

    mdli, created_at = MchInfoDecryptLogInfo.objects.get_or_create(
        application_id=mieli.application_id,
        code=mieli.code,
        code_url=mieli.code_url,
        defaults={
            'brand_pk': brand_pk,
            'model_pk': model_pk,
            'distributor_pk': distributor_pk,
            'sn': sn,
            'decrypt_count': 1,
        }
    )

    if not created_at:
        mdli.decrypt_count += 1
        mdli.save()

    unbound = ConsumeShotUnbindingInfo.objects.filter(model_id=model.model_id, sn=sn, status=True).exists()

    if v == 'v1':
        has_unexpired_activity = False
        coupon_infos = {}

    else:
        activities = ActivityInfo.objects.filter(status=True).order_by('-pk')
        coupon_infos = [act.coupon_info3 for act in activities if act.has_unexpired_activity(model.model_uni_name)]

    return response(200, data={
        'plaintext': plaintext,
        'logo_url': brand.brand_logo_url if brand else '',
        'model_imgs': model.images if model else [],
        'code_version': mieli.version,
        'goodsInfo': {
            'BrandID': brand_pk,
            'Brand': brand.brand_name if brand else '',
            'ModelID': model_pk,
            'Model': (model.model_full_name or model.model_name) if model else '',
            'DistributorID': distributor_pk,
            'SerialNo': sn,
            'img': model.imgdata1 if model else '',
            'img2': model.imgdata if model else {},
        },
        'has_unexpired_activity': False,
        'coupon_infos': coupon_infos,
        'unbound': unbound,
    })