mirror of https://github.com/jumpserver/jumpserver
38 lines
1.5 KiB
Python
38 lines
1.5 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 settings.models import Setting
|
|
from common.permissions import IsSuperUser
|
|
from common.message.backends.wecom import WeCom
|
|
|
|
from .. import serializers
|
|
|
|
|
|
class WeComTestingAPI(GenericAPIView):
|
|
permission_classes = (IsSuperUser,)
|
|
serializer_class = serializers.WeComSettingSerializer
|
|
|
|
def post(self, request):
|
|
serializer = self.serializer_class(data=request.data)
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
wecom_corpid = serializer.validated_data['WECOM_CORPID']
|
|
wecom_agentid = serializer.validated_data['WECOM_AGENTID']
|
|
wecom_corpsecret = serializer.validated_data.get('WECOM_SECRET')
|
|
|
|
if not wecom_corpsecret:
|
|
secret = Setting.objects.filter(name='WECOM_SECRET').first()
|
|
if not secret:
|
|
return Response(status=status.HTTP_400_BAD_REQUEST, data={'error': _('Secret is required')})
|
|
wecom_corpsecret = secret.cleaned_value
|
|
|
|
try:
|
|
wecom = WeCom(corpid=wecom_corpid, corpsecret=wecom_corpsecret, agentid=wecom_agentid)
|
|
wecom.send_text(['test'], 'test')
|
|
return Response(status=status.HTTP_200_OK, data={'msg': _('OK')})
|
|
except APIException as e:
|
|
return Response(status=status.HTTP_400_BAD_REQUEST, data={'error': e.detail})
|