from rest_framework import viewsets, parsers, generics
from .models import AgencyImages
from .serializers import AgencyImagesSerializer
from django.conf import settings


class AgencyImagesViewSet(generics.ListCreateAPIView):
    queryset = AgencyImages.objects.all()
    serializer_class = AgencyImagesSerializer
    parser_classes = [parsers.MultiPartParser, parsers.FormParser]

    def get_queryset(self):
        secret_key = self.request.META.get('HTTP_GTXKEY', None)
        agency_id = self.request.query_params.get('agency_id', None)
        title = self.request.query_params.get('title', None)
        document = self.request.query_params.get('document', None)
        if agency_id is not None and (secret_key is not None and secret_key == settings.GTX_KEY):
            if title is not None:
                sql_query = self.queryset.filter(agency_id=agency_id, title__icontains=title).order_by('created_at')
            elif document is not None:
                sql_query = self.queryset.filter(agency_id=agency_id, document=document).order_by('created_at')
            else:
                sql_query = self.queryset.filter(agency_id=agency_id).order_by('created_at')
            return sql_query
        return []