import django_filters
from .models import Payment, Topup
from django.db.models import Q
from django.utils import timezone


class PaymentFilter(django_filters.FilterSet):
    amount = django_filters.RangeFilter(field_name="amount")
    number_of_months = django_filters.RangeFilter(field_name="number_of_months")
    paid = django_filters.BooleanFilter(field_name="paid")
    method = django_filters.ChoiceFilter(
        choices=Payment.PAYMENT_TYPES, lookup_expr="iexact"
    )
    mib_reference = django_filters.CharFilter(lookup_expr="icontains")
    paid_at = django_filters.DateFromToRangeFilter()
    created_at = django_filters.DateFromToRangeFilter()

    class Meta:
        model = Payment
        fields = "__all__"


class TopupFilter(django_filters.FilterSet):
    amount = django_filters.RangeFilter(field_name="amount")
    paid = django_filters.BooleanFilter(field_name="paid")
    user = django_filters.CharFilter(method="filter_user_search")
    created_at = django_filters.DateFromToRangeFilter(field_name="created_at")
    is_expired = django_filters.BooleanFilter(method="filter_is_expired")

    def filter_user_search(self, queryset, name, value):
        """
        Search across multiple user fields: first_name, last_name, id_card, mobile
        """
        return queryset.filter(
            Q(user__first_name__icontains=value)
            | Q(user__last_name__icontains=value)
            | Q(user__id_card__icontains=value)
            | Q(user__mobile__icontains=value)
        )

    def filter_is_expired(self, queryset, name, value):
        """
        Filter topups based on whether they are expired or not
        """
        now = timezone.now()
        queryset = queryset.filter(paid=False)
        if value:
            return queryset.filter(expires_at__isnull=False, expires_at__lt=now)
        else:
            return queryset.filter(Q(expires_at__isnull=True) | Q(expires_at__gte=now))

    class Meta:
        model = Topup
        fields = [
            "amount",
            "paid",
            "status",
            "user",
            "created_at",
            "is_expired",
        ]