mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-05-31 10:55:03 +00:00
fa68389028
* perf: 去掉单独的flash msg perf: 修改使用库 * fix: guangbug * pref: 修改 context Co-authored-by: ibuler <ibuler@qq.com> Co-authored-by: xinwen <coderWen@126.com>
38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
from rest_framework.views import Response
|
|
from rest_framework.generics import GenericAPIView
|
|
from rest_framework.exceptions import APIException
|
|
from rest_framework import status
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from django.conf import settings
|
|
from common.permissions import IsSuperUser
|
|
from common.message.backends.dingtalk import DingTalk
|
|
|
|
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)
|
|
|
|
app_key = serializer.validated_data['DINGTALK_APPKEY']
|
|
agent_id = serializer.validated_data['DINGTALK_AGENTID']
|
|
app_secret = serializer.validated_data.get('DINGTALK_APPSECRET') \
|
|
or settings.DINGTALK_APPSECRET \
|
|
or ''
|
|
|
|
try:
|
|
dingtalk = DingTalk(appid=app_key, appsecret=app_secret, agentid=agent_id)
|
|
dingtalk.send_text(['test'], 'test')
|
|
return Response(status=status.HTTP_200_OK, data={'msg': _('Test success')})
|
|
except APIException as e:
|
|
if 'errmsg' in e.detail:
|
|
error = e.detail['errmsg']
|
|
else:
|
|
error = e.detail
|
|
return Response(status=status.HTTP_400_BAD_REQUEST, data={'error': error})
|