debusy/schedule/views.py

63 lines
2.4 KiB
Python
Raw Permalink Normal View History

2020-04-13 11:59:45 +00:00
from django.shortcuts import get_object_or_404, render
from .models import *
from django.http import Http404, HttpResponseRedirect
from django.urls import reverse
2020-04-09 12:11:12 +00:00
2020-04-13 11:59:45 +00:00
def schedule(request, date):
date = get_object_or_404(PickupDate, date=date)
member_number = request.POST and request.POST.get('member_number') \
or None
member_id = request.session.get('member_id')
if member_id is not None:
member = get_object_or_404(Member, pk=member_id)
elif member_number:
member = Member()
request.session['member_id'] = member.id
request.session.set_expiry(0)
else:
member = None
if member_number and member.member_number != member_number:
member.member_number = member_number
member.save()
timeslots = Timeslot.objects.all()
for timeslot in timeslots:
timeslot.sum = len(Pickup.objects.filter(time=timeslot))
if member:
try:
timeslot.own_pickup = \
Pickup.objects.filter(
date=date,
time=timeslot,
member=member,
).exists()
except Pickup.DoesNotExist:
timeslot.own_pickup = False
else:
timeslot.own_pickup = False
return render(request, 'schedule/schedule.html',
{ 'date': date,
'timeslots': timeslots,
})
def submit_pickup(request, date):
date = get_object_or_404(PickupDate, date=date)
request.session.set_expiry(0)
member_id = request.session.get('member_id')
if not member_id:
member = Member(member_number=request.POST.get('member_number'))
member.save()
request.session['member_id'] = member.id
else:
member = get_object_or_404(Member, pk=member_id)
timeslot = get_object_or_404(Timeslot, pk=request.POST.get('timeslot'))
Pickup(date=date, time=timeslot, member=member).save()
return HttpResponseRedirect(reverse('schedule:schedule', args=(date.date,)))
def submit_pickup_removal(request, date):
date = get_object_or_404(PickupDate, date=date)
member = get_object_or_404(Member, pk=request.session.get('member_id'))
timeslot = get_object_or_404(Timeslot, pk=request.POST.get('timeslot'))
Pickup.objects.filter(date=date, time=timeslot, member=member).delete()
return HttpResponseRedirect(reverse('schedule:schedule', args=(date.date,)))