MeridynPharma-Backend/core/apps/dashboard/views/pharmacy.py

# django
from django.db.models import Q

# rest framework
from rest_framework import viewsets, permissions
from rest_framework.decorators import action

# drf yasg
from drf_yasg import openapi
from drf_yasg.utils import swagger_auto_schema

# dashboard
from core.apps.dashboard.serializers import pharmacy as serializers

# shared
from core.apps.shared.models import Pharmacy
from core.apps.shared.utils.response_mixin import ResponseMixin


class PharmacyViewSet(viewsets.GenericViewSet, ResponseMixin):
    permission_classes = [permissions.IsAdminUser]
    queryset = Pharmacy.objects.all()

    def get_serializer_class(self):
        if self.action == "post": 
            return serializers.AdminPharmacyCreateSerializer
        elif self.action in ["patch", "put"]:
            return serializers.PharmacyUpdateSerializer
        else:
            return serializers.PharmacyListSerializer

    @swagger_auto_schema(
        tags=['Admin Pharmacies'],
        manual_parameters=[
            openapi.Parameter(
                in_=openapi.IN_QUERY,
                name="name",
                description="name bo'yicha search",
                required=False,
                type=openapi.TYPE_STRING,
            ),
            openapi.Parameter(
                in_=openapi.IN_QUERY,
                name="place",
                description="obyekt name bo'yicha search",
                required=False,
                type=openapi.TYPE_STRING,
            ),
            openapi.Parameter(
                in_=openapi.IN_QUERY,
                name="district",
                description="tuman name bo'yicha search",
                required=False,
                type=openapi.TYPE_STRING,
            ),
            openapi.Parameter(
                in_=openapi.IN_QUERY,
                name="user",
                description="qo'shgan foydalanuvchini ism va familiyasi bo'yicha qidirish",
                required=False,
                type=openapi.TYPE_STRING,
            ),
        ],
    )
    @action(detail=False, methods=['get'], url_path="list")
    def get(self, request):
        try:
            # params
            name = request.query_params.get('name', None)
            place_name = request.query_params.get('place', None)
            district_name = request.query_params.get('district', None)
            user_full_name = request.query_params.get('user', None)

            queryset = self.queryset.all()

            # filters
            if name is not None:
                queryset = queryset.filter(name__istartswith=name)
            
            if district_name is not None:
                queryset = queryset.filter(district__name__istartswith=district_name)
            
            if place_name is not None:
                queryset = queryset.filter(place__name__istartswith=place_name)

            if user_full_name is not None:
                queryset = queryset.filter(
                    Q(user__first_name__istartswith=user_full_name) |
                    Q(user__last_name__istartswith=user_full_name) 
                )

            page = self.paginate_queryset(queryset)
            if page is not None:
                serializer = self.get_serializer(page, many=True)
                return self.success_response(
                    data=self.get_paginated_response(serializer.data).data,
                    message='malumotlar fetch qilindi'
                )
            serializer = self.get_serializer(queryset, many=True)
            return self.success_response(
                data=serializer.data,
                message='malumotlar fetch qilindi'
            )
        except Exception as e:
            return self.error_response(
                data=str(e),
                message="xatolik"
            )

    @swagger_auto_schema(
        tags=['Admin Pharmacies']
    )
    @action(detail=False, methods=['post'], url_path='create')
    def post(self, request):
        try:
            serializer = self.get_serializer(data=request.data)
            if serializer.is_valid():
                obj = serializer.save()
                return self.success_response(
                    data=serializers.PharmacyListSerializer(obj).data,
                    message='malumot qoshildi'
                )
            return self.failure_response(
                data=serializer.errors,
                message='malumot qoshilmadi'
            )
        except Exception as e:
            return self.error_response(
                data=str(e),
                message="xatolik"
            )
    
    @swagger_auto_schema(
        tags=['Admin Pharmacies']
    )
    @action(detail=True, methods=['patch'], url_path='update')
    def update_pharmacy(self, request, pk=None):
        try:
            pharmacy = Pharmacy.objects.filter(id=pk).first()
            if not pharmacy:
                return self.failure_response(
                    data={},
                    message="pharmacy topilmadi",
                    status_code=404
                )
            serializer = self.get_serializer(data=request.data, instance=pharmacy)
            if serializer.is_valid():
                obj = serializer.save()
                return self.success_response(
                    data=serializers.PharmacyListSerializer(obj).data,
                    message='malumot tahrirlandi'
                )
            return self.failure_response(
                data=serializer.errors,
                message='malumot tahrirlandi'
            )
        except Exception as e:
            return self.error_response(
                data=str(e),
                message="xatolik"
            )
    
    @swagger_auto_schema(
        tags=['Admin Pharmacies']
    )
    @action(detail=True, methods=['delete'], url_path='delete')
    def delete(self, request, pk=None):
        try:
            pharmacy = Pharmacy.objects.filter(id=pk).first()
            if not pharmacy:
                return self.failure_response(
                    data={},
                    message="pharmacy topilmadi",
                    status_code=404
                )
            pharmacy.delete()
            return self.success_response(
                data={},
                message='malumot ochirildi',
                status_code=204
            )
        except Exception as e:
            return self.error_response(
                data=str(e),
                message="xatolik"
            )