mirror of
https://github.com/i701/sarlink-portal-api.git
synced 2025-07-07 06:06:31 +00:00
i701
193ce850b4
All checks were successful
Build and Push Docker Images / Build and Push Docker Images (push) Successful in 3m45s
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
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",
|
|
]
|