bookshop_admin/apps/WechatApplet/views.py

# coding=utf-8

import json
from django.db import transaction
from rest_framework.views import APIView
from rest_framework.decorators import action

from utils.custom_modelviewset import CustomModelViewSet
from utils.permission import isLogin
from utils import response_ok
from apps.log.models import BizLog
# from apps.admin import admin_log

from .serializers import WechatAppletSerializer, MsgTemplateSerializer
from apps.WechatApplet.filters import WechatAppletFilter, MsgTemplateFilter
from apps.WechatApplet.models import WechatApplet, MessageTemplate
from apps.WechatTp.models import WechatTp
from django.contrib.auth import get_user_model

User = get_user_model()
from apps.account.models import CustomerWechat


class TemplateListViewSet(APIView):
    permission_classes = [isLogin, ]

    def get(self, request):
        tp = WechatTp.getDefault()
        ret = tp.getTemplateList()
        return response_ok(ret)

class DraftTemplateListViewSet(APIView):
    permission_classes = [isLogin, ]

    def get(self, request):
        tp = WechatTp.getDefault()
        ret = tp.getDraftTemplateList()
        return response_ok(ret)


class DictListView(APIView):
    permission_classes = [isLogin, ]

    def get(self, request):
        data = {
            'temp_types': [],
            'wechat_apps': [],
        }

        for type in MessageTemplate.TYPE_CHOICE:
            item = {
                'id': type[0],
                'name': type[1],
            }
            data['temp_types'].append(item)

        rows = WechatApplet.objects.filter(is_authorize=True)
        for row in rows:
            item = {
                'id': row.id,
                'name': row.nick_name,
            }
            data['wechat_apps'].append(item)

        return response_ok(data)


class WxMsgTempView(APIView):

    def get(self, request):
        appid = request.GET.get('appid')
        data = []
        if request.user and request.user.is_authenticated:
            wechat = CustomerWechat.objects.filter(customer=request.user, wechat_app__authorizer_appid=appid).first()
            if not wechat:
                return response_ok(data)
            rows = MessageTemplate.objects.filter(wechat_app=wechat.wechat_app)
            if request.user.type == User.EMPLOYEE:
                rows = rows.filter(type__in=[MessageTemplate.CREATE_ORDER, MessageTemplate.PAY_ORDER])
            elif request.user.type == User.AGENT:
                rows = rows.filter(
                    type__in=[MessageTemplate.CONFIRM_ORDER, MessageTemplate.DELIVER, MessageTemplate.CREATE_ORDER])
            else:
                rows = rows.filter(type=MessageTemplate.CREATE_ORDER)

            rows = rows.values_list('template', flat=True)
            return response_ok(rows)
        return response_ok(data)


