django-orchestra/orchestra/contrib/orders/actions.py

175 lines
6.7 KiB
Python
Raw Normal View History

from django.contrib import admin, messages
from django.core.urlresolvers import reverse
2014-09-16 17:14:24 +00:00
from django.db import transaction
2015-07-10 13:00:51 +00:00
from django.utils import timezone
from django.utils.safestring import mark_safe
2014-10-11 16:21:51 +00:00
from django.utils.translation import ungettext, ugettext_lazy as _
from django.shortcuts import render
2014-10-11 16:21:51 +00:00
from orchestra.admin.utils import change_url
from .forms import BillSelectedOptionsForm, BillSelectConfirmationForm, BillSelectRelatedForm
class BillSelectedOrders(object):
""" Form wizard for billing orders admin action """
short_description = _("Bill selected orders")
2015-04-01 15:49:21 +00:00
verbose_name = _("Bill")
template = 'admin/orders/order/bill_selected_options.html'
__name__ = 'bill_selected_orders'
def __call__(self, modeladmin, request, queryset):
""" make this monster behave like a function """
self.modeladmin = modeladmin
self.queryset = queryset
opts = modeladmin.model._meta
app_label = opts.app_label
self.context = {
'opts': opts,
'app_label': app_label,
'queryset': queryset,
'action_checkbox_name': admin.helpers.ACTION_CHECKBOX_NAME,
}
ret = self.set_options(request)
del(self.queryset)
del(self.context)
return ret
def set_options(self, request):
form = BillSelectedOptionsForm()
if request.POST.get('step'):
form = BillSelectedOptionsForm(request.POST)
if form.is_valid():
self.options = dict(
billing_point=form.cleaned_data['billing_point'],
fixed_point=form.cleaned_data['fixed_point'],
2014-10-11 16:21:51 +00:00
proforma=form.cleaned_data['proforma'],
2014-09-19 14:47:25 +00:00
new_open=form.cleaned_data['new_open'],
)
2014-09-11 14:00:20 +00:00
if int(request.POST.get('step')) != 3:
return self.select_related(request)
else:
return self.confirmation(request)
self.context.update({
'title': _("Options for billing selected orders, step 1 / 3"),
2014-09-10 16:53:09 +00:00
'step': 1,
'form': form,
})
return render(request, self.template, self.context)
def select_related(self, request):
2014-09-15 12:15:32 +00:00
# TODO use changelist ?
related = self.queryset.get_related().select_related('account', 'service')
2014-09-14 09:52:45 +00:00
if not related:
return self.confirmation(request)
2014-09-10 16:53:09 +00:00
self.options['related_queryset'] = related
form = BillSelectRelatedForm(initial=self.options)
2014-09-10 16:53:09 +00:00
if int(request.POST.get('step')) >= 2:
form = BillSelectRelatedForm(request.POST, initial=self.options)
if form.is_valid():
select_related = form.cleaned_data['selected_related']
2014-09-11 14:00:20 +00:00
self.queryset = self.queryset | select_related
return self.confirmation(request)
self.context.update({
'title': _("Select related order for billing, step 2 / 3"),
2014-09-10 16:53:09 +00:00
'step': 2,
'form': form,
})
return render(request, self.template, self.context)
2014-09-16 17:14:24 +00:00
@transaction.atomic
def confirmation(self, request):
form = BillSelectConfirmationForm(initial=self.options)
2014-09-10 16:53:09 +00:00
if int(request.POST.get('step')) >= 3:
bills = self.queryset.bill(commit=True, **self.options)
2014-09-16 17:14:24 +00:00
for order in self.queryset:
2014-10-11 16:21:51 +00:00
self.modeladmin.log_change(request, order, _("Billed"))
if not bills:
msg = _("Selected orders do not have pending billing")
self.modeladmin.message_user(request, msg, messages.WARNING)
else:
2014-10-11 16:21:51 +00:00
num = len(bills)
if num == 1:
url = change_url(bills[0])
else:
url = reverse('admin:bills_bill_changelist')
ids = ','.join([str(b.id) for b in bills])
2014-10-11 16:21:51 +00:00
url += '?id__in=%s' % ids
2014-09-10 16:53:09 +00:00
msg = ungettext(
'<a href="{url}">One bill</a> has been created.',
'<a href="{url}">{num} bills</a> have been created.',
num).format(url=url, num=num)
msg = mark_safe(msg)
self.modeladmin.message_user(request, msg, messages.INFO)
return
2014-09-10 16:53:09 +00:00
bills = self.queryset.bill(commit=False, **self.options)
bills_with_total = []
for account, lines in bills:
total = 0
for line in lines:
discount = sum([discount.total for discount in line.discounts])
total += line.subtotal + discount
bills_with_total.append((account, total, lines))
self.context.update({
'title': _("Confirmation for billing selected orders"),
2014-09-10 16:53:09 +00:00
'step': 3,
'form': form,
2015-05-27 14:05:25 +00:00
'bills': sorted(bills_with_total, key=lambda i: -i[1]),
})
return render(request, self.template, self.context)
2014-10-16 17:14:21 +00:00
@transaction.atomic
def mark_as_ignored(modeladmin, request, queryset):
""" Mark orders as ignored """
for order in queryset:
order.mark_as_ignored()
modeladmin.log_change(request, order, 'Marked as ignored')
num = len(queryset)
msg = ungettext(
_("Selected order has been marked as ignored."),
_("%i selected orders have been marked as ignored.") % num,
num)
modeladmin.message_user(request, msg)
@transaction.atomic
def mark_as_not_ignored(modeladmin, request, queryset):
""" Mark orders as ignored """
for order in queryset:
order.mark_as_not_ignored()
modeladmin.log_change(request, order, 'Marked as not ignored')
num = len(queryset)
msg = ungettext(
_("Selected order has been marked as not ignored."),
_("%i selected orders have been marked as not ignored.") % num,
num)
modeladmin.message_user(request, msg)
2015-07-10 13:00:51 +00:00
def report(modeladmin, request, queryset):
services = {}
totals = [0, 0, None, 0]
now = timezone.now().date()
for order in queryset.select_related('service'):
name = order.service.description
active, cancelled = (1, 0) if not order.cancelled_on or order.cancelled_on > now else (0, 1)
try:
info = services[name]
except KeyError:
nominal_price = order.service.nominal_price
info = [active, cancelled, nominal_price, 1]
services[name] = info
else:
info[0] += active
info[1] += cancelled
info[3] += 1
totals[0] += active
totals[1] += cancelled
totals[3] += 1
context = {
'services': sorted(services.items(), key=lambda n: -n[1][0]),
'totals': totals,
}
return render(request, 'admin/orders/order/report.html', context)