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.feishu import FeiShu from .. import serializers class FeiShuTestingAPI(GenericAPIView): permission_classes = (IsSuperUser,) serializer_class = serializers.FeiShuSettingSerializer def post(self, request): serializer = self.serializer_class(data=request.data) serializer.is_valid(raise_exception=True) app_id = serializer.validated_data['FEISHU_APP_ID'] app_secret = serializer.validated_data.get('FEISHU_APP_SECRET') if not app_secret: secret = Setting.objects.filter(name='FEISHU_APP_SECRET').first() if secret: app_secret = secret.cleaned_value app_secret = app_secret or '' try: feishu = FeiShu(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})