implement lineup send to gamechanger

gamechanger/views.py

@@ -1,11 +1,14 @@
+from django.http import HttpResponse, HttpResponseNotAllowed, HttpResponseServerError
 from django.shortcuts import render
 from django.views.generic.edit import FormView
-from django.views.generic.list import ListView
-from .forms import PreferencesForm, AccountForm, PlayerFormSet
-from .utils.gamechanger import scrape_page
+
 from teamsnap.views import get_teamsnap_client
 
-# Create your views here.
+from .forms import AccountForm, PlayerFormSet, PreferencesForm
+from .models import Player
+from .utils import gamechanger
+
+
 class PreferencesFormView(FormView):
     template_name = "gamechanger/form.html"
     form_class = PreferencesForm
@@ -29,6 +32,7 @@ class PreferencesFormView(FormView):
 
         return initial
 
+
 class AccountFormView(FormView):
     template_name = "gamechanger/form.html"
     form_class = AccountForm
@@ -52,30 +56,119 @@ class AccountFormView(FormView):
 
         return initial
 
-def roster_view(request):
-    from pyteamsnap.api import Member
-    client = get_teamsnap_client(request)
-    season_id = request.user.gamechanger_preferences.season_id
-    team_id = request.user.gamechanger_preferences.team_id
-    teamsnap_team_id = request.user.preferences.managed_team_id
-    teamsnap_members = {
-        f"{member.data['first_name']} {member.data['last_name']}":member
-        for member in Member.search(client, team_id=teamsnap_team_id)
-    }
 
-    page = "roster"
+def roster_import(request):
+    if request.method == "GET":
+        from pyteamsnap.api import Member
 
-    d = scrape_page(season_id, team_id, page)
-    roster = d['roster']
-    initial = [{
-        'gamechanger_name':f"{player['fname']} {player['lname']}",
-        'teamsnap_name':"{first_name} {last_name}".format(**teamsnap_members[f"{player['fname']} {player['lname']}"].data),
-        'gamechanger_id':player.get('player_id'),
-        'teamsnap_member_id':teamsnap_members[f"{player['fname']} {player['lname']}"].data['id'],
-    } for player in roster]
-    formset = PlayerFormSet(initial=initial)
-    return render(request, "gamechanger/roster.html", context={'roster':roster, 'formset':formset})
-    pass
+        client = get_teamsnap_client(request)
+        season_id = request.user.gamechanger_preferences.season_id
+        team_id = request.user.gamechanger_preferences.team_id
+        teamsnap_team_id = request.user.teamsnap_preferences.managed_team_id
+        teamsnap_members = {
+            f"{member.data['first_name']} {member.data['last_name']}": member
+            for member in Member.search(client, team_id=teamsnap_team_id)
+        }
 
-def roster_save(request):
-    pass
+        page = "roster"
+
+        d = gamechanger.scrape_page(season_id, team_id, page)
+        roster = d["roster"]
+        initial = [
+            {
+                "gamechanger_name": f"{player['fname']} {player['lname']}",
+                "fname": player["fname"],
+                "lname": player["lname"],
+                "teamsnap_name": "{first_name} {last_name}".format(
+                    **teamsnap_members[f"{player['fname']} {player['lname']}"].data
+                ),
+                "id": player.get("player_id"),
+                "teamsnap_member_id": teamsnap_members[
+                    f"{player['fname']} {player['lname']}"
+                ].data["id"],
+            }
+            for player in roster
+        ]
+        formset = PlayerFormSet(initial=initial)
+        choices = [
+            (
+                teamsnap_member.data["id"],
+                f"{teamsnap_member.data['first_name']} {teamsnap_member.data['last_name']}",
+            )
+            for teamsnap_member in teamsnap_members.values()
+        ]
+        for form in formset:
+            form.fields["teamsnap_member_id"].widget.choices = choices
+            pass
+        return render(
+            request,
+            "gamechanger/roster.html",
+            context={"roster": roster, "formset": formset},
+        )
+    elif request.POST:
+        formset = PlayerFormSet(request.POST)
+        if formset.is_valid():
+            r = []
+            for form in formset:
+                data = form.cleaned_data
+                data.pop("DELETE")
+                data.pop("gamechanger_name")
+                data.pop("teamsnap_name")
+                obj, did_create = Player.objects.update_or_create(**data)
+                obj.save()
+                r.append(obj)
+            return HttpResponse(status=200)
+        else:
+            return HttpResponseServerError()
+
+        return HttpResponseServerError()
+    else:
+        return HttpResponseServerError()
+
+
+def lineup_submit(request):
+
+    from teamsnap.forms import LineupEntryFormset
+
+    if request.GET:
+        return HttpResponseNotAllowed()
+    if request.POST:
+        formset = LineupEntryFormset(request.POST)
+        if formset.is_valid():
+            lineup_data = [
+                form.cleaned_data
+                for form in formset
+                if form.cleaned_data.get("label")
+                and form.cleaned_data.get("sequence") is not None
+            ]
+            lineup_data.sort(key=lambda x: x.get("sequence"))
+            lineup = []
+
+            for lineup_entry in lineup_data:
+                d = {
+                    "player_id": lineup_entry["gamechanger_player_id"],
+                    "position": lineup_entry["label"],
+                }
+                if lineup_entry["label"] == "DH":
+                    for_whom = [
+                        e
+                        for e in lineup_data
+                        if e["position_only"] and e["label"] != "DR"
+                    ][0]
+                    d["forwhom"] = for_whom["gamechanger_player_id"]
+                    lineup.append(d)
+                    if for_whom in lineup:
+                        lineup_data.remove(for_whom)
+                    if for_whom in lineup_data:
+                        lineup_data.remove(for_whom)
+                    lineup.append(
+                        {
+                            "player_id": for_whom["gamechanger_player_id"],
+                            "position": for_whom["label"],
+                        }
+                    )
+                elif lineup_entry["label"] != "DR":
+                    lineup.append(d)
+            gamechanger.submit_lineup(request, lineup)
+            return HttpResponse(status=200)
+    return HttpResponseServerError()