class WechatAppletViewSet(CustomModelViewSet):
    permission_classes = [isLogin, ]
    queryset = WechatApplet.objects.filter(is_admin=False)
    serializer_class = WechatAppletSerializer

    def filter_queryset(self, queryset):
        f = WechatAppletFilter(self.request.GET, queryset=queryset)
        return f.qs

    @action(methods=['post'], detail=True)
    def upload_code(self, request, pk):
        # 上传代码
        template_id = request.POST.get('template_id')
        user_version = request.POST.get('user_version')
        user_desc = request.POST.get('user_desc')
        app = WechatApplet.getById(pk)
        with transaction.atomic():
            app.uploadCode(template_id, user_version, user_desc)
            BizLog.objects.addnew(self.request.user, BizLog.INSERT, u'小程序上传审核代码，id=%d' % app.id, request.POST.dict())
        return response_ok()

    @action(methods=['post'], detail=False)
    def batch_upload_code(self, request):
        # 批量上传代码
        template_id = request.POST.get('template_id')
        user_version = request.POST.get('user_version')
        user_desc = request.POST.get('user_desc')
        app_ids = json.loads(request.POST.get('ids'))

        apps = WechatApplet.objects.filter(id__in=app_ids)
        for app in apps:
            try:
                with transaction.atomic():
                    app.uploadCode(template_id, user_version, user_desc)
            except:
                pass
        # admin_log(request.user, BizLog.INSERT, u'小程序批量上传审核代码', request.POST.dict())

        return response_ok()

    @action(methods=['post'], detail=False)
    def add_to_temp(self, request):
        # 上传草稿箱模板到标准模板
        draft_id = request.POST.get('draft_id')
        tp = WechatTp.getDefault()
        tp.addToemplate(draft_id)
        with transaction.atomic():
            BizLog.objects.addnew(self.request.user, BizLog.INSERT, u'小程序上传草稿到标准模板，draft_id=%s' % draft_id)
        return response_ok()

    @action(methods=['post'], detail=False)
    def refresh_audit_status(self, request):
        # 查询最新一次提交代码的审核状态  多选更新 审核状态
        app_ids = json.loads(request.POST.get('ids'))
        for app_id in app_ids:
            app = WechatApplet.getById(app_id)
            app.refreshAuditStatus()
        return response_ok()

    @action(methods=['post'], detail=True)
    def set_secret(self, request, pk):
        secret = request.POST.get('secret')
        app = WechatApplet.getById(pk)
        with transaction.atomic():
            app.secret = secret
            app.save()
            # admin_log(request.user, BizLog.INSERT, u'设置小程序秘钥, id=%d' % app.id, secret)
        return response_ok()

    @action(methods=['post'], detail=True)
    def set_merchant(self, request, pk):
        agent_num = request.POST.get('agent_num')
        agent_key = request.POST.get('agent_key')
        app = WechatApplet.getById(pk)
        with transaction.atomic():
            app.agent_num = agent_num
            app.agent_key = agent_key
            app.save()
            # admin_log(request.user, BizLog.INSERT, u'设置小程序商户号和商户秘钥, id=%d' % app.id)
        return response_ok()

    @action(methods=['post'], detail=False)
    def add_plugin(self, request):
        app_ids = json.loads(request.POST.get('ids'))
        for app_id in app_ids:
            apps = WechatApplet.getById(app_id)
            apps.addPlugin()
        return response_ok()

    @action(methods=['post'], detail=False)
    def release(self, request):
        app_ids = json.loads(request.POST.get('ids'))
        for app_id in app_ids:
            app = WechatApplet.getById(app_id)
            app.releaseApplet()
        return response_ok()

    @action(methods=['post'], detail=True)
    def upload_cert(self, request, pk):
        file = request.FILES.get('file', None)
        app = WechatApplet.getById(pk)
        with transaction.atomic():
            app.upload_cert_file(file)
            # admin_log(request.user, BizLog.INSERT, u'设置小程序企业支付api证书, id=%d' % app.id)
        return response_ok()


class MsgTemplateViewSet(CustomModelViewSet):
    permission_classes = [isLogin, ]
    queryset = MessageTemplate.objects.filter()
    serializer_class = MsgTemplateSerializer

    def filter_queryset(self, queryset):
        f = MsgTemplateFilter(self.request.GET, queryset=queryset)
        return f.qs

    def perform_create(self, serializer):
        super(MsgTemplateViewSet, self).perform_create(serializer)
        instance = serializer.instance
        validated_data = serializer.validated_data
        BizLog.objects.addnew(self.request.user, BizLog.INSERT,
                              u'添加模板[%s]，id=%d' % (instance.wechat_app.principal_name, instance.id), validated_data)

    def perform_update(self, serializer):
        super(MsgTemplateViewSet, self).perform_update(serializer)
        instance = serializer.instance
        validated_data = serializer.validated_data
        BizLog.objects.addnew(self.request.user, BizLog.UPDATE,
                              u'修改模板[%s]，id=%d' % (instance.wechat_app.principal_name, instance.id), validated_data)

    def destroy(self, request, *args, **kwargs):
        instance = self.get_object()
        BizLog.objects.addnew(self.request.user, BizLog.DELETE,
                              u'删除模板[%s]，id=%d' % (instance.wechat_app.principal_name, instance.id))
        super(MsgTemplateViewSet, self).perform_destroy(instance)
        return response_ok()