Newer
Older
from django.conf import settings
from django.core.exceptions import PermissionDenied, SuspiciousOperation
from django.shortcuts import get_object_or_404, redirect, render
from django.urls import reverse, reverse_lazy
from django.utils.decorators import method_decorator
from django.utils.translation import ugettext as _
from django_tables2.views import RequestConfig, SingleTableView
from payments import PaymentStatus, RedirectNeeded
from rules.contrib.views import PermissionRequiredMixin
from aleksis.core.mixins import AdvancedCreateView, AdvancedDeleteView, AdvancedEditView
from .filters import InvoicesFilter
from .forms import EditClientForm, EditInvoiceGroupForm, InvoicesActionForm
from .models.invoice import Invoice, InvoiceGroup
from .tables import ClientsTable, InvoiceGroupsTable, InvoicesTable
class GetInvoicePDF(PermissionRequiredMixin, RenderPDFView):
permission_required = "tezor.print_invoice_rule"
def get_context_data(self, *args, **kwargs):
context = super().get_context_data(*args, **kwargs)
invoice = Invoice.objects.get(token=self.kwargs["token"])
class DoPaymentView(PermissionRequiredMixin, View):
model = Invoice
permission_required = "tezor.do_payment_rule"
template_name = "tezor/invoice/payment.html"
def dispatch(self, request, token):
self.object = get_object_or_404(self.model, token=token)
new_variant = request.GET.get("variant", None)
if new_variant:
if request.user.has_perm("tezor.change_payment_variant", self.object):
if new_variant in settings.PAYMENT_VARIANTS:
self.object.variant = new_variant
self.object.save()
raise SuspiciousOperation()
if self.object.status not in [
PaymentStatus.WAITING,
PaymentStatus.INPUT,
PaymentStatus.REJECTED,
]:
return redirect(self.object.get_success_url())
try:
form = self.object.get_form(data=request.POST or None)
except RedirectNeeded as redirect_to:
return redirect(str(redirect_to))
context = {
"form": form,
"payment": self.object,
}
return render(request, self.template_name, context)
class ClientListView(PermissionRequiredMixin, SingleTableView):
"""Table of all clients."""
model = Client
table_class = ClientsTable
@method_decorator(never_cache, name="dispatch")
class ClientCreateView(PermissionRequiredMixin, AdvancedCreateView):
"""Create view for clients."""
model = Client
form_class = EditClientForm
permission_required = "tezor.create_client_rule"
success_url = reverse_lazy("clients")
success_message = _("The client has been created.")
@method_decorator(never_cache, name="dispatch")
class ClientEditView(PermissionRequiredMixin, AdvancedEditView):
"""Edit view for clients."""
model = Client
form_class = EditClientForm
success_url = reverse_lazy("clients")
success_message = _("The client has been saved.")
class ClientDeleteView(PermissionRequiredMixin, AdvancedDeleteView):
"""Delete view for client."""
model = Client
permission_required = "tezor.delete_client_rule"
template_name = "core/pages/delete.html"
success_url = reverse_lazy("clients")
success_message = _("The client has been deleted.")
class ClientDetailView(PermissionRequiredMixin, DetailView):
model = Client
template_name = "tezor/client/full.html"
def get_context_data(self, object):
context = super().get_context_data()
invoice_groups = object.invoice_groups.all()
invoice_groups_table = InvoiceGroupsTable(invoice_groups)
RequestConfig(self.request).configure(invoice_groups_table)
context["invoice_groups_table"] = invoice_groups_table
return context
class InvoiceGroupDetailView(PermissionRequiredMixin, DetailView):
model = InvoiceGroup
permission_required = "tezor.view_invoice_group_rule"
template_name = "tezor/invoice_group/full.html"
def post(self, request, *args, **kwargs):
r = super().get(request, *args, **kwargs)
if self.invoices_action_form.is_valid():
action = self.invoices_action_form._get_actions_dict()[self.invoices_action_form.cleaned_data["action"]]
if request.user.has_perm(action.permission):
self.invoices_action_form.execute()
return r
def get_context_data(self, object):
context = super().get_context_data()
qs = object.invoices.all()
invoices_filter = InvoicesFilter(self.request.GET, qs)
context["filter"] = invoices_filter
invoices = invoices_filter.qs
invoices_table = InvoicesTable(invoices)
RequestConfig(self.request).configure(invoices_table)
context["invoices_table"] = invoices_table
self.invoices_action_form = InvoicesActionForm(self.request, self.request.POST or None, queryset=invoices)
context["action_form"] = self.invoices_action_form
return context
@method_decorator(never_cache, name="dispatch")
class InvoiceGroupCreateView(PermissionRequiredMixin, AdvancedCreateView):
"""Create view for invoice_groups."""
model = InvoiceGroup
form_class = EditInvoiceGroupForm
template_name = "tezor/invoice_group/create.html"
success_message = _("The invoice group has been created.")
def form_valid(self, form):
client = Client.objects.get(id=self.kwargs["pk"])
InvoiceGroup.objects.create(
client=client,
name=form.cleaned_data["name"],
template_name=form.cleaned_data["template_name"],
)
return super().form_valid(form)
def get_success_url(self):
return reverse("client_by_pk", args=[self.object.client.pk])
@method_decorator(never_cache, name="dispatch")
class InvoiceGroupEditView(PermissionRequiredMixin, AdvancedEditView):
"""Edit view for invoice_groups."""
model = InvoiceGroup
form_class = EditInvoiceGroupForm
permission_required = "tezor.edit_invoice_group_rule"
template_name = "tezor/invoice_group/edit.html"
success_url = reverse_lazy("invoice_groups")
success_message = _("The invoice group has been saved.")
class InvoiceGroupDeleteView(PermissionRequiredMixin, AdvancedDeleteView):
"""Delete view for invoice_group."""
model = InvoiceGroup
permission_required = "tezor.delete_invoice_group_rule"
template_name = "core/pages/delete.html"
success_url = reverse_lazy("invoice_groups")
success_message = _("The invoice group has been deleted.")
class InvoiceDetailView(PermissionRequiredMixin, DetailView):
model = Invoice
template_name = "tezor/invoice/full.html"
class SendInvoiceEmail(PermissionRequiredMixin, View):
permission_required = "tezor.send_invoice_email_rule"
def get(self, request, token):
email_invoice.delay(token)
url = request.META.get("HTTP_REFERRER")
if not url:
url = Invoice.objects.get(token=token).get_absolute_url()
return redirect(url)