mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-26 10:12:51 +00:00
106 lines
2.8 KiB
Python
106 lines
2.8 KiB
Python
# coding=utf-8
|
|
from drf_spectacular.types import OpenApiTypes
|
|
from drf_spectacular.utils import OpenApiParameter
|
|
|
|
from common.mixins.api_mixin import APIMixin
|
|
from common.result import ResultSerializer
|
|
from modules.models.module import ModuleCreateRequest, ModuleEditRequest
|
|
from modules.serializers.module import ModuleSerializer
|
|
|
|
|
|
class ModuleCreateResponse(ResultSerializer):
|
|
def get_data(self):
|
|
return ModuleSerializer()
|
|
|
|
|
|
class ModuleCreateAPI(APIMixin):
|
|
@staticmethod
|
|
def get_parameters():
|
|
return [
|
|
OpenApiParameter(
|
|
name="workspace_id",
|
|
description="工作空间id",
|
|
type=OpenApiTypes.STR,
|
|
location='path',
|
|
required=True,
|
|
),
|
|
OpenApiParameter(
|
|
name="source",
|
|
description="菜单",
|
|
type=OpenApiTypes.STR,
|
|
enum=["APPLICATION", "KNOWLEDGE", "TOOL"],
|
|
location='path',
|
|
required=True,
|
|
)
|
|
]
|
|
|
|
@staticmethod
|
|
def get_request():
|
|
return ModuleCreateRequest
|
|
|
|
@staticmethod
|
|
def get_response():
|
|
return ModuleCreateResponse
|
|
|
|
|
|
class ModuleReadAPI(APIMixin):
|
|
@staticmethod
|
|
def get_parameters():
|
|
return [
|
|
OpenApiParameter(
|
|
name="workspace_id",
|
|
description="工作空间id",
|
|
type=OpenApiTypes.STR,
|
|
location='path',
|
|
required=True,
|
|
),
|
|
OpenApiParameter(
|
|
name="source",
|
|
description="菜单",
|
|
type=OpenApiTypes.STR,
|
|
enum=["APPLICATION", "KNOWLEDGE", "TOOL"],
|
|
location='path',
|
|
required=True,
|
|
),
|
|
OpenApiParameter(
|
|
name="module_id",
|
|
description="模块id",
|
|
type=OpenApiTypes.STR,
|
|
location='path',
|
|
required=True,
|
|
)
|
|
]
|
|
|
|
@staticmethod
|
|
def get_response():
|
|
return ModuleCreateResponse
|
|
|
|
|
|
class ModuleEditAPI(ModuleReadAPI):
|
|
|
|
@staticmethod
|
|
def get_request():
|
|
return ModuleEditRequest
|
|
|
|
|
|
class ModuleTreeReadAPI(APIMixin):
|
|
@staticmethod
|
|
def get_parameters():
|
|
return [
|
|
OpenApiParameter(
|
|
name="workspace_id",
|
|
description="工作空间id",
|
|
type=OpenApiTypes.STR,
|
|
location='path',
|
|
required=True,
|
|
),
|
|
OpenApiParameter(
|
|
name="source",
|
|
description="菜单",
|
|
type=OpenApiTypes.STR,
|
|
enum=["APPLICATION", "KNOWLEDGE", "TOOL"],
|
|
location='path',
|
|
required=True,
|
|
)
|
|
]
|