Source code for salesman.checkout.views

from django.utils.decorators import method_decorator
from django.views.decorators.cache import never_cache
from rest_framework import mixins, status, viewsets
from rest_framework.response import Response

from salesman.core.utils import get_salesman_model

from .payment import PaymentError, payment_methods_pool
from .serializers import CheckoutSerializer

Basket = get_salesman_model('Basket')


class CheckoutViewSet(mixins.CreateModelMixin, viewsets.GenericViewSet):
    """
    Checkout API endpoint.
    """

    serializer_class = CheckoutSerializer

    def get_view_name(self):
        name = super().get_view_name()
        if name == "Checkout List":
            return "Checkout"
        return name

    def get_queryset(self):
        pass

    def get_serializer_context(self):
        context = super().get_serializer_context()
        context['basket'], _ = Basket.objects.get_or_create_from_request(self.request)
        context['basket'].update(self.request)
        return context

    @method_decorator(never_cache)
    def dispatch(self, request, *args, **kwargs):
        return super().dispatch(request, *args, **kwargs)

    def create(self, request, *args, **kwargs):
        """
        Process the checkout, handle ``PaymentError``.
        """
        try:
            return super().create(request, *args, **kwargs)
        except PaymentError as e:
            return Response({'detail': str(e)}, status=status.HTTP_402_PAYMENT_REQUIRED)

    def list(self, request, *args, **kwargs):
        """
        Show a list of payment methods with errors if they exist.
        """
        instance = {'payment_methods': payment_methods_pool.get_payments('basket')}
        serializer = self.get_serializer(instance)
        return Response(serializer.data)