# -*- coding: utf-8 -*-
# This file is part of Shuup.
#
# Copyright (c) 2012-2021, Shuup Commerce Inc. All rights reserved.
#
# This source code is licensed under the OSL-3.0 license found in the
# LICENSE file in the root directory of this source tree.
from __future__ import unicode_literals
from django import forms
from django.contrib import messages
from django.http.response import HttpResponseRedirect
from django.utils.translation import ugettext_lazy as _
from django.views.generic import DetailView, UpdateView
from django.views.generic.edit import BaseDeleteView
from shuup.admin.toolbar import PostActionButton, Toolbar
from shuup.admin.utils.forms import add_form_errors_as_messages
from shuup.admin.utils.urls import get_model_url
from shuup.core.excs import NoPaymentToCreateException
from shuup.core.models import Order, PaymentStatus, Shop
from shuup.utils.money import Money
[docs]class OrderCreatePaymentView(UpdateView):
model = Order
template_name = "shuup/admin/orders/create_payment.jinja"
context_object_name = "order"
form_class = forms.Form # Augmented manually
[docs] def get_queryset(self):
shop_ids = Shop.objects.get_for_user(self.request.user).values_list("id", flat=True)
return Order.objects.exclude(deleted=True).filter(shop_id__in=shop_ids)
[docs] def get_context_data(self, **kwargs):
context = super(OrderCreatePaymentView, self).get_context_data(**kwargs)
context["title"] = _("Create Payment -- %s") % context["order"]
context["toolbar"] = Toolbar(
[
PostActionButton(
icon="fa fa-check-circle",
form_id="create_payment",
text=_("Create Payment"),
extra_css_class="btn-success",
),
],
view=self,
)
return context
[docs]class OrderSetPaidView(DetailView):
model = Order
[docs] def get_queryset(self):
shop_ids = Shop.objects.get_for_user(self.request.user).values_list("id", flat=True)
return Order.objects.exclude(deleted=True).filter(shop_id__in=shop_ids)
[docs] def get(self, request, *args, **kwargs):
return HttpResponseRedirect(get_model_url(self.get_object()))
[docs] def post(self, request, *args, **kwargs):
order = self.object = self.get_object()
error = False
if order.payment_status not in (PaymentStatus.DEFERRED, PaymentStatus.NOT_PAID):
error = True
messages.error(self.request, _("Only orders which are not paid or deferred can be set as paid."))
if order.taxful_total_price:
error = True
messages.error(self.request, _("Only zero price orders can be set as paid without creating a payment."))
if not error:
amount = Money(0, order.shop.currency)
order.create_payment(amount, description=_("Zero amount payment"))
messages.success(self.request, _("Order marked as paid."))
return HttpResponseRedirect(get_model_url(self.get_object()))
[docs]class OrderDeletePaymentView(BaseDeleteView):
model = Order
[docs] def get_queryset(self):
shop_ids = Shop.objects.get_for_user(self.request.user).values_list("id", flat=True)
return Order.objects.incomplete().filter(shop_id__in=shop_ids)
[docs] def delete(self, request, *args, **kwargs):
order = self.get_object()
payment_id = request.POST.get("payment")
order_url = get_model_url(self.get_object())
payment = order.payments.filter(pk=payment_id).first() if payment_id else None
if not payment:
messages.error(self.request, _("Payment doesn't exist."))
return HttpResponseRedirect(order_url)
payment.delete()
order.update_payment_status()
messages.success(self.request, _("Payment deleted."))
return HttpResponseRedirect(order_url)