ob-dj-store 0.0.19__py3-none-any.whl → 0.0.23.2__py3-none-any.whl

ob_dj_store/apis/stores/filters.py

@@ -1,3 +1,5 @@
+from urllib.parse import unquote
+
 from django.contrib.contenttypes.models import ContentType
 from django.contrib.postgres.search import SearchVector
 from django.db.models import Prefetch, Q
@@ -17,6 +19,7 @@ from ob_dj_store.core.stores.models import (
     Product,
     ProductVariant,
     Store,
+    Tip,
 )
 from ob_dj_store.core.stores.models._inventory import Inventory
 from ob_dj_store.core.stores.models._wallet import Wallet
@@ -49,11 +52,7 @@ class StoreFilter(filters.FilterSet):
         return queryset.filter(address__country=value)
 
     def by_shipping_methods_names(self, queryset, name, value):
-        return queryset.filter(
-            shipping_methods__name__in=[
-                value,
-            ]
-        )
+        return queryset.filter(shipping_methods__name__in=[value,])
 
     def by_open_stores(self, queryset, name, value):
         if value:
@@ -91,6 +90,7 @@ class ProductFilter(filters.FilterSet):
     """Product filters"""
 
     category = filters.CharFilter(method="by_category")
+    q = filters.CharFilter(method="filter_search")
 
     class Meta:
         model = Product
@@ -103,6 +103,18 @@ class ProductFilter(filters.FilterSet):
     def by_category(self, queryset, name, value):
         return queryset.filter(category__name__iexact=value)
 
+    def filter_search(self, queryset, name, value):
+        language = self.request.META.get("HTTP_LANGUAGE", "").strip().upper()
+        if language == "AR":
+            value_ar = unquote(value)
+            query = Q(name_arabic__icontains=value_ar) | Q(
+                description_arabic__icontains=value_ar
+            )
+        else:
+            query = Q(name__icontains=value) | Q(description__icontains=value)
+        qs = queryset.filter(query).distinct()[:5]
+        return qs
+
 
 class VariantFilter(filters.FilterSet):
     """Variant filters"""
@@ -186,9 +198,7 @@ class CategoryFilter(filters.FilterSet):
 
     def by_type(self, queryset, name, value):
         return (
-            queryset.filter(
-                subcategories__products__type=value,
-            )
+            queryset.filter(subcategories__products__type=value,)
             .prefetch_related(
                 Prefetch(
                     "subcategories",
@@ -198,9 +208,7 @@ class CategoryFilter(filters.FilterSet):
                 ),
                 Prefetch(
                     "subcategories__products",
-                    queryset=Product.objects.filter(
-                        is_active=True,
-                    ).distinct(),
+                    queryset=Product.objects.filter(is_active=True,).distinct(),
                 ),
             )
             .distinct()
@@ -241,15 +249,13 @@ class FavoriteFilter(filters.FilterSet):
             product_variants__inventories__store__pk=value
         ).values_list("id", flat=True)
         content_type = ContentType.objects.get_for_model(Product)
-        return queryset.filter(
-            content_type=content_type,
-            object_id__in=products_ids,
-        )
+        return queryset.filter(content_type=content_type, object_id__in=products_ids,)
 
 
 class PaymentMethodFilter(filters.FilterSet):
     store = filters.CharFilter(method="by_store")
     is_digital = filters.BooleanFilter(method="by_digital")
+    country = filters.CharFilter(method="by_country")
 
     class Meta:
         models = PaymentMethod
@@ -264,13 +270,16 @@ class PaymentMethodFilter(filters.FilterSet):
             )
         return queryset
 
+    def by_country(self, queryset, name, value):
+        currency = get_currency_by_country(value)
+        if value:
+            queryset = queryset.filter(currency=currency)
+        return queryset
+
 
 class WalletFilter(filters.FilterSet):
     currency = filters.CharFilter(
-        method="by_currency",
-        validators=[
-            validate_currency,
-        ],
+        method="by_currency", validators=[validate_currency,],
     )
     country = filters.CharFilter(method="by_country")
 
@@ -314,3 +323,17 @@ class CountryPaymentMethodFilter(filters.FilterSet):
     def by_country(self, queryset, name, value):
         if value:
             return queryset.filter(country=value)
+
+
+class TipFilter(filters.FilterSet):
+
+    country = filters.CharFilter(method="by_country")
+
+    class Meta:
+        model = Tip
+        fields = []
+
+    def by_country(self, queryset, name, value):
+        if value:
+            queryset = queryset.filter(country=value)
+        return queryset