Base list pagination views

from typing import Dict, Type

from rest_framework.generics import ListAPIView
from rest_framework.response import Response

from ..paginator import PageNumberPagination
from .response_format import RestGetResponseMixin


__all__ = ('ListAPIViewPagination', )


class ListAPIViewPagination(RestGetResponseMixin, ListAPIView):
    pagination_class: Type[PageNumberPagination]
    is_paginate = True

    def list(self, request, *args, **kwargs):
        self.is_paginate = bool(request.GET.get('pagination', True))
        if not self.is_paginate:
            self.pagination_class = None
        return super().list(request, *args, **kwargs)
    
    def get_response_meta(self, response) -> Dict:
        meta = super().get_response_meta(response)
        if self.is_paginate:
            meta['pagination'] = response.data.get('pagination', {})
        return meta

    def get_response_data(self, response) -> Dict:
        if self.is_paginate:
            return response.data.get('data')
        return super().get_response_data(response)