151 lines
4.5 KiB
Python
151 lines
4.5 KiB
Python
from django.shortcuts import render, get_object_or_404, redirect
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.utils import timezone
|
|
from django.db.models.fields import DateTimeField
|
|
from django.db.models import F, Count, Q, ExpressionWrapper
|
|
from .models import ShiftRegistration, Room, Shift, Helper, Message
|
|
from django.views.generic import DetailView, ListView
|
|
from django.contrib.auth.mixins import LoginRequiredMixin
|
|
from django.contrib import messages
|
|
from .forms import BulkMessage, HelperShift
|
|
|
|
# Create your views here.
|
|
def index(request):
|
|
pass
|
|
|
|
|
|
@login_required
|
|
def shift_overview(request):
|
|
context = {}
|
|
context["running_shifts"] = [
|
|
shift
|
|
for shift in Shift.objects.annotate(
|
|
end_at=ExpressionWrapper(
|
|
F("start_at") + F("duration"),
|
|
output_field=DateTimeField(),
|
|
)
|
|
)
|
|
.filter(start_at__lte=timezone.now(), end_at__gte=timezone.now(), deleted=False)
|
|
.order_by("start_at")
|
|
]
|
|
|
|
# only Postgres supports DISTINCT on specific columns, SQLite does not support aggregates on datetime fields
|
|
context["next_shifts"] = filter(
|
|
lambda x: x is not None,
|
|
(
|
|
Shift.objects.filter(room=room, start_at__gt=timezone.now(), deleted=False)
|
|
.order_by("start_at")
|
|
.first()
|
|
for room in Room.objects.all()
|
|
),
|
|
)
|
|
|
|
return render(request, "shift_overview.html", context)
|
|
|
|
|
|
def add_helper_shift(self):
|
|
pass
|
|
|
|
|
|
@login_required
|
|
def shift_detail(request, pk):
|
|
shift = get_object_or_404(Shift, pk=pk)
|
|
form = HelperShift()
|
|
if request.method == "POST":
|
|
form = HelperShift(request.POST)
|
|
if form.is_valid():
|
|
(reg, created) = ShiftRegistration.objects.get_or_create(
|
|
helper=form.cleaned_data["helper"], shift=shift
|
|
)
|
|
if created:
|
|
messages.add_message(
|
|
request,
|
|
messages.SUCCESS,
|
|
"Helfer erfolgreich zur Schicht hinzugefügt",
|
|
)
|
|
else:
|
|
messages.add_message(
|
|
request,
|
|
messages.WARNING,
|
|
"Helfer ist bereits für diese Schicht angemeldet",
|
|
)
|
|
return redirect("team:shift", pk=shift.pk)
|
|
context = {
|
|
"shift": shift,
|
|
"add_helper_form": form,
|
|
}
|
|
return render(request, "shift_detail.html", context)
|
|
|
|
|
|
@login_required
|
|
def bulk_message(request):
|
|
form = BulkMessage()
|
|
|
|
if request.method == "POST":
|
|
form = BulkMessage(request.POST)
|
|
if form.is_valid():
|
|
try:
|
|
Message.objects.bulk_create(
|
|
(
|
|
Message(
|
|
text=form.cleaned_data["message"].replace(
|
|
"$token",
|
|
f"https://kontakt.rocks{helper.logintoken_set.first().get_absolute_url()}",
|
|
),
|
|
to=helper,
|
|
)
|
|
for helper in Helper.objects.filter(number_validated=True)
|
|
)
|
|
)
|
|
messages.add_message(
|
|
request, messages.SUCCESS, "Massen-Nachricht erfolgreich versendet"
|
|
)
|
|
except:
|
|
messages.add_message(
|
|
request,
|
|
messages.ERROR,
|
|
"Fehler beim Versenden der Massen-Nachricht",
|
|
)
|
|
|
|
context = {
|
|
"form": form,
|
|
}
|
|
return render(request, "bulk_message.html", context)
|
|
|
|
|
|
class HelperDetail(LoginRequiredMixin, DetailView):
|
|
template_name = "helper_detail.html"
|
|
model = Helper
|
|
|
|
|
|
class ShiftList(LoginRequiredMixin, ListView):
|
|
template_name = "shift_list.html"
|
|
model = Shift
|
|
title = "Alle Schichten"
|
|
|
|
def get_context_data(self, **kwargs):
|
|
context = super().get_context_data(**kwargs)
|
|
context["title"] = self.title
|
|
return context
|
|
|
|
def get_ordering(self):
|
|
return ("start_at", "room_id")
|
|
|
|
|
|
class FreeShiftList(ShiftList):
|
|
title = "Freie Schichten"
|
|
|
|
def get_queryset(self):
|
|
help_wanted = Q(required_helpers__gt=F("reg_count")) | Q(
|
|
required_helpers=0
|
|
) & Q(room__required_helpers__gt=F("reg_count"))
|
|
return (
|
|
Shift.with_reg_count()
|
|
.filter(
|
|
help_wanted,
|
|
start_at__gt=timezone.now(),
|
|
deleted = False,
|
|
)
|
|
.order_by("start_at", "room_id")
|
|
)
|