nk20/apps/note/views.py

164 lines
6.4 KiB
Python
Raw Normal View History

2020-02-18 20:30:26 +00:00
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
2019-07-17 11:34:07 +00:00
# SPDX-License-Identifier: GPL-3.0-or-later
2020-04-27 01:21:13 +00:00
import json
2019-07-17 11:34:07 +00:00
from django.conf import settings
2019-07-17 11:53:58 +00:00
from django.contrib.auth.mixins import LoginRequiredMixin
from django.contrib.contenttypes.models import ContentType
2020-07-30 13:49:59 +00:00
from django.db.models import Q
2019-07-17 11:34:07 +00:00
from django.utils.translation import gettext_lazy as _
2020-03-24 23:12:56 +00:00
from django.views.generic import CreateView, UpdateView
from django_tables2 import SingleTableView
2020-03-23 19:21:25 +00:00
from django.urls import reverse_lazy
from note_kfet.inputs import AmountInput
2020-03-20 13:43:35 +00:00
from permission.backends import PermissionBackend
from permission.views import ProtectQuerysetMixin
2020-03-20 01:14:43 +00:00
2020-03-12 22:12:49 +00:00
from .forms import TransactionTemplateForm
2020-07-25 15:25:57 +00:00
from .models import TemplateCategory, Transaction, TransactionTemplate, RecurrentTransaction, NoteSpecial
2020-03-14 14:13:58 +00:00
from .models.transactions import SpecialTransaction
2020-03-23 14:35:24 +00:00
from .tables import HistoryTable, ButtonTable
2020-02-18 11:31:15 +00:00
class TransactionCreateView(ProtectQuerysetMixin, LoginRequiredMixin, SingleTableView):
2019-07-17 11:34:07 +00:00
"""
2020-03-23 14:28:09 +00:00
View for the creation of Transaction between two note which are not :models:`transactions.RecurrentTransaction`.
e.g. for donation/transfer between people and clubs or for credit/debit with :models:`note.NoteSpecial`
2019-07-17 11:34:07 +00:00
"""
2020-03-23 14:28:09 +00:00
template_name = "note/transaction_form.html"
2019-07-17 11:53:58 +00:00
model = Transaction
2020-03-16 11:11:16 +00:00
# Transaction history table
table_class = HistoryTable
2020-07-30 15:30:21 +00:00
extra_context = {"title": _("Transfer money")}
2020-03-16 11:11:16 +00:00
def get_queryset(self, **kwargs):
2020-07-25 15:25:57 +00:00
return super().get_queryset(**kwargs).order_by("-created_at").all()[:20]
2019-07-17 11:53:58 +00:00
def get_context_data(self, **kwargs):
"""
Add some context variables in template such as page title
"""
context = super().get_context_data(**kwargs)
context['amount_widget'] = AmountInput(attrs={"id": "amount"})
2020-03-12 22:12:49 +00:00
context['polymorphic_ctype'] = ContentType.objects.get_for_model(Transaction).pk
2020-03-14 14:13:58 +00:00
context['special_polymorphic_ctype'] = ContentType.objects.get_for_model(SpecialTransaction).pk
context['special_types'] = NoteSpecial.objects\
.filter(PermissionBackend.filter_queryset(self.request.user, NoteSpecial, "view"))\
.order_by("special_type").all()
2020-02-21 17:28:21 +00:00
2020-04-06 06:58:39 +00:00
# Add a shortcut for entry page for open activities
if "activity" in settings.INSTALLED_APPS:
from activity.models import Activity
context["activities_open"] = Activity.objects.filter(open=True).filter(
PermissionBackend.filter_queryset(self.request.user, Activity, "view")).filter(
PermissionBackend.filter_queryset(self.request.user, Activity, "change")).all()
2019-07-17 11:53:58 +00:00
return context
2019-08-11 17:54:18 +00:00
class TransactionTemplateCreateView(ProtectQuerysetMixin, LoginRequiredMixin, CreateView):
2019-08-11 17:54:18 +00:00
"""
2020-04-06 06:58:39 +00:00
Create Transaction template
2019-08-11 17:54:18 +00:00
"""
model = TransactionTemplate
form_class = TransactionTemplateForm
2020-03-24 21:13:15 +00:00
success_url = reverse_lazy('note:template_list')
2020-07-30 15:30:21 +00:00
extra_context = {"title": _("Create new button")}
2020-02-18 11:31:15 +00:00
2020-03-24 23:03:48 +00:00
class TransactionTemplateListView(ProtectQuerysetMixin, LoginRequiredMixin, SingleTableView):
2019-08-11 17:54:18 +00:00
"""
2020-04-06 06:58:39 +00:00
List Transaction templates
2019-08-11 17:54:18 +00:00
"""
model = TransactionTemplate
2020-03-10 19:17:56 +00:00
table_class = ButtonTable
2020-07-30 15:30:21 +00:00
extra_context = {"title": _("Search button")}
2020-07-30 13:49:59 +00:00
def get_queryset(self, **kwargs):
"""
Filter the user list with the given pattern.
"""
qs = super().get_queryset().distinct()
if "search" in self.request.GET:
pattern = self.request.GET["search"]
qs = qs.filter(Q(name__iregex="^" + pattern) | Q(destination__club__name__iregex="^" + pattern))
qs = qs.order_by('-display', 'category__name', 'destination__club__name', 'name')
return qs
2020-02-18 11:31:15 +00:00
class TransactionTemplateUpdateView(ProtectQuerysetMixin, LoginRequiredMixin, UpdateView):
2019-08-11 17:54:18 +00:00
"""
2020-04-06 06:58:39 +00:00
Update Transaction template
2019-08-11 17:54:18 +00:00
"""
model = TransactionTemplate
2020-02-04 00:18:03 +00:00
form_class = TransactionTemplateForm
2020-03-24 22:32:48 +00:00
success_url = reverse_lazy('note:template_list')
2020-07-30 15:30:21 +00:00
extra_context = {"title": _("Update button")}
2020-02-18 11:31:15 +00:00
2020-04-27 01:21:13 +00:00
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
if "logs" in settings.INSTALLED_APPS:
from logs.models import Changelog
update_logs = Changelog.objects.filter(
model=ContentType.objects.get_for_model(TransactionTemplate),
instance_pk=self.object.pk,
action="edit",
)
price_history = []
for log in update_logs.all():
old_dict = json.loads(log.previous)
new_dict = json.loads(log.data)
old_price = old_dict["amount"]
new_price = new_dict["amount"]
if old_price != new_price:
price_history.append(dict(price=old_price, time=log.timestamp))
price_history.append(dict(price=self.object.amount, time=None))
price_history.reverse()
context["price_history"] = price_history
return context
2020-03-24 23:03:48 +00:00
class ConsoView(ProtectQuerysetMixin, LoginRequiredMixin, SingleTableView):
2020-02-04 00:18:03 +00:00
"""
2020-03-23 14:28:09 +00:00
The Magic View that make people pay their beer and burgers.
(Most of the magic happens in the dark world of Javascript see consos.js)
2020-02-04 00:18:03 +00:00
"""
model = Transaction
2020-02-04 00:18:03 +00:00
template_name = "note/conso_form.html"
2020-07-30 15:30:21 +00:00
extra_context = {"title": _("Consumptions")}
2020-03-10 07:07:09 +00:00
# Transaction history table
table_class = HistoryTable
2020-02-04 00:18:03 +00:00
def get_queryset(self, **kwargs):
2020-07-25 15:25:57 +00:00
return super().get_queryset(**kwargs).order_by("-created_at")[:20]
2020-02-04 00:18:03 +00:00
def get_context_data(self, **kwargs):
"""
Add some context variables in template such as page title
"""
context = super().get_context_data(**kwargs)
2020-07-25 15:25:57 +00:00
categories = TemplateCategory.objects.order_by('name').all()
for category in categories:
category.templates_filtered = category.templates.filter(
PermissionBackend().filter_queryset(self.request.user, TransactionTemplate, "view")
).filter(display=True).order_by('name').all()
context['categories'] = [cat for cat in categories if cat.templates_filtered]
context['highlighted'] = TransactionTemplate.objects.filter(highlighted=True).filter(
2020-03-20 01:14:43 +00:00
PermissionBackend().filter_queryset(self.request.user, TransactionTemplate, "view")
2020-07-25 15:25:57 +00:00
).order_by('name').all()
context['polymorphic_ctype'] = ContentType.objects.get_for_model(RecurrentTransaction).pk
2020-02-04 00:18:03 +00:00
# select2 compatibility
2020-02-21 18:46:56 +00:00
context['no_cache'] = True
2020-02-04 00:18:03 +00:00
return context