centurion_erp/app/access/viewsets/team_user.py

from drf_spectacular.utils import extend_schema, extend_schema_view, OpenApiParameter, OpenApiResponse

from access.models.team import Team

from access.serializers.team_user import (    # pylint: disable=W0611:unused-import
    TeamUsers,
    TeamUserModelSerializer,
    TeamUserViewSerializer
)

from api.viewsets.common import ModelViewSet



@extend_schema_view(
    create=extend_schema(
        summary = 'Create a user within this team',
        description='',
        parameters = [
            OpenApiParameter(
                name = 'organization_id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'team_id',
                location = 'path',
                type = int
            ),
        ],
        responses = {
            # 200: OpenApiResponse(description='Allready exists', response=TeamUserViewSerializer),
            201: OpenApiResponse(description='Created', response=TeamUserViewSerializer),
            # 400: OpenApiResponse(description='Validation failed.'),
            403: OpenApiResponse(description='User is missing add permissions'),
        }
    ),
    destroy = extend_schema(
        summary = 'Delete a user from this team',
        description = '',
        parameters = [
            OpenApiParameter(
                name = 'id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'organization_id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'team_id',
                location = 'path',
                type = int
            ),
        ],
        responses = {
            204: OpenApiResponse(description=''),
            403: OpenApiResponse(description='User is missing delete permissions'),
        }
    ),
    list = extend_schema(
        summary = 'Fetch all users from this team',
        description='',
        parameters = [
            OpenApiParameter(
                name = 'organization_id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'team_id',
                location = 'path',
                type = int
            ),
        ],
        responses = {
            200: OpenApiResponse(description='', response=TeamUserViewSerializer),
            403: OpenApiResponse(description='User is missing view permissions'),
        }
    ),
    retrieve = extend_schema(
        summary = 'Fetch a single user from this team',
        description='',
        parameters = [
            OpenApiParameter(
                name = 'id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'organization_id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'team_id',
                location = 'path',
                type = int
            ),
        ],
        responses = {
            200: OpenApiResponse(description='', response=TeamUserViewSerializer),
            403: OpenApiResponse(description='User is missing view permissions'),
        }
    ),
    update = extend_schema(exclude = True),
    partial_update = extend_schema(
        summary = 'Update a user within this team',
        description = '',
        parameters = [
            OpenApiParameter(
                name = 'id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'organization_id',
                location = 'path',
                type = int
            ),
            OpenApiParameter(
                name = 'team_id',
                location = 'path',
                type = int
            ),
        ],
        responses = {
            200: OpenApiResponse(description='', response=TeamUserViewSerializer),
            # 201: OpenApiResponse(description='Created', response=OrganizationViewSerializer),
            # # 400: OpenApiResponse(description='Validation failed.'),
            403: OpenApiResponse(description='User is missing change permissions'),
        }
    ),
)
class ViewSet( ModelViewSet ):

    filterset_fields = [
        'manager',
        'team__organization',
    ]

    search_fields = []

    model = TeamUsers

    parent_model = Team

    parent_model_pk_kwarg = 'team_id'

    view_description = 'Users belonging to a single team'


    def get_back_url(self) -> str:

        if(
            getattr(self, '_back_url', None) is None
        ):

            return_model =Team.objects.get(
                pk = self.kwargs['team_id']
            )

            self._back_url = str(
                return_model.get_url( self.request )
            )

        return self._back_url


    def get_queryset(self):

        if self.queryset is not None:

            return self.queryset

        self.queryset = super().get_queryset()

        self.queryset = self.queryset.filter(
            team_id = self.kwargs['team_id']
        )

        return self.queryset


    def get_serializer_class(self):

        if (
            self.action == 'list'
            or self.action == 'retrieve'
        ):

            self.serializer_class = globals()[str( self.model._meta.verbose_name).replace(' ' , '') + 'ViewSerializer']

        else:

            self.serializer_class = globals()[str( self.model._meta.verbose_name).replace(' ' , '') + 'ModelSerializer']


        return self.serializer_class


    def get_return_url(self):

        if getattr(self, '_get_return_url', None):

            return self._get_return_url

        if self.kwargs.get('pk', None) is None:

            return_model = Team.objects.get(
                pk = self.kwargs['team_id']
            )

            self._get_return_url = return_model.get_url( self.request )

            return self._get_return_url

        return None