ob-dj-store 0.0.11.8__py3-none-any.whl → 0.0.11.10__py3-none-any.whl

ob_dj_store/apis/stores/filters.py

@@ -5,10 +5,12 @@ from django.utils.timezone import now
 from django_filters import rest_framework as filters
 from rest_framework.exceptions import ValidationError
 
+from config import settings as store_settings
 from ob_dj_store.core.stores.models import (
     Category,
     Favorite,
     Order,
+    PaymentMethod,
     Product,
     ProductVariant,
     Store,
@@ -128,6 +130,7 @@ class CategoryFilter(filters.FilterSet):
                         is_active=True,
                     ).distinct(),
                 ),
+                "subcategories__products__images",
             )
             .distinct()
         )
@@ -174,3 +177,25 @@ class FavoriteFilter(filters.FilterSet):
             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")
+
+    class Meta:
+        models = PaymentMethod
+
+    def by_store(self, queryset, name, value):
+        return queryset.filter(stores=value)
+
+    def by_digital(self, queryset, name, value):
+        if value:
+            return queryset.filter(
+                payment_provider__in=[
+                    store_settings.TAP_CREDIT_CARD,
+                    store_settings.TAP_KNET,
+                    store_settings.TAP_ALL,
+                ]
+            )
+        return queryset

ob_dj_store/apis/stores/rest/serializers/serializers.py

@@ -39,6 +39,7 @@ from ob_dj_store.core.stores.models import (
     ShippingMethod,
     Store,
     Tax,
+    Wallet,
 )
 from ob_dj_store.core.stores.models._inventory import Inventory
 from ob_dj_store.core.stores.utils import distance
@@ -288,7 +289,7 @@ class OrderSerializer(serializers.ModelSerializer):
         if payment_method:
             if payment_method.payment_provider == store_settings.WALLET:
                 try:
