257 lines
9.7 KiB
Python
257 lines
9.7 KiB
Python
from django.shortcuts import render, redirect
|
|
|
|
from .teamsnap.api import Event as TsApiEvent
|
|
from .teamsnap.api import TeamSnap
|
|
from .models import User, Member, Team, Event, Location, LineupEntry, Opponent, Availability
|
|
from lib.views import BenchcoachListView
|
|
from .forms import EventForm, MemberForm, OpponentForm, TeamForm, LocationForm
|
|
from django.urls import reverse
|
|
from django.db.models import Case, When
|
|
from django.views import View
|
|
from django.http import HttpResponse
|
|
import benchcoachproject.models
|
|
import benchcoach.models
|
|
from django.http import JsonResponse
|
|
from .utils.teamsnap_object_utils import update_users, update_teams, update_events, update_members, update_locations, update_availabilities
|
|
from django.contrib import messages
|
|
from django.template.loader import render_to_string
|
|
from .utils.import_teamsnap import update_team, update_event, update_member, update_location, update_opponent, update_availability, update_teamsnap_object
|
|
from django.forms import modelformset_factory
|
|
import django.db.models
|
|
from django.contrib.auth.decorators import login_required
|
|
|
|
|
|
|
|
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)
|
|
return redirect(event.edit_url)
|
|
|
|
def home(request):
|
|
current_benchcoach_user = request.user
|
|
current_teamsnap_user = request.user.profile.teamsnap_user
|
|
current_teamsnap_team = request.user.profile.teamsnapsettings.managed_team
|
|
teamsnap_objects = {}
|
|
for obj in [Availability, Event, LineupEntry, Location, Member, Opponent, Team, User]:
|
|
teamsnap_objects[obj.__name__.lower()] = {}
|
|
teamsnap_objects[obj.__name__.lower()]['object_count']=obj.objects.count()
|
|
|
|
context= {
|
|
'benchcoach_user': current_benchcoach_user,
|
|
'teamsnap_user': current_teamsnap_user,
|
|
'teamsnap_team':current_teamsnap_team,
|
|
'teamsnap_objects': teamsnap_objects
|
|
}
|
|
return render(request, 'teamsnap/home.html', context)
|
|
|
|
class TeamsnapObjTableView(View):
|
|
Model = None
|
|
Form = None
|
|
template = 'teamsnap/table.html'
|
|
options = {
|
|
'event': (Event, EventForm),
|
|
'location': (Location, LocationForm),
|
|
'member': (Member, MemberForm),
|
|
'opponent': (Opponent, OpponentForm),
|
|
'team': (Team, TeamForm)
|
|
}
|
|
|
|
def post(self, request, object):
|
|
self.Model, self.Form = self.options[object]
|
|
self.Formset = modelformset_factory(
|
|
model=self.Model,
|
|
form=self.Form,
|
|
extra=0
|
|
)
|
|
formset = self.Formset(request.POST, request.FILES)
|
|
if formset.is_valid():
|
|
formset.save()
|
|
return HttpResponse(200)
|
|
else:
|
|
return HttpResponse(422)
|
|
pass
|
|
|
|
def get(self, request, object):
|
|
self.Model, self.Form = self.options[object]
|
|
self.Formset = modelformset_factory(
|
|
model=self.Model,
|
|
form=self.Form,
|
|
extra=0
|
|
)
|
|
qs = self.Model.objects.all()
|
|
formset = self.Formset(queryset=qs)
|
|
return render(request, self.template, context={'formset': formset})
|
|
|
|
@login_required()
|
|
def update_teamsnapdb_from_teamsnapapi(request, object_name, object_id=None):
|
|
TOKEN = request.user.profile.teamsnap_access_token
|
|
USER_ID = request.user.profile.teamsnap_user.id
|
|
TEAM_ID = request.user.profile.teamsnapsettings.managed_team.id
|
|
CLIENT = TeamSnap(token=TOKEN)
|
|
|
|
Object = {
|
|
obj.__name__.lower():obj
|
|
for obj in
|
|
[Availability, Event, LineupEntry, Location, Member, Opponent, Team, User]
|
|
}.get(object_name)
|
|
|
|
r = {}
|
|
|
|
for Obj in [Object]:
|
|
r[Obj.__name__.lower()] = []
|
|
a = Obj.ApiObject.search(CLIENT, team_id=TEAM_ID)
|
|
for _a in a:
|
|
obj, created = Obj.update_or_create_from_teamsnap_api(_a.data)
|
|
r[Obj.__name__.lower()].append((obj, created))
|
|
|
|
for object_name, results in r.items():
|
|
if len(r) == 0:
|
|
messages.error(request, f"Error! No {object_name} objects created or updated")
|
|
else:
|
|
result = [created for obj, created in results]
|
|
messages.success(request,
|
|
f"Success! {sum(result)} {object_name} objects created, {len(result) - sum(result)} {object_name} objects updated.")
|
|
|
|
return redirect('teamsnap home')
|
|
|
|
@login_required()
|
|
def send_to_benchcoach(request, object_name):
|
|
Object = {
|
|
obj.__name__.lower(): obj
|
|
for obj in
|
|
[Availability, Event, LineupEntry, Location, Member, Opponent, Team, User]
|
|
}.get(object_name)
|
|
|
|
TEAM_ID = request.user.profile.teamsnapsettings.managed_team.id
|
|
r = {}
|
|
|
|
r[object_name]=[]
|
|
|
|
if object_name == 'team':
|
|
for team in Object.objects.filter(id=TEAM_ID):
|
|
r[object_name] += update_opponent(team, create_benchcoach_object=True, create_related=True)
|
|
|
|
if object_name == 'opponent':
|
|
for team in Object.objects.filter(team_id=TEAM_ID):
|
|
r[object_name] += update_team(team, create_benchcoach_object=True, create_related=True)
|
|
|
|
if object_name == 'location':
|
|
for location in Location.objects.filter(team_id=TEAM_ID):
|
|
r[object_name] += update_location(location, create_benchcoach_object=True, create_related=True)
|
|
|
|
if object_name == 'member':
|
|
for member in Member.objects.filter(team_id=TEAM_ID, is_non_player=False):
|
|
r[object_name] += update_member(member, create_benchcoach_object=True, create_related=True)
|
|
|
|
if object_name == 'event':
|
|
for event in Event.objects.filter(team_id=TEAM_ID):
|
|
r[object_name] += update_event(event, create_benchcoach_object=True, create_related=True)
|
|
|
|
if object_name == 'availability':
|
|
for availability in Availability.objects.filter(team_id=TEAM_ID, member__is_non_player=False):
|
|
r[object_name] += update_availability(availability, create_benchcoach_object=True, create_related=True)
|
|
|
|
for object_name, results in r.items():
|
|
if len(r) == 0:
|
|
messages.error(request, f"Error! No {object_name} objects created or updated")
|
|
else:
|
|
result = [created for obj, created in results]
|
|
messages.success(request,
|
|
f"Success! {sum(result)} {object_name} objects created, {len(result) - sum(result)} {object_name} objects updated.")
|
|
|
|
return redirect('teamsnap home')
|
|
|
|
def sync_teamsnapdb_with_teamsnapapi(request):
|
|
'''
|
|
This sync the internal TeamSnap Database with the TeamSnap API
|
|
'''
|
|
TOKEN = request.user.profile.teamsnap_access_token
|
|
USER_ID = request.user.profile.teamsnap_user.id
|
|
TEAM_ID = request.user.profile.teamsnapsettings.managed_team.id
|
|
CLIENT = TeamSnap(token=TOKEN)
|
|
|
|
r = {}
|
|
|
|
for Obj in [User]:
|
|
r[Obj.__name__] = []
|
|
a = Obj.ApiObject.search(CLIENT, id=USER_ID)
|
|
for _a in a:
|
|
obj, created = Obj.update_or_create_from_teamsnap_api(_a.data)
|
|
r[Obj.__name__].append((obj, created))
|
|
|
|
for Obj in [Event, Availability]:
|
|
r[Obj.__name__] = []
|
|
a = Obj.ApiObject.search(CLIENT, team_id=TEAM_ID)
|
|
for _a in a:
|
|
print(f"importing {_a}")
|
|
obj, created = Obj.update_or_create_from_teamsnap_api(_a.data)
|
|
r[Obj.__name__].append((obj, created))
|
|
|
|
|
|
for object_name, results in r.items():
|
|
if len(r) == 0:
|
|
messages.error(request, f"Error! No {object_name} objects created or updated")
|
|
else:
|
|
result = [created for obj, created in results]
|
|
messages.success(request, f"Success! {sum(result)} {object_name} objects created, {len(result)-sum(result)} {object_name} objects updated.")
|
|
|
|
data = {
|
|
'msg': render_to_string('messages.html', {}, request),
|
|
}
|
|
|
|
return JsonResponse(data)
|
|
|
|
def sync_teamsnapdb_to_benchcoachdb(request):
|
|
'''
|
|
This syncs the internal BenchCoach Database and the TeamSnap Database
|
|
'''
|
|
TEAM_ID = request.user.profile.teamsnapsettings.managed_team.id
|
|
r={}
|
|
|
|
r['team/opponent'] = []
|
|
for team in Opponent.objects.filter(team_id=TEAM_ID):
|
|
r['team/opponent'] += update_opponent(team, create_benchcoach_object=True, create_related=True)
|
|
|
|
for team in Team.objects.filter(id=TEAM_ID):
|
|
r['team/opponent'] += update_team(team, create_benchcoach_object=True, create_related=True)
|
|
|
|
r['location'] = []
|
|
for location in Location.objects.filter(team_id=TEAM_ID):
|
|
r['location'] += update_location(location, create_benchcoach_object=True, create_related=True)
|
|
|
|
r['member'] = []
|
|
for member in Member.objects.filter(team_id=TEAM_ID, is_non_player=False):
|
|
r['member'] += update_member(member, create_benchcoach_object=True, create_related=True)
|
|
|
|
r['event'] = []
|
|
for event in Event.objects.filter(team_id=TEAM_ID):
|
|
r['event'] += update_event(event, create_benchcoach_object=True, create_related=True)
|
|
|
|
r['availability'] = []
|
|
for availability in Availability.objects.filter(team_id=TEAM_ID, member__is_non_player=False):
|
|
r['availability'] += update_availability(availability, create_benchcoach_object=True, create_related=True)
|
|
pass
|
|
|
|
for object_name, results in r.items():
|
|
if len(r) == 0:
|
|
messages.error(request, f"Error! No {object_name} objects created or updated")
|
|
else:
|
|
result = [created for obj, created in results]
|
|
messages.success(request, f"Success! {sum(result)} {object_name} objects created, {len(result)-sum(result)} {object_name} objects updated.")
|
|
|
|
data = {
|
|
'msg': render_to_string('messages.html', {}, request),
|
|
}
|
|
return JsonResponse(data)
|
|
|
|
|
|
|
|
|
|
|
|
|