from decimal import Decimal from django.db import models from django.http import HttpResponse from django.shortcuts import get_object_or_404, render from django.urls import reverse from django.utils.text import gettext_lazy as _ from django.views.generic import TemplateView from config.mixins import HTMXFormComponent from shop.filters import ProductFilter from shop.models import ( CartItem, CustomerAddress, Product, ProductPrice, ShippingMethod, WishlistedProduct, ) from users.forms.change_password import ChangePasswordForm from users.forms.info import UserInfoForm from web.forms import CustomerAddressForm from web.mixins import FilteredQuerysetMixin, PaginatedQuerysetMixin from web.settings import ANONYMOUS_CART_ID_COOKIE_NAME from web.utils import get_or_create_cart class ListProducts(TemplateView, FilteredQuerysetMixin, PaginatedQuerysetMixin): template_name = "web/list_products.html" queryset = Product.objects.filter(hidden=False).prefetch_related( "images", ) filter_class = ProductFilter def get_context_data(self, **kwargs): qs = self.get_queryset() page = self.get_paginated_queryset(qs) return { "current_page": page.number, "num_pages": page.paginator.num_pages, "page": page, "has_next_page": page.has_next(), "has_previous_page": page.has_previous(), } class ListWishlistedProducts( TemplateView, FilteredQuerysetMixin, PaginatedQuerysetMixin ): template_name = "web/list_products.html" queryset = Product.objects.filter(hidden=False).prefetch_related( "images", ) filter_class = ProductFilter def get_queryset(self): product_ids = WishlistedProduct.objects.filter( user=self.request.user ).values_list("product", flat=True) return super().get_queryset().filter(pk__in=product_ids).distinct() def get_context_data(self, **kwargs): qs = self.get_queryset() page = self.get_paginated_queryset(qs) return { "current_page": page.number, "num_pages": page.paginator.num_pages, "page": page, "has_next_page": page.has_next(), "has_previous_page": page.has_previous(), } def cart_dropdown(request, *args, **kwargs): cart, created = get_or_create_cart(request) response = render( request, "components/cart/cart_navbar.html", context={ "cart": cart, }, ) if not request.user.is_authenticated: response.set_cookie(ANONYMOUS_CART_ID_COOKIE_NAME, cart.uuid, samesite="strict") return response def cart(request, *args, **kwargs): cart, created = get_or_create_cart(request) items = CartItem.objects.filter(cart=cart) base_total = Decimal("0.00") tax_total = Decimal("0.00") for item in items: price = ProductPrice.objects.filter(product=item.product, current=True).first() base_total += price.price * item.quantity tax_total += round( price.price * Decimal(price.tax.value / 100) * item.quantity, 2 ) total = base_total + tax_total if request.user.is_authenticated: billing_address = CustomerAddress.objects.filter( user=request.user, address_type=CustomerAddress.Types.BILLING, default=True ).first() shipping_address = CustomerAddress.objects.filter( user=request.user, address_type=CustomerAddress.Types.SHIPPING, ).first() else: billing_address = None shipping_address = None shipping_methods = ShippingMethod.objects.all() response = render( request, "components/cart/cart.html", context={ "cart": cart, "items": items, "total": total, "base_total": base_total, "tax_total": tax_total, "billing_address": billing_address, "shipping_address": shipping_address, "shipping_methods": shipping_methods, }, ) if not request.user.is_authenticated: response.set_cookie(ANONYMOUS_CART_ID_COOKIE_NAME, cart.uuid, samesite="strict") return response class WishlistButton(TemplateView): template_name = "components/products/wishlist_button.html" def get_context_data(self, **kwargs): product = get_object_or_404(Product, pk=self.kwargs.get("pk")) is_wishlisted = WishlistedProduct.objects.filter( user=self.request.user, product=product ).exists() return { "is_wishlisted": is_wishlisted, "product": product, } class UserInfoFormComponentView(HTMXFormComponent): form_class = UserInfoForm hx_trigger = "updated_user_info" save_button_text = _("Actualizar datos") def get_object(self) -> models.Model: return self.request.user def get_view_name(self): return reverse("web:user_info_component") def get_initial_values(self, instance): return { "email": instance.email, "first_name": instance.first_name, "last_name": instance.last_name, } class ChangePasswordFormComponentView(HTMXFormComponent): form_class = ChangePasswordForm hx_trigger = "changed_password" save_button_text = _("Cambiar contraseƱa") def get_form(self, initial=True): if initial: return self.form_class(user=self.request.user) return self.form_class(self.request.user, self.request.POST) def get_object(self) -> models.Model: return self.request.user def get_view_name(self): return reverse("web:change_password_component") class ListCustomerAddressComponent(TemplateView): template_name = "components/users/retrieve_update_address.html" def get_context_data(self, **kwargs): return { "customer_addresses": CustomerAddress.objects.filter(user=self.request.user) } def update_customer_address(request, pk): customer_address = get_object_or_404(CustomerAddress, pk=pk, user=request.user) form = CustomerAddressForm(request.POST, instance=customer_address) if form.is_valid(): form.save() return HttpResponse(status=200, headers={"HX-Trigger": "updated_addresses"}) return HttpResponse(form.errors, status=400) list_products = ListProducts.as_view() list_wishlisted_products = ListWishlistedProducts.as_view() wishlist_button = WishlistButton.as_view() user_info_component = UserInfoFormComponentView.as_view() change_password_component = ChangePasswordFormComponentView.as_view() list_customer_addresses_component = ListCustomerAddressComponent.as_view()