yzk_wechat_event/apps/jqr/views.py

import requests
from rest_framework import viewsets
from rest_framework.decorators import action
from rest_framework.response import Response
from django.conf import settings

from apps.jqr.parsers import WechatContentParser
from apps.jqr.renderers import WechatPublicContentRenderer
from apps.jqr.serializers import WechatPublicTokenSerializer, WechatEncryptSerializer
from apps.qc.models import QcCorpinfo


class WechatWorkerViewSet(viewsets.GenericViewSet):

    @action(methods=['GET'], detail=False, serializer_class=WechatPublicTokenSerializer,
            url_path=r'callback/(?P<path>\w+)', )
    def verify(self, request, path, *args, **kwargs):
        if not settings.DEBUG:
            params = request.query_params
            # cbtest.xbtool.cn
            url = f'https://46i859447w.yicp.fun/event/callback/{path}'
            res = requests.get(url, params=params)
            data = res.text
            return Response(data=int(data))

        [business, corpid, agentid, uid] = path.split('_')
        corp = QcCorpinfo.objects.get(corpid=corpid, agentid=agentid, uid=uid)
        context = self.get_serializer_context()
        context['corp'] = corp
        serializer = self.get_serializer(data=request.query_params, context=context)
        serializer.is_valid(raise_exception=True)
        enchostr = serializer.data.get('echostr')
        return Response(data=int(enchostr))
        # return Response(data=enchostr)

    @verify.mapping.post
    def message(self, request, path, *args, **kwargs):
        print(request.data)
        if not settings.DEBUG:
            data = request.data
            # cbtest.xbtool.cn
            url = f'https://46i859447w.yicp.fun/event/callback/{path}'
            res = requests.post(url, data=data, headers=request.headers)
            return Response(data=res.json())
        serializer = WechatEncryptSerializer(data=request.data)
        serializer.is_valid(raise_exception=True)
        return Response(data=serializer.data)

    def get_parsers(self):
        if self.request.method == 'POST':
            return [WechatContentParser()]
        return super().get_parsers()

    def get_renderers(self):
        if self.request.method == 'POST':
            return [WechatPublicContentRenderer()]
        return super().get_renderers()