mirror of
https://github.com/1Panel-dev/MaxKB.git
synced 2025-12-25 17:22:55 +00:00
108 lines
5.4 KiB
Python
108 lines
5.4 KiB
Python
# coding=utf-8
|
||
"""
|
||
@project: maxkb
|
||
@Author：虎
|
||
@file： Team.py
|
||
@date：2023/9/25 17:13
|
||
@desc:
|
||
"""
|
||
from drf_yasg.utils import swagger_auto_schema
|
||
from rest_framework.decorators import action
|
||
from rest_framework.views import APIView
|
||
from rest_framework.views import Request
|
||
|
||
from common.auth import TokenAuth, has_permissions
|
||
from common.constants.permission_constants import PermissionConstants
|
||
from common.log.log import log
|
||
from common.response import result
|
||
from setting.serializers.team_serializers import TeamMemberSerializer, get_response_body_api, \
|
||
UpdateTeamMemberPermissionSerializer
|
||
from django.utils.translation import gettext_lazy as _
|
||
|
||
from setting.views.common import get_member_operation_object, get_member_operation_object_batch
|
||
|
||
|
||
class TeamMember(APIView):
|
||
authentication_classes = [TokenAuth]
|
||
|
||
@action(methods=['GET'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Get a list of team members'),
|
||
operation_id=_('Get a list of team members'),
|
||
responses=result.get_api_response(get_response_body_api()),
|
||
tags=[_('Team')])
|
||
@has_permissions(PermissionConstants.TEAM_READ)
|
||
def get(self, request: Request):
|
||
return result.success(TeamMemberSerializer(data={'team_id': str(request.user.id)}).list_member())
|
||
|
||
@action(methods=['POST'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Add member'),
|
||
operation_id=_('Add member'),
|
||
request_body=TeamMemberSerializer().get_request_body_api(),
|
||
responses=result.get_default_response(),
|
||
tags=[_('Team')])
|
||
@has_permissions(PermissionConstants.TEAM_CREATE)
|
||
@log(menu='Team', operate='Add member',
|
||
get_operation_object=lambda r, k: {'name': r.data.get('username_or_email')})
|
||
def post(self, request: Request):
|
||
team = TeamMemberSerializer(data={'team_id': str(request.user.id)})
|
||
return result.success((team.add_member(**request.data)))
|
||
|
||
class Batch(APIView):
|
||
authentication_classes = [TokenAuth]
|
||
|
||
@action(methods=['POST'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Add members in batches'),
|
||
operation_id=_('Add members in batches'),
|
||
request_body=TeamMemberSerializer.get_bach_request_body_api(),
|
||
responses=result.get_api_array_response(TeamMemberSerializer.get_response_body_api()),
|
||
tags=[_('Team')])
|
||
@has_permissions(PermissionConstants.TEAM_CREATE)
|
||
@log(menu='Team', operate='Add members in batches',
|
||
get_operation_object=lambda r, k: get_member_operation_object_batch(r.data))
|
||
def post(self, request: Request):
|
||
return result.success(
|
||
TeamMemberSerializer(data={'team_id': request.user.id}).batch_add_member(request.data))
|
||
|
||
class Operate(APIView):
|
||
authentication_classes = [TokenAuth]
|
||
|
||
@action(methods=['GET'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Get team member permissions'),
|
||
operation_id=_('Get team member permissions'),
|
||
manual_parameters=TeamMemberSerializer.Operate.get_request_params_api(),
|
||
tags=[_('Team')])
|
||
@has_permissions(PermissionConstants.TEAM_READ)
|
||
def get(self, request: Request, member_id: str):
|
||
return result.success(TeamMemberSerializer.Operate(
|
||
data={'member_id': member_id, 'team_id': str(request.user.id)}).list_member_permission())
|
||
|
||
@action(methods=['PUT'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Update team member permissions'),
|
||
operation_id=_('Update team member permissions'),
|
||
request_body=UpdateTeamMemberPermissionSerializer().get_request_body_api(),
|
||
responses=result.get_default_response(),
|
||
manual_parameters=TeamMemberSerializer.Operate.get_request_params_api(),
|
||
tags=[_('Team')]
|
||
)
|
||
@has_permissions(PermissionConstants.TEAM_EDIT)
|
||
@log(menu='Team', operate='Update team member permissions',
|
||
get_operation_object=lambda r, k: get_member_operation_object(k.get('member_id'))
|
||
)
|
||
def put(self, request: Request, member_id: str):
|
||
return result.success(TeamMemberSerializer.Operate(
|
||
data={'member_id': member_id, 'team_id': str(request.user.id)}).edit(request.data))
|
||
|
||
@action(methods=['DELETE'], detail=False)
|
||
@swagger_auto_schema(operation_summary=_('Remove member'),
|
||
operation_id=_('Remove member'),
|
||
manual_parameters=TeamMemberSerializer.Operate.get_request_params_api(),
|
||
responses=result.get_default_response(),
|
||
tags=[_('Team')]
|
||
)
|
||
@has_permissions(PermissionConstants.TEAM_DELETE)
|
||
@log(menu='Team', operate='Remove member',
|
||
get_operation_object=lambda r, k: get_member_operation_object(k.get('member_id')))
|
||
def delete(self, request: Request, member_id: str):
|
||
return result.success(TeamMemberSerializer.Operate(
|
||
data={'member_id': member_id, 'team_id': str(request.user.id)}).delete())
|