shoppy/web/views/components.py

from decimal import Decimal

from django.http import HttpResponse
from django.shortcuts import get_object_or_404, render
from django.views.generic import TemplateView

from shop.filters import ProductFilter
from shop.models import (
    CartItem,
    CustomerAddress,
    Product,
    ProductPrice,
    ShippingMethod,
    WishlistedProduct,
)
from users.forms.info import UserInfoForm
from web.forms import CustomerAddressForm
from web.mixins import FilteredQuerysetMixin, PaginatedQuerysetMixin
from web.models import WebSettings
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 {
            "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 {
            "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 UserInfoComponentView(TemplateView):
    template_name = "components/users/info.html"
    form_class = UserInfoForm

    def get_context_data(self, **kwargs):
        web_settings = WebSettings.load()
        user = self.request.user
        form = self.form_class(
            initial={
                "email": user.email,
                "first_name": user.first_name,
                "last_name": user.last_name,
            }
        )

        return {
            "form": form,
        }

    def post(self, request, *args, **kwargs):
        user = self.request.user
        form = self.form_class(request.POST)

        if form.is_valid():
            user.email = form.cleaned_data.get("email")
            user.first_name = form.cleaned_data.get("first_name")
            user.last_name = form.cleaned_data.get("last_name")
            user.save()
            return HttpResponse(status=200)

        return render(request, self.template_name, {"form": form})


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 = UserInfoComponentView.as_view()
list_customer_addresses_component = ListCustomerAddressComponent.as_view()