58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
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()
|