mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 10:12:51 +00:00
73 lines
3.3 KiB
Python
73 lines
3.3 KiB
Python
from django.db.models import QuerySet
|
|
from drf_spectacular.utils import extend_schema
|
|
from rest_framework.request import Request
|
|
from rest_framework.views import APIView
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from application.api.application_api_key import ApplicationKeyCreateAPI
|
|
from application.models import ApplicationApiKey
|
|
from application.serializers.application_api_key import ApplicationKeySerializer
|
|
from common.auth import TokenAuth
|
|
from common.auth.authentication import has_permissions
|
|
from common.constants.permission_constants import PermissionConstants
|
|
from common.log.log import log
|
|
from common.result import result, success
|
|
|
|
|
|
def get_application_operation_object(application_api_key_id):
|
|
application_api_key_model = QuerySet(model=ApplicationApiKey).filter(id=application_api_key_id).first()
|
|
if application_api_key_model is not None:
|
|
return {
|
|
"name": application_api_key_model.name
|
|
}
|
|
return {}
|
|
|
|
|
|
class ApplicationKey(APIView):
|
|
authentication_classes = [TokenAuth]
|
|
|
|
@extend_schema(
|
|
methods=['POST'],
|
|
description=_('Create application ApiKey'),
|
|
summary=_('Create application ApiKey'),
|
|
operation_id=_('Create application ApiKey'), # type: ignore
|
|
parameters=ApplicationKeyCreateAPI.get_parameters(),
|
|
tags=[_('Application Api Key')] # type: ignore
|
|
)
|
|
@log(menu='Application', operate="Add ApiKey",
|
|
get_operation_object=lambda r, k: get_application_operation_object(k.get('application_api_key_id')))
|
|
@has_permissions(PermissionConstants.APPLICATION_OVERVIEW_API_KEY.get_workspace_application_permission())
|
|
def post(self, request: Request, application_id: str, workspace_id: str):
|
|
return result.success(ApplicationKeySerializer(
|
|
data={'application_id': application_id, 'user_id': request.user.id,
|
|
'workspace_id': workspace_id}).generate())
|
|
|
|
@extend_schema(
|
|
methods=['GET'],
|
|
description=_('GET application ApiKey List'),
|
|
summary=_('Create application ApiKey List'),
|
|
operation_id=_('Create application ApiKey List'), # type: ignore
|
|
parameters=ApplicationKeyCreateAPI.get_parameters(),
|
|
tags=[_('Application Api Key')] # type: ignore
|
|
)
|
|
@has_permissions(PermissionConstants.APPLICATION_OVERVIEW_API_KEY.get_workspace_application_permission())
|
|
def get(self, request: Request, application_id: str, workspace_id: str):
|
|
return result, success(ApplicationKeySerializer(
|
|
data={'application_id': application_id, 'user_id': request.user.id,
|
|
'workspace_id': workspace_id}).list())
|
|
|
|
class Operate(APIView):
|
|
authentication_classes = [TokenAuth]
|
|
|
|
@extend_schema(
|
|
methods=['GET'],
|
|
description=_('GET application ApiKey List'),
|
|
summary=_('Create application ApiKey List'),
|
|
operation_id=_('Create application ApiKey List'), # type: ignore
|
|
parameters=ApplicationKeyCreateAPI.get_parameters(),
|
|
tags=[_('Application Api Key')] # type: ignore
|
|
)
|
|
@has_permissions(PermissionConstants.APPLICATION_OVERVIEW_API_KEY.get_workspace_application_permission())
|
|
def put(self, request: Request, application_id: str, workspace_id: str):
|
|
return result.success(ApplicationKeySerializer.Operate())
|