mirror of https://gitlab.crans.org/bde/nk20
151 lines
4.6 KiB
Python
151 lines
4.6 KiB
Python
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from dal import autocomplete
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.db.models import Q
|
|
from django.urls import reverse
|
|
from django.utils.translation import gettext_lazy as _
|
|
from django.views.generic import CreateView, ListView, UpdateView
|
|
|
|
from .models import Transaction, TransactionTemplate, Alias, TemplateTransaction
|
|
from .forms import TransactionForm, TransactionTemplateForm, ConsoForm
|
|
|
|
|
|
class TransactionCreate(LoginRequiredMixin, CreateView):
|
|
"""
|
|
Show transfer page
|
|
|
|
TODO: If user have sufficient rights, they can transfer from an other note
|
|
"""
|
|
model = Transaction
|
|
form_class = TransactionForm
|
|
|
|
def get_context_data(self, **kwargs):
|
|
"""
|
|
Add some context variables in template such as page title
|
|
"""
|
|
context = super().get_context_data(**kwargs)
|
|
context['title'] = _('Transfer money from your account '
|
|
'to one or others')
|
|
|
|
context['no_cache'] = True
|
|
|
|
return context
|
|
|
|
def get_form(self, form_class=None):
|
|
"""
|
|
If the user has no right to transfer funds, then it won't have the choice of the source of the transfer.
|
|
"""
|
|
form = super().get_form(form_class)
|
|
|
|
if False: # TODO: fix it with "if %user has no right to transfer funds"
|
|
del form.fields['source']
|
|
form.user = self.request.user
|
|
|
|
return form
|
|
|
|
def get_success_url(self):
|
|
return reverse('note:transfer')
|
|
|
|
|
|
class NoteAutocomplete(autocomplete.Select2QuerySetView):
|
|
"""
|
|
Auto complete note by aliases
|
|
"""
|
|
def get_queryset(self):
|
|
"""
|
|
Quand une personne cherche un alias, une requête est envoyée sur l'API dédiée à l'auto-complétion.
|
|
Cette fonction récupère la requête, et renvoie la liste filtrée des aliases.
|
|
"""
|
|
# Un utilisateur non connecté n'a accès à aucune information
|
|
if not self.request.user.is_authenticated:
|
|
return Alias.objects.none()
|
|
|
|
qs = Alias.objects.all()
|
|
|
|
# self.q est le paramètre de la recherche
|
|
if self.q:
|
|
qs = qs.filter(Q(name__regex=self.q) | Q(normalized_name__regex=Alias.normalize(self.q)))\
|
|
.order_by('normalized_name').distinct()
|
|
|
|
# Filtrage par type de note (user, club, special)
|
|
note_type = self.forwarded.get("note_type", None)
|
|
if note_type:
|
|
types = str(note_type).lower()
|
|
if "user" in types:
|
|
qs = qs.filter(note__polymorphic_ctype__model="noteuser")
|
|
elif "club" in types:
|
|
qs = qs.filter(note__polymorphic_ctype__model="noteclub")
|
|
elif "special" in types:
|
|
qs = qs.filter(note__polymorphic_ctype__model="notespecial")
|
|
else:
|
|
qs = qs.none()
|
|
|
|
return qs
|
|
|
|
def get_result_label(self, result):
|
|
# Gère l'affichage de l'alias dans la recherche
|
|
res = result.name
|
|
note_name = str(result.note)
|
|
if res != note_name:
|
|
res += " (aka. " + note_name + ")"
|
|
return res
|
|
|
|
def get_result_value(self, result):
|
|
# Le résultat renvoyé doit être l'identifiant de la note, et non de l'alias
|
|
return str(result.note.pk)
|
|
|
|
|
|
class TransactionTemplateCreateView(LoginRequiredMixin, CreateView):
|
|
"""
|
|
Create TransactionTemplate
|
|
"""
|
|
model = TransactionTemplate
|
|
form_class = TransactionTemplateForm
|
|
|
|
|
|
class TransactionTemplateListView(LoginRequiredMixin, ListView):
|
|
"""
|
|
List TransactionsTemplates
|
|
"""
|
|
model = TransactionTemplate
|
|
form_class = TransactionTemplateForm
|
|
|
|
|
|
class TransactionTemplateUpdateView(LoginRequiredMixin, UpdateView):
|
|
"""
|
|
"""
|
|
model = TransactionTemplate
|
|
form_class = TransactionTemplateForm
|
|
|
|
|
|
class ConsoView(LoginRequiredMixin, CreateView):
|
|
"""
|
|
Consume
|
|
"""
|
|
model = TemplateTransaction
|
|
template_name = "note/conso_form.html"
|
|
form_class = ConsoForm
|
|
|
|
def get_context_data(self, **kwargs):
|
|
"""
|
|
Add some context variables in template such as page title
|
|
"""
|
|
context = super().get_context_data(**kwargs)
|
|
context['transaction_templates'] = TransactionTemplate.objects.filter(display=True) \
|
|
.order_by('category')
|
|
context['title'] = _("Consommations")
|
|
|
|
# select2 compatibility
|
|
context['no_cache'] = True
|
|
|
|
return context
|
|
|
|
def get_success_url(self):
|
|
"""
|
|
When clicking a button, reload the same page
|
|
"""
|
|
return reverse('note:consos')
|
|
|