from rest_framework import viewsets, permissions
from .models import FeaturedService
from .serializers import FeaturedServiceSerializer


class FeaturedServiceViewSet(viewsets.ModelViewSet):
    queryset = FeaturedService.objects.filter(publish_status='published')
    serializer_class = FeaturedServiceSerializer
    permission_classes = [permissions.AllowAny]
    
    def get_queryset(self):
        queryset = FeaturedService.objects.all()
        region_id = self.request.query_params.get('region')
        category = self.request.query_params.get('category')
        
        if region_id:
            queryset = queryset.filter(region_id=region_id)
        if category:
            queryset = queryset.filter(category=category)
        
        return queryset
    
    def perform_create(self, serializer):
        service = serializer.save(submitter=self.request.user)
        service.submit_for_moderator_review()