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


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")

    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)
        )

    class Meta:
        model = Topup  # Assuming Topup is a subclass of Payment
        fields = [
            "amount",
            "paid",
            "user",
            "created_at",
        ]