mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.6 KiB
45 lines
1.6 KiB
from django.utils.translation import gettext_lazy as _ |
|
from rest_framework import status |
|
from rest_framework.exceptions import APIException |
|
from rest_framework.generics import GenericAPIView |
|
from rest_framework.views import Response |
|
|
|
from common.sdk.im.feishu import FeiShu |
|
from common.sdk.im.lark import Lark |
|
from settings.models import Setting |
|
from .. import serializers |
|
|
|
|
|
class FeiShuTestingAPI(GenericAPIView): |
|
category = 'FEISHU' |
|
serializer_class = serializers.FeiShuSettingSerializer |
|
rbac_perms = { |
|
'POST': 'settings.change_auth' |
|
} |
|
|
|
def post(self, request): |
|
serializer = self.serializer_class(data=request.data) |
|
serializer.is_valid(raise_exception=True) |
|
|
|
app_id = serializer.validated_data[f'{self.category}_APP_ID'] |
|
app_secret = serializer.validated_data.get(f'{self.category}_APP_SECRET') |
|
|
|
if not app_secret: |
|
secret = Setting.objects.filter(name=f'{self.category}_APP_SECRET').first() |
|
if secret: |
|
app_secret = secret.cleaned_value |
|
|
|
app_secret = app_secret or '' |
|
|
|
auth_cls = FeiShu if self.category == 'FEISHU' else Lark |
|
|
|
try: |
|
feishu = auth_cls(app_id=app_id, app_secret=app_secret) |
|
feishu.send_text(['test'], 'test') |
|
return Response(status=status.HTTP_200_OK, data={'msg': _('Test success')}) |
|
except APIException as e: |
|
try: |
|
error = e.detail['errmsg'] |
|
except: |
|
error = e.detail |
|
return Response(status=status.HTTP_400_BAD_REQUEST, data={'error': error})
|
|
|