mirror of
https://github.com/ae-utbm/sith3.git
synced 2024-06-29 09:58:04 +00:00
123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
from django.shortcuts import render
|
|
from django.views.generic import ListView, DetailView, RedirectView
|
|
from django.views.generic.edit import UpdateView, CreateView, DeleteView
|
|
from django.forms.models import modelform_factory
|
|
from django.forms import CheckboxSelectMultiple
|
|
from django.core.urlresolvers import reverse_lazy
|
|
from django.contrib.auth.forms import AuthenticationForm
|
|
from django.utils import timezone
|
|
from django.conf import settings
|
|
|
|
from datetime import timedelta
|
|
|
|
from core.views import CanViewMixin, CanEditMixin, CanEditPropMixin
|
|
from subscription.models import Subscriber
|
|
from counter.models import Counter
|
|
|
|
class CounterDetail(DetailView):
|
|
"""
|
|
The public (barman) view
|
|
"""
|
|
model = Counter
|
|
template_name = 'counter/counter_detail.jinja'
|
|
pk_url_kwarg = "counter_id"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
"""
|
|
Get the barman list for the template
|
|
|
|
Also handle the timeout
|
|
"""
|
|
context = super(CounterDetail, self).get_context_data(**kwargs)
|
|
context['login_form'] = AuthenticationForm()
|
|
print(self.object.id)
|
|
print(list(Counter.barmen_session.keys()))
|
|
if str(self.object.id) in list(Counter.barmen_session.keys()):
|
|
if (timezone.now() - Counter.barmen_session[str(self.object.id)]['time']) < timedelta(minutes=settings.SITH_BARMAN_TIMEOUT):
|
|
context['barmen'] = []
|
|
for b in Counter.barmen_session[str(self.object.id)]['users']:
|
|
context['barmen'].append(Subscriber.objects.filter(id=b).first())
|
|
Counter.barmen_session[str(self.object.id)]['time'] = timezone.now()
|
|
else:
|
|
Counter.barmen_session[str(self.object.id)]['users'] = {}
|
|
else:
|
|
context['barmen'] = []
|
|
return context
|
|
|
|
class CounterLogin(RedirectView):
|
|
"""
|
|
Handle the login of a barman
|
|
|
|
Logged barmen are stored in the class-wide variable 'barmen_session', in the Counter model
|
|
"""
|
|
permanent = False
|
|
def post(self, request, *args, **kwargs):
|
|
"""
|
|
Register the logged user as barman for this counter
|
|
"""
|
|
self.counter_id = kwargs['counter_id']
|
|
# TODO: make some checks on the counter type
|
|
form = AuthenticationForm(request, data=request.POST)
|
|
if form.is_valid():
|
|
user = Subscriber.objects.filter(username=form.cleaned_data['username']).first()
|
|
if self.counter_id not in Counter.barmen_session.keys():
|
|
Counter.barmen_session[self.counter_id] = {'users': {user.id}, 'time': timezone.now()}
|
|
else:
|
|
Counter.barmen_session[self.counter_id]['users'].add(user.id)
|
|
else:
|
|
print("Error logging the barman") # TODO handle that nicely
|
|
return super(CounterLogin, self).post(request, *args, **kwargs)
|
|
|
|
def get_redirect_url(self, *args, **kwargs):
|
|
return reverse_lazy('counter:details', args=args, kwargs=kwargs)
|
|
|
|
class CounterLogout(RedirectView):
|
|
permanent = False
|
|
def post(self, request, *args, **kwargs):
|
|
"""
|
|
Unregister the user from the barman
|
|
"""
|
|
self.counter_id = kwargs['counter_id']
|
|
Counter.barmen_session[str(self.counter_id)]['users'].remove(int(request.POST['user_id']))
|
|
return super(CounterLogout, self).post(request, *args, **kwargs)
|
|
|
|
def get_redirect_url(self, *args, **kwargs):
|
|
return reverse_lazy('counter:details', args=args, kwargs=kwargs)
|
|
|
|
class CounterListView(CanViewMixin, ListView):
|
|
"""
|
|
A list view for the admins
|
|
"""
|
|
model = Counter
|
|
template_name = 'counter/counter_list.jinja'
|
|
|
|
class CounterEditView(CanEditMixin, UpdateView):
|
|
"""
|
|
Edit a counter's main informations (for the counter's admin)
|
|
"""
|
|
model = Counter
|
|
form_class = modelform_factory(Counter, fields=['name', 'club', 'type', 'products'],
|
|
widgets={'products':CheckboxSelectMultiple})
|
|
pk_url_kwarg = "counter_id"
|
|
template_name = 'counter/counter_edit.jinja'
|
|
|
|
class CounterCreateView(CanEditMixin, CreateView):
|
|
"""
|
|
Create a counter (for the admins)
|
|
"""
|
|
model = Counter
|
|
form_class = modelform_factory(Counter, fields=['name', 'club', 'type', 'products'],
|
|
widgets={'products':CheckboxSelectMultiple})
|
|
template_name = 'counter/counter_edit.jinja'
|
|
|
|
class CounterDeleteView(CanEditMixin, DeleteView):
|
|
"""
|
|
Delete a counter (for the admins)
|
|
"""
|
|
model = Counter
|
|
pk_url_kwarg = "counter_id"
|
|
template_name = 'core/delete_confirm.jinja'
|
|
success_url = reverse_lazy('counter:admin_list')
|
|
|
|
|