Source code for eggplant.market.views.payment

import logging

from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.shortcuts import get_object_or_404, redirect, render
from django.utils.decorators import method_decorator
from django.utils.translation import ugettext as _
from django.views.generic.detail import DetailView
from eggplant.core.views import LoginRequiredMixin
from getpaid.forms import PaymentMethodForm

from ..models import Payment

log = logging.getLogger(__name__)


@login_required
def payment_list(request):
    payments = Payment.objects.filter(account__user_profiles=request.user.profile).order_by('-created')
    ctx = {
        'payments': payments
    }
    return render(request, 'eggplant/market/payment_list.html', ctx)


@login_required
def payment_info(request, pk=None):
    payment = get_object_or_404(Payment, pk=pk, account__user_profiles=request.user.profile)
    ctx = {
        'payments': [payment]
    }
    return render(request, 'eggplant/market/payment_list.html', ctx)


class PaymentView(LoginRequiredMixin, DetailView):
    model = Payment
    template_name = 'eggplant/market/payment_detail.html'

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

    def get_queryset(self):
        return Payment.objects.filter(account__user_profiles=self.request.user.profile)

    def get_context_data(self, **kwargs):
        context = super(PaymentView, self).get_context_data(**kwargs)
        context['payment_form'] = PaymentMethodForm(
            self.object.amount.currency,
            initial={'payment': self.object}
        )
        return context

payment_detail = PaymentView.as_view()


def payment_accepted(request, pk=None):
    get_object_or_404(Payment, pk=pk, account__user_profiles=request.user.profile)
    messages.info(request, _("Your payment has been accepted and"
                             " it's being processed."))
    return redirect('eggplant:market:payments_list')


@login_required
def payment_rejected(request, pk=None):
    get_object_or_404(Payment, pk=pk, account__user_profiles=request.user.profile)
    messages.error(request, _("Your payment has been cancelled."))
    return redirect("eggplant:market:payments_list")