40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
from django.db.models import Q
|
|
from django.db.models.functions import TruncDate
|
|
from apps.product.models import Quota
|
|
|
|
|
|
class QuotaSearch:
|
|
def __init__(self, query: str = None, start_date: str = None, end_date: str = None):
|
|
self.query = (query or '').strip()
|
|
self.start_date = start_date
|
|
self.end_date = end_date
|
|
|
|
def search(self):
|
|
""" multi term search & filter by date range """
|
|
|
|
queryset = QuotaDistribution.objects.all() # noqa
|
|
|
|
if self.start_date or self.end_date:
|
|
queryset = queryset.annotate(date_only=TruncDate('create_date'))
|
|
if self.start_date:
|
|
queryset = queryset.filter(date_only__gte=self.start_date)
|
|
if self.end_date:
|
|
queryset = queryset.filter(date_only__lte=self.end_date)
|
|
|
|
# convert string to list of words
|
|
if self.query:
|
|
keywords = [word.strip() for word in self.query.split(',') if word.strip()]
|
|
if keywords:
|
|
combined_q = Q()
|
|
for keyword in keywords:
|
|
combined_q |= Q(registerer_organization__name__icontains=keyword)
|
|
combined_q |= Q(quota_id__icontains=keyword)
|
|
combined_q |= Q(product__name__icontains=keyword)
|
|
combined_q |= Q(sale_type__icontains=keyword)
|
|
combined_q |= Q(sale_unit__unit__icontains=keyword)
|
|
combined_q |= Q(group__icontains=keyword)
|
|
|
|
queryset = queryset.filter(combined_q)
|
|
|
|
return queryset.distinct()
|