-                    wallet = user.wallets.get(country=attrs["store"].address.country)
+                    wallet = user.wallets.get(currency=attrs["store"].currency)
                 except ObjectDoesNotExist:
                     raise serializers.ValidationError(
                         {
@@ -603,7 +604,9 @@ class ProductSerializer(FavoriteMixin, serializers.ModelSerializer):
         return data
 
 
-class ProductListSerializer(ProductSerializer):
+class ProductListSerializer(serializers.ModelSerializer):
+    product_images = ProductMediaSerializer(many=True, source="images")
+
     class Meta:
         model = Product
         fields = (
@@ -974,7 +977,11 @@ class FavoriteSerializer(serializers.ModelSerializer):
             "extras",
             "object_id",
             "object_type",
+            "name",
         )
+        extra_kwargs = {
+            "name": {"required": True},
+        }
 
     def _lookup_validation(self, data):
         content_type = ContentType.objects.get_for_model(type(data["content_object"]))
@@ -1001,6 +1008,7 @@ class FavoriteSerializer(serializers.ModelSerializer):
 
     def validate(self, attrs):
         validated_data = super().validate(attrs)
+        name = validated_data["name"]
         object_type = validated_data["object_type"]
         if object_type not in store_settings.FAVORITE_TYPES and hasattr(
             sys.modules[__name__], object_type
@@ -1035,14 +1043,50 @@ class FavoriteSerializer(serializers.ModelSerializer):
         validated_data = {
             "content_object": object_instance,
             "extras": extras,
+            "name": name,
         }
         self._lookup_validation(validated_data)
         return validated_data
 
     def create(self, validated_data):
-        favorite = Favorite.add_favorite(
-            validated_data["content_object"],
-            self.context["request"].user,
-            validated_data["extras"],
-        )
+        try:
+            favorite = Favorite.add_favorite(
+                content_object=validated_data["content_object"],
+                user=self.context["request"].user,
+                name=validated_data["name"],
+                extras=validated_data["extras"],
+            )
+        except ValidationError as e:
+            raise serializers.ValidationError(detail=e.message_dict)
         return favorite
+
+
+class WalletSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Wallet
+        fields = ["id", "user", "balance", "name", "image", "image_thumbnail_medium"]
+        extra_kwargs = {
+            "user": {"read_only": True},
+            "image_thumbnail_medium": {"read_only": True},
+        }
+
+
+class WalletTopUpSerializer(serializers.Serializer):
+    amount = serializers.DecimalField(
+        max_digits=10, decimal_places=2, min_value=1, required=True
+    )
+    payment_method = serializers.PrimaryKeyRelatedField(
+        queryset=PaymentMethod.objects.filter(
+            payment_provider__in=[
+                store_settings.TAP_CREDIT_CARD,
+                store_settings.TAP_KNET,
+                store_settings.TAP_ALL,
+            ]
+        ),
+        required=True,
+    )
+
+    def top_up_wallet(self, wallet):
+        amount = self.validated_data["amount"]
+        payment_method = self.validated_data["payment_method"]
+        return wallet.top_up_wallet(amount=amount, payment_method=payment_method)

ob_dj_store/apis/stores/urls.py

@@ -16,6 +16,7 @@ from ob_dj_store.apis.stores.views import (
     TaxViewSet,
     TransactionsViewSet,
     VariantView,
+    WalletViewSet,
 )
 
 app_name = "stores"
@@ -28,9 +29,6 @@ stores_router.register(r"order", OrderView, basename="order")
 stores_router.register(r"product", ProductView, basename="product")
 stores_router.register(r"variant", VariantView, basename="variant")
 stores_router.register(r"inventory", InventoryView, basename="inventory")
-stores_router.register(
-    r"payment-method", PaymentMethodViewSet, basename="payment-method"
-)
 stores_router.register(
     r"shipping-method", ShippingMethodViewSet, basename="shipping-method"
 )
@@ -40,6 +38,8 @@ router.register(r"transaction", TransactionsViewSet, basename="transaction")
 router.register(r"cart", CartView, basename="cart")
 router.register(r"cart-item", CartItemView, basename="cart-item")
 router.register(r"favorite", FavoriteViewSet, basename="favorite")
+router.register(r"wallet", WalletViewSet, basename="wallet")
+router.register(r"payment-method", PaymentMethodViewSet, basename="payment-method")
 urlpatterns = [
     path(r"", include(router.urls)),
     path(r"", include(stores_router.urls)),

ob_dj_store/apis/stores/views.py

@@ -26,6 +26,7 @@ from ob_dj_store.apis.stores.filters import (
     FavoriteFilter,
     InventoryFilter,
     OrderFilter,
+    PaymentMethodFilter,
     ProductFilter,
     StoreFilter,
     VariantFilter,
@@ -47,6 +48,8 @@ from ob_dj_store.apis.stores.rest.serializers.serializers import (
     ShippingMethodSerializer,
     StoreSerializer,
     TaxSerializer,
+    WalletSerializer,
+    WalletTopUpSerializer,
 )
 from ob_dj_store.core.stores.models import (
     Cart,
@@ -62,6 +65,7 @@ from ob_dj_store.core.stores.models import (
     ShippingMethod,
     Store,
     Tax,
+    Wallet,
 )
 from ob_dj_store.core.stores.models._inventory import Inventory
 
@@ -188,29 +192,6 @@ class StoreView(
         serializer = self.get_serializer(page, many=True)
         return self.get_paginated_response(serializer.data)
 
-    @swagger_auto_schema(
-        operation_summary="Add or Remove Store from Favorites",
-        operation_description="""
-            Add or Remove Store from Favorites
-        """,
-        tags=[
-            "Store",
-        ],
-    )
-    @action(
-        detail=True,
-        methods=["GET"],
-        url_path="favorite",
-    )
-    def favorite(self, request, *args, **kwargs):
-        instance = self.get_object()
-        try:
-            Favorite.objects.favorite_for_user(instance, request.user).delete()
-        except Favorite.DoesNotExist:
-            Favorite.add_favorite(instance, request.user)
-        serializer = StoreSerializer(instance=instance, context={"request": request})
-        return Response(serializer.data)
-
     @swagger_auto_schema(
         operation_summary="Retrieve count of store's products",
         operation_description="""
@@ -674,6 +655,9 @@ class CategoryViewSet(
     filterset_class = CategoryFilter
     lookup_value_regex = "[0-9]+"
 
+    def get_queryset(self):
+        return super().get_queryset().prefetch_related("products__images")
+
     def get_object(self):
         store_id = self.request.query_params.get("store", None)
         instance_pk = self.kwargs["pk"]
@@ -699,6 +683,7 @@ class CategoryViewSet(
                     "products",
                     queryset=product_queryset.distinct(),
                 ),
+                "subcategories__products__images",
             ).get(pk=instance_pk)
         except Category.DoesNotExist:
             raise Http404("No Category matches the given query.")
@@ -800,14 +785,9 @@ class PaymentMethodViewSet(
     permission_classes = [
         permissions.IsAuthenticated,
     ]
-    queryset = PaymentMethod.objects.all()
-
-    def get_queryset(self):
-        try:
-            store = Store.objects.get(pk=self.kwargs["store_pk"])
-        except ObjectDoesNotExist:
-            raise ValidationError(_(f"Store does not Exist"))
-        return store.payment_methods.filter(is_active=True)
+    queryset = PaymentMethod.objects.filter(is_active=True)
+    filterset_class = PaymentMethodFilter
+    lookup_value_regex = "[0-9]+"
 
     @swagger_auto_schema(
         operation_summary="List Payment Methods",
@@ -949,3 +929,84 @@ class FavoriteViewSet(
     )
     def destroy(self, request, *args, **kwargs):
         return super().destroy(request, *args, **kwargs)
+
+
+class WalletViewSet(
+    mixins.ListModelMixin,
+    mixins.RetrieveModelMixin,
+    mixins.UpdateModelMixin,
+    viewsets.GenericViewSet,
+):
+    queryset = Wallet.objects.all()
+    serializer_class = WalletSerializer
+    permission_classes = [
+        permissions.IsAuthenticated,
+    ]
+
+    def get_queryset(self):
+        return Wallet.objects.filter(user=self.request.user)
+
+    @swagger_auto_schema(
+        operation_summary="Get User Wallet",
+        operation_description="""
+                Get User Wallet
+            """,
+        tags=[
+            "Wallet",
+        ],
+    )
+    def retrieve(
+        self, request: Request, *args: typing.Any, **kwargs: typing.Any
+    ) -> Response:
+        return super().retrieve(request=request, *args, **kwargs)
+
+    @swagger_auto_schema(
+        operation_summary="Update user wallet",
+        operation_description="""
+            Update a wallet
+        """,
+        tags=[
+            "Wallet",
+        ],
+    )
+    def partial_update(self, request: Request, *args: typing.Any, **kwargs: typing.Any):
+        return super().partial_update(request, *args, **kwargs)
+
+    @swagger_auto_schema(
+        operation_summary="List User Wallets",
+        operation_description="""
+                List User Wallets
+            """,
+        tags=[
+            "Wallet",
+        ],
+    )
+    def list(
+        self, request: Request, *args: typing.Any, **kwargs: typing.Any
+    ) -> Response:
+        return super().list(request=request, *args, **kwargs)
+
+    @swagger_auto_schema(
+        operation_summary="top up a wallet",
+        operation_description="""
+                top up a user wallet with tap payment
+            """,
+        tags=[
+            "Wallet",
+        ],
+    )
+    @action(
+        methods=["POST"],
+        detail=True,
+        url_path="top-up",
+        url_name="top-up",
+        serializer_class=WalletTopUpSerializer,
+    )
+    def top_up_wallet(
+        self, request: Request, *args: typing.Any, **kwargs: typing.Any
+    ) -> Response:
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        instance = self.get_object()
+        payment_url = serializer.top_up_wallet(instance)
+        return Response({"payment_url": payment_url}, status=status.HTTP_200_OK)

ob_dj_store/core/stores/admin.py

@@ -104,12 +104,6 @@ class CategoryAdmin(admin.ModelAdmin):
         "is_active",
     ]
 
-    def save_model(self, request, obj, form, change) -> None:
-        from ob_dj_store.core.stores.receivers import create_media_thumbnails
-
-        create_media_thumbnails(None, obj, None)
-        return super().save_model(request, obj, form, change)
-
 
 class ProductVariantAdmin(admin.ModelAdmin):
     inlines = [
@@ -122,6 +116,9 @@ class ProductVariantAdmin(admin.ModelAdmin):
     ]
     search_fields = ["name", "product__name", "sku"]
 
+    def get_queryset(self, request):
+        return super().get_queryset(request).prefetch_related("inventories")
+
 
 class ProductAdmin(admin.ModelAdmin):
     list_display = ["id", "name", "category", "type", "is_active"]
@@ -249,6 +246,9 @@ class InventoryAdmin(admin.ModelAdmin):
         "is_uncountable",
     ]
 
+    def get_queryset(self, request):
+        return super().get_queryset(request).prefetch_related("store")
+
 
 class TaxAdmin(admin.ModelAdmin):
     list_display = [

ob_dj_store/core/stores/admin_inlines.py

@@ -22,10 +22,10 @@ class InventoryInlineAdmin(admin.TabularInline):
     extra = 1
 
     def get_queryset(self, request):
-        qs = super().get_queryset(request)
-        return qs.select_related(
-            "store",
-            "variant",
+        return (
+            super()
+            .get_queryset(request)
+            .select_related("variant", "store", "variant__product")
         )
 
 

ob_dj_store/core/stores/gateway/tap/models.py

@@ -93,7 +93,7 @@ class TapPayment(models.Model):
 
     @property
     def amount(self):
-        return self.payment.amount
+        return self.payment.total_payment
 
     def callback_update(self, tap_payload):
         if self.status == self.Status.INITIATED:

ob_dj_store/core/stores/managers.py

@@ -6,7 +6,6 @@ from django.db import models
 from django.utils.translation import gettext_lazy as _
 
 from config import settings
-from ob_dj_store.core.stores.utils import get_country_by_currency
 
 
 class ActiveMixin:
@@ -77,8 +76,7 @@ class PaymentManager(models.Manager):
             return instance
         elif gateway == settings.WALLET:
             try:
-                country = get_country_by_currency(currency)
-                wallet = kwargs["user"].wallets.get(country=country)
+                wallet = kwargs["user"].wallets.get(currency=currency)
             except ObjectDoesNotExist:
                 raise ValidationError({"wallet": _("Wallet Not Found")})
             WalletTransaction.objects.create(

ob_dj_store/core/stores/migrations/0062_auto_20230226_2005.py

@@ -0,0 +1,43 @@
+# Generated by Django 3.2.8 on 2023-02-26 17:05
+
+from django.conf import settings
+from django.db import migrations, models
+
+import ob_dj_store.core.stores.utils
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ("stores", "0061_auto_20230223_1435"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="store",
+            name="currency",
+            field=models.CharField(
+                default="KWD",
+                max_length=3,
+                validators=[ob_dj_store.core.stores.utils.validate_currency],
+            ),
+        ),
+        migrations.AddField(
+            model_name="wallet",
+            name="currency",
+            field=models.CharField(
+                default="KWD",
+                max_length=3,
+                validators=[ob_dj_store.core.stores.utils.validate_currency],
+            ),
+        ),
+        migrations.AlterUniqueTogether(
+            name="wallet",
+            unique_together={("user", "currency")},
+        ),
+        migrations.RemoveField(
+            model_name="wallet",
+            name="country",
+        ),
+    ]

ob_dj_store/core/stores/migrations/0063_alter_store_payment_methods.py

@@ -0,0 +1,23 @@
+# Generated by Django 3.2.8 on 2023-02-27 11:15
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("stores", "0062_auto_20230226_2005"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="store",
+            name="payment_methods",
+            field=models.ManyToManyField(
+                blank=True,
+                help_text="Payment methods within the store",
+                related_name="stores",
+                to="stores.PaymentMethod",
+            ),
+        ),
+    ]

ob_dj_store/core/stores/migrations/0064_auto_20230228_1814.py

@@ -0,0 +1,24 @@
+# Generated by Django 3.2.8 on 2023-02-28 15:14
+
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ("stores", "0063_alter_store_payment_methods"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="favorite",
+            name="name",
+            field=models.CharField(blank=True, max_length=200, null=True),
+        ),
+        migrations.AlterUniqueTogether(
+            name="favorite",
+            unique_together={("name", "user")},
+        ),
+    ]

ob_dj_store/core/stores/migrations/0065_auto_20230228_1932.py

@@ -0,0 +1,34 @@
+# Generated by Django 3.2.8 on 2023-02-28 16:32
+
+from django.db import migrations, models
+
+import ob_dj_store.utils.helpers
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("stores", "0064_auto_20230228_1814"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="wallet",
+            name="image",
+            field=models.ImageField(
+                blank=True,
+                null=True,
+                upload_to=ob_dj_store.utils.helpers.wallet_media_upload_to,
+            ),
+        ),
+        migrations.AddField(
+            model_name="wallet",
+            name="image_thumbnail_medium",
+            field=models.ImageField(blank=True, null=True, upload_to="wallets/"),
+        ),
+        migrations.AddField(
+            model_name="wallet",
+            name="name",
+            field=models.CharField(blank=True, max_length=200, null=True),
+        ),
+    ]

ob_dj_store/core/stores/models/_favorite.py

@@ -16,6 +16,7 @@ class Favorite(DjangoModelCleanMixin, models.Model):
     user = models.ForeignKey(
         get_user_model(), related_name="favorites", on_delete=models.CASCADE
     )
+    name = models.CharField(max_length=200, null=True, blank=True)
     content_type = models.ForeignKey(ContentType, on_delete=models.CASCADE)
     content_object = GenericForeignKey("content_type", "object_id")
     object_id = models.PositiveIntegerField()
@@ -26,18 +27,20 @@ class Favorite(DjangoModelCleanMixin, models.Model):
     class Meta:
         verbose_name = _("favorite")
         verbose_name_plural = _("favorites")
+        unique_together = (("name", "user"),)
 
     def __str__(self):
         return f"{self.user} favorites {self.content_object}"
 
     @classmethod
-    def add_favorite(cls, content_object, user, extras=[]):
+    def add_favorite(cls, content_object, user, name, extras=[]):
         content_type = ContentType.objects.get_for_model(type(content_object))
         favorite = Favorite(
             user=user,
             content_type=content_type,
             object_id=content_object.pk,
             content_object=content_object,
+            name=name,
         )
         favorite.save()
         for extra in extras:

ob_dj_store/core/stores/models/_order.py

@@ -119,6 +119,8 @@ class Order(DjangoModelCleanMixin, models.Model):
 
     @property
     def total_amount(self):
+        if self.type_of_order == Order.OrderType.WALLET.value:
+            return Decimal(self.extra_infos["amount"])
         amount = Decimal(
             sum(map(lambda item: Decimal(item.total_amount) or 0, self.items.all()))
         )

ob_dj_store/core/stores/models/_payment.py

@@ -110,17 +110,36 @@ class Payment(models.Model):
         return f"Payment (PK={self.pk})"
 
     def mark_paid(self):
+        """
+        mark payment and orders as paid then perform post payment actions
+        if wallet fill up we add wallet transaction to the user
+        if physical order we decrease the quantity from inventory and delete items from cart
+        """
+        from ob_dj_store.core.stores.models._wallet import WalletTransaction
+
         self.status = self.PaymentStatus.SUCCESS
-        orders = list(self.orders.all())
-        cart = self.user.cart
-        for order in orders:
+        orders = self.orders.all()
+        if orders[0].type_of_order == Order.OrderType.WALLET.value:
+            order = orders[0]
+            currency = order.extra_infos["currency"]
+            wallet = self.user.wallets.get(currency=currency)
+            WalletTransaction.objects.create(
+                wallet=wallet,
+                amount=self.amount,  # TODO: cunfused about how we hundle if he want to fill up other currency
+                type=WalletTransaction.TYPE.CREDIT,
+            )
             order.status = Order.OrderStatus.PAID
-            for item in order.items.all():
-                if item.inventory:
-                    item.inventory.decrease(item.quantity)
             order.save()
-            items = order.store.store_items.filter(cart=cart)
-            items.delete()
+        else:
+            cart = self.user.cart
+            for order in orders:
+                order.status = Order.OrderStatus.PAID
+                for item in order.items.all():
+                    if item.inventory:
+                        item.inventory.decrease(item.quantity)
+                order.save()
+                items = order.store.store_items.filter(cart=cart)
+                items.delete()
         self.payment_post_at = timezone.now()
         self.save()
 
@@ -130,14 +149,15 @@ class Payment(models.Model):
 
     @property
     def total_payment(self):
+        orders = self.orders.all()
         sum_orders = Decimal(
-            sum(map(lambda order: Decimal(order.total_amount) or 0, self.orders.all()))
+            sum(map(lambda order: Decimal(order.total_amount) or 0, orders))
         )
         if self.payment_tax.rate == Tax.Rates.PERCENTAGE:
             perc = Decimal(sum_orders * self.payment_tax.value / 100)
-            return sum_orders + perc
+            return round(sum_orders + perc, 3)
 
-        return sum_orders + Decimal(self.payment_tax.value)
+        return round(sum_orders + self.payment_tax.value, 3)
 
     @property
     def type_of_order(self):

ob_dj_store/core/stores/models/_product.py

@@ -319,8 +319,6 @@ class ProductMedia(DjangoModelCleanMixin, models.Model):
         verbose_name = _("product media")
         verbose_name_plural = _("Product medias")
 
-    def save(self, **kwargs):
-        from ob_dj_store.core.stores.receivers import create_media_thumbnails
-
-        create_media_thumbnails(None, self, None)
-        return super().save(**kwargs)
+    @property
+    def name(self):
+        return f"{self.product.name}_{self.order_value}"

ob_dj_store/core/stores/models/_store.py

@@ -12,7 +12,7 @@ from ob_dj_store.core.stores.managers import (
     ShippingMethodManager,
     StoreManager,
 )
-from ob_dj_store.core.stores.utils import get_currency_by_country
+from ob_dj_store.core.stores.utils import validate_currency
 from ob_dj_store.utils.helpers import store_media_upload_to
 from ob_dj_store.utils.model import DjangoModelCleanMixin
 
@@ -114,7 +114,7 @@ class Store(DjangoModelCleanMixin, models.Model):
     )
     payment_methods = models.ManyToManyField(
         PaymentMethod,
-        related_name="payment_methods",
+        related_name="stores",
         blank=True,
         help_text=_("Payment methods within the store"),
     )
@@ -137,6 +137,13 @@ class Store(DjangoModelCleanMixin, models.Model):
         help_text=_("This is the min price to get a free delivery"),
     )
     image = models.ImageField(upload_to=store_media_upload_to, null=True, blank=True)
+    currency = models.CharField(
+        max_length=3,
+        default="KWD",
+        validators=[
+            validate_currency,
+        ],
+    )
 
     created_at = models.DateTimeField(auto_now_add=True)
     updated_at = models.DateTimeField(auto_now=True)
@@ -150,12 +157,6 @@ class Store(DjangoModelCleanMixin, models.Model):
     def __str__(self) -> typing.Text:
         return f"Store {self.name} (PK={self.pk})"
 
-    @property
-    def currency(self):
-        if self.address:
-            return get_currency_by_country(self.address.country.name)
-        return None
-
     @property
     def current_opening_hours(self):
         return next(

ob_dj_store/core/stores/models/_wallet.py

@@ -6,10 +6,11 @@ from django.conf import settings
 from django.db import models
 from django.db.models.functions import Coalesce
 from django.utils.translation import gettext_lazy as _
-from django_countries.fields import CountryField
 
 from ob_dj_store.core.stores.managers import WalletTransactionManager
-from ob_dj_store.core.stores.utils import get_currency_by_country
+from ob_dj_store.core.stores.models._store import PaymentMethod
+from ob_dj_store.core.stores.utils import validate_currency
+from ob_dj_store.utils.helpers import wallet_media_upload_to
 
 logger = logging.getLogger(__name__)
 
@@ -20,7 +21,21 @@ class Wallet(models.Model):
         on_delete=models.CASCADE,
         related_name="wallets",
     )
-    country = CountryField(_("the wallet's country"), null=True, blank=True)
+    name = models.CharField(max_length=200, null=True, blank=True)
+    image = models.ImageField(upload_to=wallet_media_upload_to, null=True, blank=True)
+    image_thumbnail_medium = models.ImageField(
+        upload_to="wallets/", null=True, blank=True
+    )
+    currency = models.CharField(
+        max_length=3,
+        default="KWD",
+        validators=[
+            validate_currency,
+        ],
+    )
+
+    class Meta:
+        unique_together = ("user", "currency")
 
     def __str__(self) -> typing.Text:
         return f"Wallet(PK={self.pk})"
@@ -48,11 +63,29 @@ class Wallet(models.Model):
         )
         return query["balance"]
 
-    @property
-    def currency(self):
-        if self.country:
-            return get_currency_by_country(self.country.name)
-        return None
+    def top_up_wallet(self, amount: Decimal, payment_method: PaymentMethod):
+        from ob_dj_store.core.stores.models import Order, Payment
+
+        user = self.user
+        order = Order.objects.create(
+            customer=user,
+            payment_method=payment_method,
+            extra_infos={
+                "is_wallet_fill_up": True,
+                "amount": str(amount),
+                "currency": self.currency,
+            },
+        )
+        payment = Payment.objects.create(
+            orders=[
+                order,
+            ],
+            user=user,
+            currency=self.currency,
+            method=payment_method,
+            amount=amount,
+        )
+        return payment.payment_url
 
 
 class WalletTransaction(models.Model):

ob_dj_store/core/stores/receivers.py

@@ -1,9 +1,16 @@
 from django.conf import settings
-from django.db.models.signals import post_save
+from django.db.models.signals import post_save, pre_save
 from django.dispatch import receiver
 
 from config import settings as store_settings
-from ob_dj_store.core.stores.models import Cart, Category, Order, OrderHistory, Wallet
+from ob_dj_store.core.stores.models import (
+    Cart,
+    Category,
+    Order,
+    OrderHistory,
+    ProductMedia,
+    Wallet,
+)
 from ob_dj_store.utils.utils import resize_image
 
 
@@ -17,8 +24,8 @@ def create_customer_cart_and_wallet_handler(sender, instance, created, **kwargs)
         return
     cart = Cart(customer=instance)
     cart.save()
-    country = getattr(instance, "country", None)
-    Wallet.objects.create(user=instance, country=country)
+    for currency in store_settings.WALLET_CURRENCIES:
+        instance.wallets.create(currency=currency)
 
 
 # add receiver to ProductVariant to create inventory
@@ -37,19 +44,66 @@ def create_order_history_handler(sender, instance, created, **kwargs):
 
 
 @receiver(
-    post_save,
+    pre_save,
     sender=Category,
     dispatch_uid="create_category_thumbnails",
 )
-def create_media_thumbnails(sender, instance, created, **kwargs):
+def create_category_thumbnails(sender, instance, **kwargs):
+    medium_dim = getattr(store_settings, "THUMBNAIL_MEDIUM_DIMENSIONS", None)
+    small_dim = getattr(store_settings, "THUMBNAIL_SMALL_DIMENSIONS", None)
+    if instance.image:
+        if medium_dim:
+            instance.image_thumbnail_medium = resize_image(
+                instance.image,
+                dim=medium_dim,
+                size_name="medium",
+                image_name=instance.name,
+            )
+        if small_dim:
+            instance.image_thumbnail_small = resize_image(
+                instance.image,
+                dim=small_dim,
+                size_name="small",
+                image_name=instance.name,
+            )
+
+
+@receiver(
+    pre_save,
+    sender=ProductMedia,
+    dispatch_uid="create_product_media_thumbnails",
+)
+def create_product_media_thumbnails(sender, instance, **kwargs):
     medium_dim = getattr(store_settings, "THUMBNAIL_MEDIUM_DIMENSIONS", None)
     small_dim = getattr(store_settings, "THUMBNAIL_SMALL_DIMENSIONS", None)
     if instance.image:
         if medium_dim:
             instance.image_thumbnail_medium = resize_image(
-                instance.image, dim=medium_dim, size_name="medium"
+                instance.image,
+                dim=medium_dim,
+                size_name="medium",
+                image_name=instance.name,
             )
         if small_dim:
             instance.image_thumbnail_small = resize_image(
-                instance.image, dim=small_dim, size_name="small"
+                instance.image,
+                dim=small_dim,
+                size_name="small",
+                image_name=instance.name,
             )
+
+
+@receiver(
+    pre_save,
+    sender=Wallet,
+    dispatch_uid="create_wallet_thumbnails",
+)
+def create_wallet_thumbnails(sender, instance, **kwargs):
+    medium_dim = getattr(store_settings, "THUMBNAIL_MEDIUM_DIMENSIONS", None)
+    if instance.image and medium_dim:
+        instance.image_thumbnail_medium = resize_image(
+            instance.image,
+            dim=medium_dim,
+            size_name="medium",
+            image_name=f"{instance.currency}_{instance.user.username}",
+        )

ob_dj_store/core/stores/utils.py

@@ -1,6 +1,8 @@
 import math
 
 import pycountry
+from django.core.exceptions import ValidationError
+from django.utils.translation import gettext_lazy as _
 
 
 def get_data_dict(instance):
@@ -53,13 +55,19 @@ def distance(origin, destination):
 
 def get_currency_by_country(country):
     country = pycountry.countries.get(name=country)
-    if not country:
-        return None
-    return pycountry.currencies.get(numeric=country.numeric).alpha_3
+    currency = pycountry.currencies.get(numeric=country.numeric)
+    return currency.alpha_3
 
 
 def get_country_by_currency(currency):
     currency = pycountry.currencies.get(alpha_3=currency)
-    if not currency:
-        return None
-    return pycountry.countries.get(numeric=currency.numeric).alpha_2
+    country = pycountry.countries.get(numeric=currency.numeric)
+    return country.alpha_2
+
+
+def validate_currency(value):
+    if not pycountry.currencies.get(alpha_3=value):
+        raise ValidationError(
+            _("%(value)s is not a currency"),
+            params={"value": value},
+        )

ob_dj_store/utils/helpers.py

@@ -19,7 +19,7 @@ def import_class_from_string(serializer_class_name):
 
 def product_media_upload_to(instance, filename):
     ext = filename.split(".")[-1]
-    return f"product_media/{instance.product.id}_{instance.order_value}_{int(now().timestamp())}.{ext}"
+    return f"product_media/{instance.product.name}_{instance.order_value}_{int(now().timestamp())}.{ext}"
 
 
 def category_media_upload_to(instance, filename):
@@ -32,3 +32,9 @@ def store_media_upload_to(instance, filename):
     ext = filename.split(".")[-1]
     if instance:
         return f"store_media/{instance.name}_{int(now().timestamp())}.{ext}"
+
+
+def wallet_media_upload_to(instance, filename):
+    ext = filename.split(".")[-1]
+    if instance:
+        return f"wallets/{instance.currency}_{instance.user.username}_{int(now().timestamp())}.{ext}"

ob_dj_store/utils/utils.py

@@ -3,13 +3,14 @@ import os
 from io import BytesIO
 
 from django.core.files.uploadedfile import InMemoryUploadedFile
+from django.utils.timezone import now
 from PIL import Image, UnidentifiedImageError
 from pilkit.processors import ResizeToFill
 
 logger = logging.getLogger(__name__)
 
 
-def resize_image(image, dim: dict, size_name: str):
+def resize_image(image, dim: dict, size_name: str, image_name: str):
     try:
         img = Image.open(image)
     except UnidentifiedImageError as e:
@@ -34,7 +35,7 @@ def resize_image(image, dim: dict, size_name: str):
     new_image = InMemoryUploadedFile(
         output,
         "ImageField",
-        f"{filename}_{size_name}.{img_format.lower()}",
+        f"{image_name}_{int(now().timestamp())}_{size_name}.{img_format.lower()}",
         content_type,
         output.__sizeof__(),
         None,

{ob_dj_store-0.0.11.8.dist-info → ob_dj_store-0.0.11.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-dj-store
-Version: 0.0.11.8
+Version: 0.0.11.10
 Summary: OBytes django application for managing ecommerce stores.
 Home-page: https://www.obytes.com/
 Author: OBytes

{ob_dj_store-0.0.11.8.dist-info → ob_dj_store-0.0.11.10.dist-info}/RECORD

@@ -1,28 +1,28 @@
 ob_dj_store/apis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ob_dj_store/apis/stores/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ob_dj_store/apis/stores/filters.py,sha256=Kx4szVD8ethRcOHyUE4k3qkWNq1jercGw1dgieug7Cc,4991
-ob_dj_store/apis/stores/urls.py,sha256=wwItq02d8xXiDaQs2p2CrhgOTW2jrMD75EFbMTZuspM,1526
-ob_dj_store/apis/stores/views.py,sha256=2v6oood8PDsoQrNb-qAEGtmEzp2yBsdBORiMQn6G4gI,28202
-ob_dj_store/apis/stores/rest/serializers/serializers.py,sha256=qUh0uFJ9YJQv8QspAWZxviQ1s_BcAXQa_vIkCE_NHmU,34908
+ob_dj_store/apis/stores/filters.py,sha256=s4US_TysnIgdT8pue_1jlG3V33WhYmCymXK3DuC0rqY,5746
+ob_dj_store/apis/stores/urls.py,sha256=cPForgFpOgOGCUVAk6DZcZ7qOooMf1zpDIr1BA0L_8A,1593
+ob_dj_store/apis/stores/views.py,sha256=zoEgQg3qKVlhhHPj96_UoNE-B8Pf8NK4AuUEmTwOhgg,29833
+ob_dj_store/apis/stores/rest/serializers/serializers.py,sha256=43c_0NyKVZDfmI32f2DR3m6i0G2y3G_yxZEuJBvdAL4,36402
 ob_dj_store/apis/tap/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ob_dj_store/apis/tap/serializers.py,sha256=KPrBK4h2-fWvEVf6vOj2ww5-USV9WqpyYicIqoHIiXI,1065
 ob_dj_store/apis/tap/urls.py,sha256=bnOTv6an11kxpo_FdqlhsizlGPLVpNxBjCyKcf3_C9M,367
 ob_dj_store/apis/tap/views.py,sha256=VnVquybTHlJquxsC0RNTy20dtLXalchO0SlGjSDaBng,2666
 ob_dj_store/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ob_dj_store/core/stores/__init__.py,sha256=-izNGrxNn_nn3IQXd5pkuES9lSF-AHYb14yhNPozYCI,65
-ob_dj_store/core/stores/admin.py,sha256=ZgVudDVomfKNR_BEhUzk5cz2Exx8G0K5H-PsPEnA7Mk,7383
-ob_dj_store/core/stores/admin_inlines.py,sha256=8pnJRzU3Zc0Il-BHm92hj-JEIdFZFj-Sdf9BAa5ne9g,1538
+ob_dj_store/core/stores/admin.py,sha256=4um-O_30xpnfcZHe_sNOyvSTnqLSbeExA2E1QppLEZQ,7354
+ob_dj_store/core/stores/admin_inlines.py,sha256=SLj8mMa-Sc3oP693R0W9c3Ocyk7eb34IyXbC4E6LL8M,1557
 ob_dj_store/core/stores/apps.py,sha256=i6D2lcqlluP6FwJSBpekf7BQVzQURwSy_Y97SgqWTkM,434
-ob_dj_store/core/stores/managers.py,sha256=qAAas3VTmppPyLVrTdIEW_ilPujcvqq4EtAkmmH_Kyg,6483
-ob_dj_store/core/stores/receivers.py,sha256=DR5-qDsAXLitSkc4_4Ade7ziGCeHUR53e8jocmOIWsk,1685
+ob_dj_store/core/stores/managers.py,sha256=cDr8TtPxayLwTFNgFs9I87ZXh7ssqM-yAmvEcN9a5LU,6359
+ob_dj_store/core/stores/receivers.py,sha256=5AYMW8ul5htJPp4gBwRAH1VyT5ugZvDMJmed2sostBM,3194
 ob_dj_store/core/stores/settings_validation.py,sha256=s9BPEdyCL2aslZwZGLAWEvJBzXlUo0p2TbC1_0E_SDo,327
-ob_dj_store/core/stores/utils.py,sha256=P_aSRj8ap9z6m1ZsX8iNEpo57yWbVTkg93yT5oQwXgo,1571
+ob_dj_store/core/stores/utils.py,sha256=_FwZEIwKdfj3CuYHCz3wKqq5TBb8xak7UiiCB1oggKc,1850
 ob_dj_store/core/stores/gateway/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ob_dj_store/core/stores/gateway/tap/__init__.py,sha256=5Z6azpb6tmr1nRvKwQWzlYw9ruvw-9ZMBWRqEngDKTM,40
 ob_dj_store/core/stores/gateway/tap/admin.py,sha256=Jzt50VacsSaxJb-ly4sJ0F4NOujixlS4_3DgkdU0JSk,384
 ob_dj_store/core/stores/gateway/tap/apps.py,sha256=kWwPjuAJeEmEasVDzUbvRsGaQWL-aYe4JDHNLvCVXPs,212
 ob_dj_store/core/stores/gateway/tap/managers.py,sha256=nJkKPzd_bgTttmZERiWWHiQW0WWcfSHUxNkOKmyrjDY,830
-ob_dj_store/core/stores/gateway/tap/models.py,sha256=BGxHY_-X4kLFLePuHGezvTAyBot3iQndigXtyW0RChg,3449
+ob_dj_store/core/stores/gateway/tap/models.py,sha256=11SYWK-I7NyrJUQAJeTp3jzO94jwWNm_y5QZXd4UFkA,3456
 ob_dj_store/core/stores/gateway/tap/utils.py,sha256=DB-UVxQT3Sdb17g5PF5a9Ghy6GgKADxcJWt1MxPmlt4,2493
 ob_dj_store/core/stores/gateway/tap/migrations/0001_initial.py,sha256=_303R1R2tVVdPSMtwpuLUxvWbxQ2BML1Gsd-VhfnZEM,4808
 ob_dj_store/core/stores/gateway/tap/migrations/0002_auto_20220815_1610.py,sha256=gRsekTbqUH4h5yg5d1zWuai7Wiv6-rfsmjnXundgin8,674
@@ -90,23 +90,27 @@ ob_dj_store/core/stores/migrations/0058_attributechoice_is_default.py,sha256=nv_
 ob_dj_store/core/stores/migrations/0059_auto_20230217_2006.py,sha256=ZaTZRC10L9uzMuya0rXHcsvJzsfF6H9-kcbzrrKVZMQ,1304
 ob_dj_store/core/stores/migrations/0060_alter_orderitem_product_variant.py,sha256=wyGWekRoR3XAGFfNScNV1jvqMz_HgWACCeiodUa_fGA,607
 ob_dj_store/core/stores/migrations/0061_auto_20230223_1435.py,sha256=Rqi0XMNw21mVlSzmWo132xaJfamUeyZ1l0_wdr6ipg0,796
+ob_dj_store/core/stores/migrations/0062_auto_20230226_2005.py,sha256=ApzGG5teYcLd4horeCzxIWnVPTeRoMNlJProUhYHacY,1187
+ob_dj_store/core/stores/migrations/0063_alter_store_payment_methods.py,sha256=QKIrNVtjttvYn4IO1GpGkA2gX4cyqjHNbFQ7q7cQqno,571
+ob_dj_store/core/stores/migrations/0064_auto_20230228_1814.py,sha256=6fgusQrMykN_tlNBxiJqLNvrHBA0yI32JK0kQ4VfNpQ,644
+ob_dj_store/core/stores/migrations/0065_auto_20230228_1932.py,sha256=r3-ThPhh_Lf4J3tYbCLiRbLSui5VSqwaVCtzvEZes_8,919
 ob_dj_store/core/stores/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ob_dj_store/core/stores/models/__init__.py,sha256=1lEOqmvQlQJZj9nnObTAQqgnPP1yVGvM8H9yjglZ0VI,1568
 ob_dj_store/core/stores/models/_address.py,sha256=qS5TQ9Z12zx_4CrrHvG8PoYVkdiOq_MtbKR14WKh3Hw,1661
 ob_dj_store/core/stores/models/_cart.py,sha256=qgjg5MXTMRIZRJJDx5BQ1pa9__ylfrnYMMRGbB2R53c,4555
-ob_dj_store/core/stores/models/_favorite.py,sha256=k5WlwSiYsLFICh4cvfkuo63yiVmyU8HbZ9X60_c26WI,2419
+ob_dj_store/core/stores/models/_favorite.py,sha256=bpXlFBNK8jsQkZG1RiDAbNTNSoVP0NOUUxZI8c46QzI,2561
 ob_dj_store/core/stores/models/_feedback.py,sha256=eCUVgprNK5hSRKOS4M_pdR7QH2-rqhoYevlpykhCOLg,1472
 ob_dj_store/core/stores/models/_inventory.py,sha256=vAXSpCyUdYDIWgUEUUObQfhAcCcQO6j6zATrHf5dPuQ,3928
-ob_dj_store/core/stores/models/_order.py,sha256=B1MpTal3nC3tgNzy8cDsjB2usMQMoN_WTvY3VUniwXM,7900
-ob_dj_store/core/stores/models/_payment.py,sha256=Rxo3vlpCzyqQjtCWjlQctgx15tuWf2e2zurURmBqGa0,5189
-ob_dj_store/core/stores/models/_product.py,sha256=R8ZW_12iofS8LrlVQ2zI-hDQE-gRsxXTTsQ_7gVIjQA,11034
-ob_dj_store/core/stores/models/_store.py,sha256=ysu_UQuJODj8JR9mnYmQPas5hi3kkUm9IRTBumNEX9s,7356
-ob_dj_store/core/stores/models/_wallet.py,sha256=cnLC7NgHk2dDwgwV-vLkiVN0cBP0LjsPwDVVK2wp0ro,2640
+ob_dj_store/core/stores/models/_order.py,sha256=qtdUpG30dZeDGeh0Z8JhzyyqQIuxTdLfyzd0-xaoo3o,8018
+ob_dj_store/core/stores/models/_payment.py,sha256=dCOXOYkgxOGJqp0nxEvzHmdZxR_srFUIZAhrB--SP1U,6143
+ob_dj_store/core/stores/models/_product.py,sha256=yGVp3vl5rCReG4c8ogWy2NUcsTJeqCxP0tm6lQyYx7Y,10928
+ob_dj_store/core/stores/models/_store.py,sha256=g7QXKbZI886Pdw6smuheVep7Vh1ke42PEhJhnxW4byA,7334
+ob_dj_store/core/stores/models/_wallet.py,sha256=4LUyWPlgRedthWij-Nkn4diYvWHU755IOBLgpjZOyQ8,3679
 ob_dj_store/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ob_dj_store/utils/helpers.py,sha256=5zR395i-hTxEvgCscQEm2ueLB1ZTRpeY5faGPJI673M,1019
+ob_dj_store/utils/helpers.py,sha256=02igVH2DSDTZYa6kFSTmBnJeXfTdgjRCRSXQ7mvmCGo,1224
 ob_dj_store/utils/model.py,sha256=DV7hOhTaZL3gh9sptts2jTUFlTArKG3i7oPioq9HLFE,303
-ob_dj_store/utils/utils.py,sha256=nHIa7oIIuZee3Vlf3UDcUU_myR3-UuFZ_zgo822Cy5I,1200
-ob_dj_store-0.0.11.8.dist-info/METADATA,sha256=DwF5WSq53xrMpj10Ir0-V14xZionUWPQhQartef6rIA,2827
-ob_dj_store-0.0.11.8.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-ob_dj_store-0.0.11.8.dist-info/top_level.txt,sha256=CZG3G0ptTkzGnc0dFYN-ZD7YKdJBmm47bsmGwofD_lk,12
-ob_dj_store-0.0.11.8.dist-info/RECORD,,
+ob_dj_store/utils/utils.py,sha256=euWeNI39P48jGNwKoBCC5AmwQUmgDeb92t0fQtPm-5g,1282
+ob_dj_store-0.0.11.10.dist-info/METADATA,sha256=pjF7LcR9ltkOqvsxXXp5ZGjaJRR47kGrO_SyxVtGrZ4,2828
+ob_dj_store-0.0.11.10.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+ob_dj_store-0.0.11.10.dist-info/top_level.txt,sha256=CZG3G0ptTkzGnc0dFYN-ZD7YKdJBmm47bsmGwofD_lk,12
+ob_dj_store-0.0.11.10.dist-info/RECORD,,