from django.shortcuts import render
from dal import autocomplete

from .models import (
    Pais, Estado, Municipio, Parroquia
)


class PaisAutocomplete(autocomplete.Select2QuerySetView):
    def get_queryset(self):
        # No se olvide de filtrar los resultados en función del visitante !
        if not self.request.user.is_authenticated():
            return Pais.objects.none()

        qs = Pais.objects.all()
        if self.q:
            qs = qs.filter(nombre__istartswith=self.q)

        return qs


class EstadoAutocomplete(autocomplete.Select2QuerySetView):
    def get_queryset(self):
        # No se olvide de filtrar los resultados en función del visitante !
        if not self.request.user.is_authenticated():
            return Estado.objects.none()

        qs = Estado.objects.all()

        if self.q:
            qs = qs.filter(nombre__istartswith=self.q)

        return qs


class MunicipioAutocomplete(autocomplete.Select2QuerySetView):
    def get_queryset(self):
        # No se olvide de filtrar los resultados en función del visitante !
        if not self.request.user.is_authenticated():
            return Municipio.objects.none()

        qs = Municipio.objects.all()

        if self.q:
            qs = qs.filter(nombre__istartswith=self.q)

        return qs


class ParroquiaAutocomplete(autocomplete.Select2QuerySetView):
    def get_queryset(self):
        # No se olvide de filtrar los resultados en función del visitante !
        if not self.request.user.is_authenticated():
            return Parroquia.objects.none()

        qs = Parroquia.objects.all()

        if self.q:
            qs = qs.filter(nombre__istartswith=self.q)

        return qs