UnisKB/apps/system_manage/views/email_setting.py

# coding=utf-8
"""
    @project: maxkb
    @Author：虎
    @file： system_setting.py
    @date：2024/3/19 16:01
    @desc:
"""
from drf_spectacular.utils import extend_schema
from rest_framework.request import Request
from rest_framework.views import APIView

from common.auth import TokenAuth
from common.auth.authentication import has_permissions
from common.constants.permission_constants import PermissionConstants

from django.utils.translation import gettext_lazy as _

from common.result import result
from models_provider.api.model import DefaultModelResponse
from system_manage.api.email_setting import EmailSettingAPI
from system_manage.serializers.email_setting import EmailSettingSerializer


class SystemSetting(APIView):
    class Email(APIView):
        authentication_classes = [TokenAuth]

        @extend_schema(methods=['PUT'],
                       summary=_('Create or update email settings'),
                       description=_('Create or update email settings'),
                       operation_id=_('Create or update email settings'),  # type: ignore
                       request=EmailSettingAPI.get_request(),
                       responses=EmailSettingAPI.get_response(),
                       tags=[_('Email Settings')])  # type: ignore
        @has_permissions(PermissionConstants.EMAIL_SETTING_EDIT)
        def put(self, request: Request):
            return result.success(
                EmailSettingSerializer.Create(
                    data=request.data).update_or_save())

        @extend_schema(
            methods=['POST'],
            summary=_('Test email settings'),
            operation_id=_('Test email settings'),  # type: ignore
            request=EmailSettingAPI.get_request(),
            responses=DefaultModelResponse.get_response(),
            tags=[_('Email Settings')]  # type: ignore
        )
        @has_permissions(PermissionConstants.EMAIL_SETTING_EDIT)
        def post(self, request: Request):
            return result.success(
                EmailSettingSerializer.Create(
                    data=request.data).is_valid())

        @extend_schema(methods=['GET'],
                       summary=_('Get email settings'),
                       description=_('Get email settings'),
                       operation_id=_('Get email settings'),  # type: ignore
                       responses=DefaultModelResponse.get_response(),
                       tags=[_('Email Settings')])  # type: ignore
        @has_permissions(PermissionConstants.EMAIL_SETTING_READ)
        def get(self, request: Request):
            return result.success(
                EmailSettingSerializer.one())