add opponent views and forms
This commit is contained in:
@@ -6,14 +6,13 @@ from allauth.socialaccount.providers.oauth2.views import (
|
||||
OAuth2CallbackView,
|
||||
OAuth2LoginView,
|
||||
)
|
||||
from django.http import HttpResponse, HttpResponseNotAllowed, HttpResponseServerError
|
||||
from django.shortcuts import redirect, render
|
||||
from django.http import HttpResponse
|
||||
from django.shortcuts import redirect, render, reverse
|
||||
from django.views.generic import ListView
|
||||
from django.views.generic.edit import FormView
|
||||
|
||||
from gamechanger.models import Player as GamechangerPlayer
|
||||
|
||||
from .forms import PreferencesForm
|
||||
from .models import Preferences
|
||||
from .forms import OpponentForm, PreferencesForm
|
||||
from .models import Opponent, Preferences
|
||||
from .provider import TeamsnapProvider
|
||||
from .utils import get_teamsnap_client
|
||||
|
||||
@@ -60,7 +59,9 @@ class PreferencesFormView(FormView):
|
||||
# It should return an HttpResponse.
|
||||
if form.data["user"] == str(self.request.user.id):
|
||||
form.save()
|
||||
return super().form_valid(form)
|
||||
return super().form_valid(form)
|
||||
else:
|
||||
return super().form_invalid(form)
|
||||
|
||||
def get_initial(self):
|
||||
"""
|
||||
@@ -180,238 +181,6 @@ def view_event(request, event_id, team_id=None):
|
||||
)
|
||||
|
||||
|
||||
def edit_lineup(request, event_ids, team_id):
|
||||
import re
|
||||
|
||||
from pyteamsnap.api import (
|
||||
Availability,
|
||||
AvailabilitySummary,
|
||||
Event,
|
||||
EventLineup,
|
||||
EventLineupEntry,
|
||||
Member,
|
||||
)
|
||||
|
||||
from teamsnap.forms import LineupEntryFormset
|
||||
|
||||
client = get_teamsnap_client(request)
|
||||
|
||||
event_ids = str(event_ids).split(",")
|
||||
|
||||
ts_bulkload = client.bulk_load(
|
||||
team_id=team_id,
|
||||
types=[Event, EventLineup, EventLineupEntry, AvailabilitySummary, Member],
|
||||
event__id=",".join(event_ids),
|
||||
)
|
||||
event_ids = [int(i) for i in event_ids]
|
||||
|
||||
contexts = []
|
||||
for event_id in event_ids:
|
||||
ts_event = [
|
||||
i for i in ts_bulkload if isinstance(i, Event) and i.data["id"] == event_id
|
||||
][0]
|
||||
ts_availabilities = Availability.search(client, event_id=ts_event.data["id"])
|
||||
ts_lineup_entries = EventLineupEntry.search(client, event_id=event_id)
|
||||
|
||||
ts_members = [i for i in ts_bulkload if isinstance(i, Member)]
|
||||
ts_member_lookup = {m.data["id"]: m for m in ts_members}
|
||||
gc_player_lookup = {
|
||||
m.data["id"]: getattr(
|
||||
GamechangerPlayer.objects.filter(
|
||||
teamsnap_member_id=m.data["id"]
|
||||
).first(),
|
||||
"id",
|
||||
None,
|
||||
)
|
||||
for m in ts_members
|
||||
}
|
||||
ts_availability_lookup = {m.data["member_id"]: m for m in ts_availabilities}
|
||||
ts_lineup_entries_lookup = {m.data["member_id"]: m for m in ts_lineup_entries}
|
||||
|
||||
members = []
|
||||
|
||||
for lineup_entry in ts_lineup_entries:
|
||||
members.append(
|
||||
{
|
||||
"member": getattr(
|
||||
ts_member_lookup[lineup_entry.data["member_id"]], "data"
|
||||
),
|
||||
"availability": getattr(
|
||||
ts_availability_lookup.get(lineup_entry.data["member_id"], {}),
|
||||
"data",
|
||||
{},
|
||||
),
|
||||
"lineup_entry": getattr(lineup_entry, "data", {}),
|
||||
}
|
||||
)
|
||||
|
||||
in_lineup_already = [m["member"] for m in members]
|
||||
|
||||
for member in ts_members:
|
||||
if member.data not in in_lineup_already:
|
||||
members.append(
|
||||
{
|
||||
"member": getattr(member, "data"),
|
||||
"availability": getattr(
|
||||
ts_availability_lookup.get(member.data["id"], {}),
|
||||
"data",
|
||||
{},
|
||||
),
|
||||
"lineup_entry": getattr(
|
||||
ts_lineup_entries_lookup.get(member.data["id"], {}),
|
||||
"data",
|
||||
{},
|
||||
),
|
||||
}
|
||||
)
|
||||
|
||||
members = sorted(
|
||||
members,
|
||||
key=lambda d: (
|
||||
{None: 3, 0: 2, 2: 1, 1: 0}.get( # No Response # No # Maybe # Yes
|
||||
d["availability"].get("status_code")
|
||||
),
|
||||
d["member"].get("last_name"),
|
||||
),
|
||||
)
|
||||
|
||||
initial = []
|
||||
for member in members:
|
||||
if not member["member"]["is_non_player"]:
|
||||
if re.search(
|
||||
r"([A-Z0-9]+)(?:\s+\[(.*)\])?",
|
||||
member["lineup_entry"].get("label", ""),
|
||||
):
|
||||
position, position_note = re.search(
|
||||
r"([A-Z0-9]+)(?:\s+\[(.*)\])?",
|
||||
member["lineup_entry"].get("label", ""),
|
||||
).groups()
|
||||
else:
|
||||
position, position_note = ("", "")
|
||||
position_only = position_note == "PO"
|
||||
initial.append(
|
||||
{
|
||||
"event_lineup_entry_id": member["lineup_entry"].get("id"),
|
||||
"event_lineup_id": member["lineup_entry"].get(
|
||||
"event_lineup_id"
|
||||
),
|
||||
"event_id": event_id,
|
||||
"position_only": position_only,
|
||||
"member_id": member["member"]["id"],
|
||||
"sequence": member["lineup_entry"].get("sequence"),
|
||||
"label": position,
|
||||
"gamechanger_player_id": gc_player_lookup.get(
|
||||
member["member"]["id"]
|
||||
),
|
||||
}
|
||||
)
|
||||
|
||||
formset = LineupEntryFormset(initial=initial)
|
||||
|
||||
for form in formset:
|
||||
form.member = ts_member_lookup.get(form["member_id"].initial)
|
||||
form.availability = ts_availability_lookup.get(form["member_id"].initial)
|
||||
|
||||
formset_startinglineup = [
|
||||
form
|
||||
for form in formset
|
||||
if form.initial.get("event_lineup_entry_id")
|
||||
and not form.initial.get("position_only")
|
||||
]
|
||||
formset_startinglineup = sorted(
|
||||
formset_startinglineup, key=lambda d: d.initial.get("sequence", 100)
|
||||
)
|
||||
formset_startingpositiononly = [
|
||||
form
|
||||
for form in formset
|
||||
if form.initial.get("event_lineup_entry_id")
|
||||
and form not in formset_startinglineup
|
||||
]
|
||||
formset_startingpositiononly = sorted(
|
||||
formset_startingpositiononly, key=lambda d: d.initial.get("sequence", 100)
|
||||
)
|
||||
formset_bench = [
|
||||
form
|
||||
for form in formset
|
||||
if form not in formset_startinglineup
|
||||
and form not in formset_startingpositiononly
|
||||
and form.availability.data["status_code"] in [2, 1]
|
||||
]
|
||||
formset_out = [
|
||||
form
|
||||
for form in formset
|
||||
if form not in formset_startinglineup
|
||||
and form not in formset_bench
|
||||
and form not in formset_startingpositiononly
|
||||
and not form.member.data["is_non_player"]
|
||||
]
|
||||
|
||||
contexts.append(
|
||||
{
|
||||
"event": ts_event,
|
||||
"formset": formset,
|
||||
"formset_bench": formset_bench,
|
||||
"formset_startinglineup": formset_startinglineup,
|
||||
"formset_startingpositionalonly": formset_startingpositiononly,
|
||||
"formset_out": formset_out,
|
||||
}
|
||||
)
|
||||
|
||||
return render(request, "lineup/edit.html", context={"contexts": contexts})
|
||||
|
||||
|
||||
def submit_lineup(request, team_id, event_id):
|
||||
from pyteamsnap.api import EventLineup, EventLineupEntry
|
||||
|
||||
from teamsnap.forms import LineupEntryFormset
|
||||
|
||||
client = get_teamsnap_client(request)
|
||||
ts_lineup = EventLineup.search(client, event_id=event_id)
|
||||
event_lineup_id = ts_lineup[0].data["id"]
|
||||
if request.GET:
|
||||
return HttpResponseNotAllowed()
|
||||
if request.POST:
|
||||
formset = LineupEntryFormset(request.POST)
|
||||
if formset.is_valid():
|
||||
r = []
|
||||
for form in formset:
|
||||
data = form.cleaned_data
|
||||
if data.get("event_lineup_entry_id"):
|
||||
event_lineup_entry = EventLineupEntry.get(
|
||||
client, id=data.get("event_lineup_entry_id")
|
||||
)
|
||||
if data.get("position_only"):
|
||||
data["label"] = data["label"] + " [PO]"
|
||||
event_lineup_entry.data.update(data)
|
||||
if not data.get("sequence") and not data.get("label"):
|
||||
try:
|
||||
r.append(event_lineup_entry.delete())
|
||||
except Exception as e:
|
||||
raise e
|
||||
else:
|
||||
try:
|
||||
r.append(event_lineup_entry.put())
|
||||
except Exception as e:
|
||||
raise e
|
||||
pass
|
||||
elif data.get("sequence") is not None and data.get("label"):
|
||||
event_lineup_entry = EventLineupEntry.new(client)
|
||||
if data.get("position_only"):
|
||||
data["label"] = data["label"] + " [PO]"
|
||||
event_lineup_entry.data.update(data)
|
||||
event_lineup_entry.data.update({"event_lineup_id": event_lineup_id})
|
||||
try:
|
||||
r.append(event_lineup_entry.post())
|
||||
except Exception as e:
|
||||
raise e
|
||||
else:
|
||||
pass
|
||||
else:
|
||||
pass
|
||||
return HttpResponse(status=200)
|
||||
return HttpResponseServerError()
|
||||
|
||||
|
||||
def multi_lineup_choose(request, team_id):
|
||||
from django.forms import formset_factory
|
||||
from pyteamsnap.api import Event
|
||||
@@ -464,3 +233,86 @@ def multi_lineup_choose(request, team_id):
|
||||
"lineup/multiple_choose.html",
|
||||
context={"formset": formset, "team_id": team_id},
|
||||
)
|
||||
|
||||
|
||||
class OpponentFormView(FormView):
|
||||
pk_url_kwarg = "opponent_id"
|
||||
template_name = "opponent.html"
|
||||
form_class = OpponentForm
|
||||
|
||||
def get_success_url(self):
|
||||
team_id = self.kwargs.get("team_id")
|
||||
return reverse("teamsnap_opponents", kwargs={"team_id": team_id})
|
||||
|
||||
def form_valid(self, form):
|
||||
# This method is called when valid form data has been POSTed.
|
||||
# It should return an HttpResponse.
|
||||
import pyteamsnap
|
||||
|
||||
ts_client = get_teamsnap_client(self.request)
|
||||
user = pyteamsnap.api.Me(ts_client)
|
||||
team_id = int(self.kwargs.get("team_id"))
|
||||
opponent_id = int(self.kwargs.get("opponent_id"))
|
||||
opponent = pyteamsnap.api.Opponent.get(ts_client, opponent_id)
|
||||
if (
|
||||
team_id in user.data["managed_team_ids"]
|
||||
and opponent.data["team_id"] == team_id
|
||||
):
|
||||
form.save()
|
||||
return super().form_valid(form)
|
||||
else:
|
||||
return super().form_invalid(form)
|
||||
|
||||
def get_form(self):
|
||||
"""
|
||||
Returns the initial data to use for forms on this view.
|
||||
"""
|
||||
|
||||
try:
|
||||
opponent, _ = Opponent.objects.get_or_create(
|
||||
id=self.kwargs.get("opponent_id")
|
||||
)
|
||||
form = OpponentForm(instance=opponent, **self.get_form_kwargs())
|
||||
|
||||
except Opponent.DoesNotExist:
|
||||
form = super().get_form(self.form_class)
|
||||
# form.widgets['id'] = forms.HiddenInput()
|
||||
|
||||
return form
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
import pyteamsnap
|
||||
|
||||
ts_client = get_teamsnap_client(self.request)
|
||||
context = super().get_context_data(**kwargs)
|
||||
opponent = pyteamsnap.api.Opponent.get(
|
||||
ts_client, self.kwargs.get("opponent_id")
|
||||
)
|
||||
context["opponent"] = opponent
|
||||
return context
|
||||
|
||||
|
||||
class Opponents(ListView):
|
||||
model = Opponent
|
||||
template_name = "opponents.html"
|
||||
pass
|
||||
|
||||
def get_queryset(self):
|
||||
import pyteamsnap
|
||||
|
||||
ts_client = get_teamsnap_client(self.request)
|
||||
team_id = self.kwargs.get("team_id")
|
||||
ts_opponents = pyteamsnap.api.Opponent.search(ts_client, team_id=team_id)
|
||||
ts_opponents.sort(key=lambda t: t.data.get("name"))
|
||||
opponents = []
|
||||
for opponent in ts_opponents:
|
||||
o, _ = Opponent.objects.get_or_create(id=opponent.data["id"])
|
||||
opponents.append((opponent, o))
|
||||
|
||||
return opponents
|
||||
|
||||
def get_context_data(self, **kwargs):
|
||||
team_id = self.kwargs.get("team_id")
|
||||
context = super().get_context_data(**kwargs)
|
||||
context["team_id"] = team_id
|
||||
return context
|
||||
|
||||
Reference in New Issue
Block a user