cleanup and added comments
This commit is contained in:
@@ -1,33 +1,19 @@
|
||||
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 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
|
||||
from .utils.teamsnap_sync_engine import TeamsnapSyncEngine
|
||||
|
||||
def edit_event(request, id):
|
||||
'''
|
||||
redirect to teamsnap.com page for editing of event.
|
||||
:param request:
|
||||
:param id:
|
||||
:return:
|
||||
'''
|
||||
event = Event.objects.get(id = id)
|
||||
return redirect(event.edit_url)
|
||||
|
||||
@@ -61,121 +47,8 @@ def home(request):
|
||||
}
|
||||
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 sync(request):
|
||||
object_name = request.POST.get('object_name')
|
||||
object_id = request.POST.get('object_id')
|
||||
if request.POST and request.POST.get('object_name') and request.POST.get('object_id'):
|
||||
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_name = request.POST.get('object_name')
|
||||
object_id = request.POST.get('object_id')
|
||||
Object = {
|
||||
obj.__name__.lower(): obj
|
||||
for obj in
|
||||
[Availability, Event, LineupEntry, Location, Member, Opponent, Team]
|
||||
}.get(object_name)
|
||||
|
||||
r={}
|
||||
r[Object.__name__.lower()] = []
|
||||
a = Object.ApiObject.search(CLIENT, id=object_id)
|
||||
if a and len(a) == 1:
|
||||
obj, created = Object.update_or_create_from_teamsnap_api(a[0].data)
|
||||
r[Object.__name__.lower()].append((obj, created))
|
||||
|
||||
for object_name, results in r.items():
|
||||
if len(results) == 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(request.POST.get('next'))
|
||||
|
||||
return HttpResponse('404')
|
||||
|
||||
|
||||
@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()] = []
|
||||
if not object_id:
|
||||
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))
|
||||
else:
|
||||
a = Obj.ApiObject.search(CLIENT, id=object_id)[0]
|
||||
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(results) == 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=None, object_id=None): #TODO rename
|
||||
def sync_from_teamsnap(request, object_name=None, object_id=None):
|
||||
if request.POST:
|
||||
next = request.POST.get('next')
|
||||
object_name = request.POST.get('object_name')
|
||||
@@ -238,90 +111,7 @@ def send_to_benchcoach(request, object_name=None, object_id=None): #TODO rename
|
||||
else:
|
||||
return HttpResponse(404)
|
||||
|
||||
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)
|
||||
|
||||
from .utils.teamsnap_sync_engine import TeamsnapSyncEngine
|
||||
def import_teamsnap(request):
|
||||
TEAM_ID = request.user.profile.teamsnapsettings.managed_team.id
|
||||
TOKEN = request.user.profile.teamsnap_access_token
|
||||
|
||||
Reference in New Issue
Block a user