RasadDam_Backend/apps/core/api.py

from rest_framework import viewsets

from apps.authentication.mixins.region_filter import RegionFilterMixin, get_organization_by_user
from apps.core.models import MobileTest, SystemConfig
from apps.core.serializers import MobileTestSerializer, SystemConfigSerializer


class BaseViewSet(RegionFilterMixin, viewsets.ModelViewSet):
    """
    All view sets in the project should inherit from this class.
    It applies region-based filtering automatically to GET (list) requests.
    """

    def get_queryset(self):
        queryset = super().get_queryset()
        request = self.request
        user = request.user
        org = get_organization_by_user(user)
        user_relation = user.user_relation.filter(organization=org)

        if self.request.method.lower() == 'get' and not self.kwargs.get('pk'):
            queryset = self.filter_by_region(queryset, org=True)
            print(queryset)

        if user_relation.exists():
            if not user_relation.first().role.type.key == 'ADM':
                model_name = queryset.model.__name__.lower()

                if model_name == 'userrelations':  # noqa
                    queryset = queryset.exclude(id=user_relation.first().id)

                elif model_name == 'organization':
                    queryset = queryset.exclude(id=user_relation.first().organization.id)

        return queryset


class MobileTestViewSet(viewsets.ModelViewSet):
    queryset = MobileTest.objects.all()
    serializer_class = MobileTestSerializer


class SystemConfigViewSet(viewsets.ModelViewSet):
    queryset = SystemConfig.objects.all()
    serializer_class = SystemConfigSerializer