from django.contrib.auth.decorators import login_required from django.db.models import Count from django.urls import reverse_lazy from django.utils.decorators import method_decorator from django.utils.translation import ugettext as _ from django.shortcuts import render, get_object_or_404, redirect from django.contrib import messages from django.core.exceptions import PermissionDenied from django.views.generic import DeleteView from guardian.decorators import permission_required_or_403 as permission_required from djing.global_base_views import OrderedFilteredList from .models import Tariff, PeriodicPay from djing import lib from djing.lib.decorators import only_admins from . import forms login_decs = login_required, only_admins @method_decorator(login_decs, name='dispatch') @method_decorator(permission_required('tariff_app.view_tariff'), name='dispatch') class TariffsListView(OrderedFilteredList): """ Show Services(Tariffs) list """ template_name = 'tariff_app/tarifs.html' context_object_name = 'tariflist' model = Tariff queryset = Tariff.objects.annotate(usercount=Count('linkto_tariff__abon')) @login_required @only_admins @permission_required('tariff_app.change_tariff') def edit_tarif(request, tarif_id=0): tarif_id = lib.safe_int(tarif_id) if tarif_id == 0: if not request.user.has_perm('tariff_app.add_tariff'): raise PermissionDenied tarif = None else: if not request.user.has_perm('tariff_app.change_tariff'): raise PermissionDenied tarif = get_object_or_404(Tariff, pk=tarif_id) if request.method == 'POST': frm = forms.TariffForm(request.POST, instance=tarif) if frm.is_valid(): service = frm.save() if tarif is None: request.user.log(request.META, 'csrv', '"%(title)s", "%(descr)s", %(amount).2f' % { 'title': service.title or '-', 'descr': service.descr or '-', 'amount': service.amount or 0.0 }) messages.success(request, _('Service has been saved')) return redirect('tarifs:edit', tarif_id=service.pk) else: messages.warning(request, _('Some fields were filled incorrect, please try again')) else: frm = forms.TariffForm(instance=tarif) return render(request, 'tariff_app/editTarif.html', { 'form': frm, 'tarif_id': tarif_id }) @method_decorator(login_decs, name='dispatch') @method_decorator(permission_required('tariff_app.delete_tariff'), name='dispatch') class TariffDeleteView(DeleteView): model = Tariff pk_url_kwarg = 'tid' success_url = reverse_lazy('tarifs:home') def delete(self, request, *args, **kwargs): res = super().delete(request, *args, **kwargs) request.user.log(request.META, 'dsrv', '"%(title)s", "%(descr)s", %(amount).2f' % { 'title': self.object.title or '-', 'descr': self.object.descr or '-', 'amount': self.object.amount or 0.0 }) messages.success(request, _('Service has been deleted')) return res def get_context_data(self, **kwargs): kwargs['tid'] = self.kwargs.get('tid') return super().get_context_data(**kwargs) @method_decorator(login_decs, name='dispatch') @method_decorator(permission_required('tariff_app.view_periodicpay'), name='dispatch') class PeriodicPaysListView(OrderedFilteredList): context_object_name = 'pays' model = PeriodicPay template_name = 'tariff_app/periodic_pays/list.html' @login_required @only_admins def periodic_pay(request, pay_id=0): if pay_id != 0: pay_inst = get_object_or_404(PeriodicPay, pk=pay_id) if not request.user.has_perm('tariff_app.change_periodicpay'): raise PermissionDenied else: pay_inst = None if not request.user.has_perm('tariff_app.add_periodicpay'): raise PermissionDenied if request.method == 'POST': frm = forms.PeriodicPayForm(request.POST, instance=pay_inst) if frm.is_valid(): new_periodic_pay = frm.save() if pay_inst is None: comment = _('New periodic pay successfully created') else: comment = _('Periodic pay has been changed') messages.success(request, comment) return redirect('tarifs:periodic_pay_edit', new_periodic_pay.pk) else: messages.error(request, _('Some fields were filled incorrect, please try again')) else: frm = forms.PeriodicPayForm(instance=pay_inst) return render(request, 'tariff_app/periodic_pays/add_edit.html', { 'pay_instance': pay_inst, 'form': frm })