add lineup entry (analogous to positioning)

teamsnap/views.py

@@ -1,9 +1,19 @@
 from django.shortcuts import render, redirect
 
 # from .teamsnap.api import TeamSnap, Team, Event, Availability
-from .models import User, Member, Team, Event, Location
+from .models import User, Member, Team, Event, Location, LineupEntry
 from django.views.generic.list import ListView
 from lib.views import BenchcoachListView
+from .forms import LineupEntryForm, LineupEntryFormSet
+from django.forms.models import model_to_dict
+from django.urls import reverse
+from django.db.models import Case, When
+
+def queryset_from_ids(Model, id_list):
+    #https://stackoverflow.com/questions/4916851/django-get-a-queryset-from-array-of-ids-in-specific-order
+    preserved = Case(*[When(pk=pk, then=pos) for pos, pk in enumerate(id_list)])
+    queryset = Model.objects.filter(pk__in=id_list).order_by(preserved)
+    return queryset
 
 def edit_event(request, id):
     event = Event.objects.get(id = id)
@@ -17,19 +27,82 @@ class EventsListView(BenchcoachListView):
     title_strf = '{item.formatted_title}'
     body_strf = "{item.start_date:%a, %b %-d, %-I:%M %p},\n{item.location.name}"
 
-    # def get_context_data(self):
-    #     context = super().get_context_data()
-    #     for item in context['items']:
-    #         item['buttons'].append(
-    #             {
-    #                 'label': 'Edit Lineup',
-    #                 'href': reverse('edit lineup', args=[item['id']])
-    #             }
-    #         )
-    #     return context
+    def get_context_data(self):
+        context = super().get_context_data()
+        for item in context['items']:
+            item['buttons'].append(
+                {
+                    'label': 'Edit Lineup',
+                    'href': reverse('teamsnap edit lineup', args=[item['id']])
+                }
+            )
+        return context
 
-class TeamListView(ListView):
-    model = Team
+class TeamListView(BenchcoachListView):
+    Model = Team
+    edit_url = 'teamsnap edit team'
+    list_url = 'teamsnap list teams'
+    page_title = "TeamSnap Teams"
 
-class LocationListView(ListView):
-    model = Location
+class LocationListView(BenchcoachListView):
+    Model = Location
+    edit_url = 'teamsnap edit location'
+    list_url = 'teamsnap list locations'
+    page_title = "TeamSnap Locations"
+
+def edit_lineup(request, event_id):
+
+    if request.method == 'POST':
+        # create a form instance and populate it with data from the request:
+        formset = LineupEntryFormSet(request.POST)
+        for form in formset:
+            if form.is_valid():
+                # process the data in form.cleaned_data as required
+                # ...
+                # redirect to a new URL:
+
+                if isinstance(form.cleaned_data['id'], LineupEntry):
+                    positioning_id = form.cleaned_data.pop('id').id #FIXME this is a workaround, not sure why it is necessary
+                    positioning = LineupEntry.objects.filter(id=positioning_id)
+                    positioning.update(**form.cleaned_data)
+                    did_create = False
+                else:
+                    positioning = LineupEntry.objects.create(**form.cleaned_data, event_id=event_id)
+                    did_create = True
+            else:
+                pass
+        return render(request, 'success.html', {'call_back':'teamsnap edit lineup','id':event_id, 'errors':[error for error in formset.errors if error]}, status=200)
+            # return render(request, 'success.html', {'call_back':'schedule'})
+    event = Event.objects.get(id=event_id)
+    members = Member.objects.filter(is_non_player=False).prefetch_related('availability_set', 'lineupentry_set')
+    # players_d.sort(key=lambda d: (-d['availability'].available, d['last_name']))
+
+    for member in members:
+        LineupEntry.objects.get_or_create(member_id=member.id, event_id=event_id)
+
+    qs_starting_lineup = LineupEntry.objects.filter(event_id=event_id, sequence__isnull=False, sequence__gt=0).order_by('sequence')
+    qs_bench = LineupEntry.objects.filter(event_id=event_id, sequence=0).prefetch_related('member__availability_set').order_by('member__last_name')
+
+    # This is all a compromise to get the sorting just the way I wanted. THERE'S GOT TO BE A BETTER WAY
+    ids_starting_lineup = [item.id for item in qs_starting_lineup]
+    ids_bench_available = [item.id for item in qs_bench
+                           if item.member.availability_set.get(event_id=event_id).status_code == 1]
+    ids_bench_maybe = [item.id for item in qs_bench
+                           if item.member.availability_set.get(event_id=event_id).status_code == 2]
+    ids_bench_no = [item.id for item in qs_bench
+                       if item.member.availability_set.get(event_id=event_id).status_code == 0]
+    ids_bench_unknown = [item.id for item in qs_bench
+                    if item.member.availability_set.get(event_id=event_id).status_code is None]
+    qset = queryset_from_ids(LineupEntry, ids_starting_lineup + ids_bench_available + ids_bench_maybe + ids_bench_no + ids_bench_unknown)
+
+    formset = LineupEntryFormSet(queryset=qset)
+
+    for f in formset:
+        if f.instance.member_id:
+            f.availability = f.instance.member.availability_set.get(event_id=event_id)
+            # f.statline = f.instance.member.statline_set.get()
+
+    return render(request, 'teamsnap/lineup.html', {'title': 'Lineup',
+                                                   'event': event,
+                                                   'formset': formset,
+                                                   })