mirror of https://github.com/jumpserver/jumpserver
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
|
import requests
|
||
|
|
|
||
|
|
from rest_framework.views import Response
|
||
|
|
from rest_framework.generics import GenericAPIView
|
||
|
|
from django.utils.translation import gettext_lazy as _
|
||
|
|
|
||
|
|
from common.permissions import IsSuperUser
|
||
|
|
from common.message.backends.dingtalk import URL
|
||
|
|
|
||
|
|
from .. import serializers
|
||
|
|
|
||
|
|
|
||
|
|
class DingTalkTestingAPI(GenericAPIView):
|
||
|
|
permission_classes = (IsSuperUser,)
|
||
|
|
serializer_class = serializers.DingTalkSettingSerializer
|
||
|
|
|
||
|
|
def post(self, request):
|
||
|
|
serializer = self.serializer_class(data=request.data)
|
||
|
|
serializer.is_valid(raise_exception=True)
|
||
|
|
|
||
|
|
dingtalk_appkey = serializer.validated_data['DINGTALK_APPKEY']
|
||
|
|
dingtalk_agentid = serializer.validated_data['DINGTALK_AGENTID']
|
||
|
|
dingtalk_appsecret = serializer.validated_data['DINGTALK_APPSECRET']
|
||
|
|
|
||
|
|
try:
|
||
|
|
params = {'appkey': dingtalk_appkey, 'appsecret': dingtalk_appsecret}
|
||
|
|
resp = requests.get(url=URL.GET_TOKEN, params=params)
|
||
|
|
if resp.status_code != 200:
|
||
|
|
return Response(status=400, data={'error': resp.json()})
|
||
|
|
|
||
|
|
data = resp.json()
|
||
|
|
errcode = data['errcode']
|
||
|
|
if errcode != 0:
|
||
|
|
return Response(status=400, data={'error': data['errmsg']})
|
||
|
|
|
||
|
|
return Response(status=200, data={'msg': _('OK')})
|
||
|
|
except Exception as e:
|
||
|
|
return Response(status=400, data={'error': str(e)})
|