chore(fallback): mark team-only shifts as such in fallback shifts view

shiftregister/fallback/forms.py

@@ -0,0 +1,10 @@
+from django import forms
+
+
+class TradeForm(forms.Form):
+    assignment_id = forms.IntegerField(
+        widget=forms.NumberInput(attrs={"class": "input", "placeholder": "Schicht-ID"})
+    )
+    pin = forms.IntegerField(
+        widget=forms.PasswordInput(attrs={"class": "input", "placeholder": "Deine PIN"})
+    )

shiftregister/fallback/templates/my_fallback_shifts.html

@@ -75,17 +75,12 @@ Diese Schichtzuteilung wurde maschinell erstellt und ist auch ohne Unterschrift
             <td>{{ shift.room.name }} </td>
             <td>{{ shift.registration_count }}/{{ shift.required_helpers|default:shift.room.required_helpers }}</td>
             <td>
-{% for fa in shift.fallbackassignment_set.all %}
-    {% if fa.traded_to %}
-        {{ fa.traded_to.name }}
-    {% else %}
-        {{ fa.team_member.name }}
-    {% endif %}
-    {% if not forloop.last %}, {% endif %}
-{% endfor %}
+{% for fa in shift.fallbackassignment_set.all %}{% if fa.traded_to %}{{ fa.traded_to.name }}{% else %}{{ fa.team_member.name }}{% endif %}{% if not forloop.last %}, {% endif %}{% endfor %}
             </td>
             <td>
-{% if shift.registration_count < shift.required_helpers|default:shift.room.required_helpers %}
+{% if shift.restricted %}
+                <button class="button is-primary is-small mr-0" disabled>Nur Team</button>
+{% elif shift.registration_count < shift.required_helpers|default:shift.room.required_helpers %}
                 <a class="button is-primary is-small mr-0" href="{% url 'shift' shift.id %}">Mithelfen</a>
 {% endif %}
             </td>

shiftregister/fallback/views.py

@@ -1,23 +1,15 @@
 from base64 import urlsafe_b64decode
 
-from django import forms
 from django.contrib import messages
 from django.contrib.auth.decorators import login_required
 from django.db.models import Count, Q
+from django.db.models.functions import Coalesce
 from django.http import HttpResponse
 from django.shortcuts import get_object_or_404, redirect, render
 from django.urls import reverse
 
-from shiftregister.fallback.models import FallbackAssignment, TeamMember
-
-
-class TradeForm(forms.Form):
-    assignment_id = forms.IntegerField(
-        widget=forms.NumberInput(attrs={"class": "input", "placeholder": "Schicht-ID"})
-    )
-    pin = forms.IntegerField(
-        widget=forms.PasswordInput(attrs={"class": "input", "placeholder": "Deine PIN"})
-    )
+from .forms import TradeForm
+from .models import FallbackAssignment, TeamMember
 
 
 def my_fallback_shifts(request, team_member_id):
@@ -70,11 +62,14 @@ def my_fallback_shifts(request, team_member_id):
                     messages.error(request, "Ungültige Schicht-ID")
 
     assignments = (
-        FallbackAssignment.objects.filter(
+        FallbackAssignment.objects.select_related(
+            "shift__event__calendar", "traded_to", "team_member"
+        )
+        .annotate(restricted=Coalesce("shift__event__calendar__restricted", False))
+        .filter(
             Q(team_member=team_member, traded_to__isnull=True)
             | Q(traded_to=team_member)
         )
-        .prefetch_related("shift", "traded_to", "team_member")
         .order_by("shift__start_at")
     )