mirror of
https://github.com/nottinghamtec/PyRIGS.git
synced 2026-01-17 13:32:15 +00:00
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from django.core.urlresolvers import reverse_lazy
|
|
from django.http import Http404, HttpResponseRedirect
|
|
from django.views import generic
|
|
from django.shortcuts import get_object_or_404
|
|
from django.contrib import messages
|
|
|
|
from RIGS import models
|
|
|
|
|
|
class InvoiceIndex(generic.ListView):
|
|
model = models.Invoice
|
|
template_name = 'RIGS/invoice_list.html'
|
|
|
|
def get_queryset(self):
|
|
active = self.model.objects.filter(void=False).select_related('payment_set')
|
|
set = []
|
|
for invoice in active:
|
|
if invoice.balance != 0:
|
|
set.append(invoice)
|
|
return set
|
|
|
|
|
|
class InvoiceDetail(generic.DetailView):
|
|
model = models.Invoice
|
|
|
|
|
|
class InvoiceVoid(generic.View):
|
|
def get(self, request, *args, **kwargs):
|
|
pk = kwargs.get('pk')
|
|
object = get_object_or_404(models.Invoice, pk=pk)
|
|
object.void = not object.void
|
|
object.save()
|
|
|
|
if object.void:
|
|
return HttpResponseRedirect(reverse_lazy('invoice_list'))
|
|
return HttpResponseRedirect(reverse_lazy('invoice_detail', kwargs={'pk': object.pk}))
|
|
|
|
|
|
class PaymentCreate(generic.CreateView):
|
|
model = models.Payment
|
|
|
|
def get_initial(self):
|
|
initial = super(generic.CreateView, self).get_initial()
|
|
invoicepk = self.request.GET.get('invoice', self.request.POST.get('invoice', None))
|
|
if invoicepk == None:
|
|
raise Http404()
|
|
invoice = get_object_or_404(models.Invoice, pk=invoicepk)
|
|
initial.update({'invoice': invoice})
|
|
return initial
|
|
|
|
def get_success_url(self):
|
|
messages.info(self.request, "location.reload()")
|
|
return reverse_lazy('closemodal')
|
|
|
|
|
|
class PaymentDelete(generic.DeleteView):
|
|
model = models.Payment
|
|
|
|
def get_success_url(self):
|
|
return self.request.POST.get('next') |