revamp teamsnap syncing
This commit is contained in:
@@ -1,354 +0,0 @@
|
||||
import django.db.models
|
||||
from typing import List, Tuple
|
||||
from benchcoach.models import Availability, Player, Team, Positioning, Event, Venue
|
||||
from teamsnap.teamsnap.api import TeamSnap
|
||||
import teamsnap.models
|
||||
|
||||
from benchcoach.utils.sync_engine import AbstractSyncEngine
|
||||
|
||||
class TeamsnapSyncEngine(AbstractSyncEngine):
|
||||
models = [
|
||||
Availability,
|
||||
Player,
|
||||
Team,
|
||||
# Positioning, # Not Implemented
|
||||
Event,
|
||||
Venue
|
||||
]
|
||||
|
||||
def __init__(self, managed_team_teamsnap_id, teamsnap_token):
|
||||
self.managed_teamsnap_team_id = managed_team_teamsnap_id
|
||||
self.client = TeamSnap(token=teamsnap_token)
|
||||
|
||||
def _update_teamsnapdb_from_teamsnapapi(self, teamsnap_instance):
|
||||
ApiObject = {
|
||||
teamsnap.models.Availability:teamsnap.teamsnap.api.Availability,
|
||||
teamsnap.models.Member:teamsnap.teamsnap.api.Member,
|
||||
teamsnap.models.Team:teamsnap.teamsnap.api.Team,
|
||||
teamsnap.models.Opponent:teamsnap.teamsnap.api.Opponent,
|
||||
# teamsnap.models.LineupEntry # Not Implemented
|
||||
teamsnap.models.Event:teamsnap.teamsnap.api.Event,
|
||||
teamsnap.models.Location:teamsnap.teamsnap.api.Location
|
||||
}.get(teamsnap_instance._meta.model)
|
||||
teamsnap_model = teamsnap_instance._meta.model
|
||||
new_data = ApiObject.get(client=self.client, id=teamsnap_instance.id).data
|
||||
obj, created = self._update_or_create_from_teamsnapapi(teamsnap_model, new_data)
|
||||
return [(obj, created)]
|
||||
|
||||
def _update_or_create_from_teamsnapapi(self, teamsnap_model, teamsnap_data, create_benchcoach_object = False):
|
||||
related_objects = {}
|
||||
fields = ['id', 'created_at', 'updated_at']
|
||||
if teamsnap_model == teamsnap.models.Event:
|
||||
fields += [
|
||||
'label',
|
||||
'start_date',
|
||||
'formatted_title',
|
||||
'points_for_opponent',
|
||||
'points_for_team',
|
||||
'is_game',
|
||||
'game_type'
|
||||
]
|
||||
if teamsnap_data.get('location_id'):
|
||||
related_objects['location'] = self._update_or_create_from_teamsnapapi(
|
||||
teamsnap.models.Location,
|
||||
{'id':teamsnap_data['location_id']}
|
||||
)
|
||||
if teamsnap_data.get('opponent_id'):
|
||||
related_objects['opponent'] = self._update_or_create_from_teamsnapapi(
|
||||
teamsnap.models.Opponent,
|
||||
{'id':teamsnap_data['opponent_id']}
|
||||
)
|
||||
|
||||
elif teamsnap_model == teamsnap.models.Opponent:
|
||||
fields += ['name']
|
||||
|
||||
elif teamsnap_model == teamsnap.models.Team:
|
||||
fields += ['name']
|
||||
|
||||
elif teamsnap_model == teamsnap.models.Location:
|
||||
fields += ['name']
|
||||
|
||||
elif teamsnap_model == teamsnap.models.Member:
|
||||
fields += [
|
||||
'first_name',
|
||||
'last_name',
|
||||
'jersey_number',
|
||||
'is_non_player'
|
||||
]
|
||||
elif teamsnap_model == teamsnap.models.Availability:
|
||||
fields += ['status_code']
|
||||
|
||||
related_objects['member'] = self._update_or_create_from_teamsnapapi(
|
||||
teamsnap.models.Member,
|
||||
{'id': teamsnap_data['member_id']}
|
||||
)
|
||||
|
||||
related_objects['event'] = self._update_or_create_from_teamsnapapi(
|
||||
teamsnap.models.Event,
|
||||
{'id': teamsnap_data['event_id']}
|
||||
)
|
||||
|
||||
else:
|
||||
raise ValueError
|
||||
|
||||
if teamsnap_data.get('team_id'):
|
||||
related_objects['team'] = self._update_or_create_from_teamsnapapi(teamsnap.models.Team,
|
||||
{"id":teamsnap_data['team_id']})
|
||||
|
||||
data = {field: teamsnap_data[field] for field in fields if teamsnap_data.get(field) != None}
|
||||
id = data.pop('id')
|
||||
instance, created = teamsnap_model.objects.update_or_create(id=id, defaults=data)
|
||||
r_related_objects = []
|
||||
for related_object_name, related_objectcreated_list in related_objects.items():
|
||||
related_object, created = related_objectcreated_list[0] #FIXME This can't be right, do we need a list for related?
|
||||
setattr(instance, related_object_name, related_object)
|
||||
r_related_objects.append((related_object, created))
|
||||
instance.save()
|
||||
# if create_benchcoach_object:
|
||||
# ben
|
||||
return [(instance, created)] + r_related_objects
|
||||
|
||||
def _update_teamsnapdb_to_benchcoachdb(self, benchcoach_instance, teamsnap_instance,
|
||||
create_if_doesnt_exist: bool = False) -> List[Tuple[django.db.models.Model, bool]]:
|
||||
''' Function to update from a teamsnap object to Benchcoach object.
|
||||
|
||||
:param d: The information to update.
|
||||
:param teamsnap_object: The teamsnap object from which to update.
|
||||
:param create_benchcoach_object: If true, will create the benchcoach object if it doesn't exist
|
||||
:param create_related: This is here for decoration only. It doesn't do anything.
|
||||
:return: a list of tuples in the form (obj, did_create) for created or modified objects.
|
||||
'''
|
||||
|
||||
if isinstance(teamsnap_instance, teamsnap.models.Event):
|
||||
benchcoach_model = Event
|
||||
|
||||
d = {
|
||||
'start': teamsnap_instance.start_date,
|
||||
}
|
||||
|
||||
if teamsnap_instance.team:
|
||||
if teamsnap_instance.team.benchcoach_object:
|
||||
if teamsnap_instance.game_type == "Home":
|
||||
d['home_team'] = teamsnap_instance.team.benchcoach_object
|
||||
elif teamsnap_instance.game_type == "Away":
|
||||
d['away_team'] = teamsnap_instance.team.benchcoach_object
|
||||
elif not teamsnap_instance.team.benchcoach_object:
|
||||
raise Team.DoesNotExist
|
||||
|
||||
if teamsnap_instance.opponent:
|
||||
if teamsnap_instance.opponent.benchcoach_object:
|
||||
if teamsnap_instance.game_type == 'Home':
|
||||
d['away_team'] = teamsnap_instance.opponent.benchcoach_object
|
||||
elif teamsnap_instance.game_type == 'Away':
|
||||
d['home_team'] = teamsnap_instance.opponent.benchcoach_object
|
||||
elif not teamsnap_instance.opponent.benchcoach_object:
|
||||
raise Team.DoesNotExist
|
||||
|
||||
if teamsnap_instance.location:
|
||||
if teamsnap_instance.location.benchcoach_object:
|
||||
if teamsnap_instance.location:
|
||||
d['venue'] = teamsnap_instance.location.benchcoach_object
|
||||
elif not teamsnap_instance.location.benchcoach_object:
|
||||
raise Venue.DoesNotExist
|
||||
|
||||
elif isinstance(teamsnap_instance, teamsnap.models.Opponent):
|
||||
benchcoach_model = Team
|
||||
d = {
|
||||
'name': teamsnap_instance.name,
|
||||
}
|
||||
|
||||
elif isinstance(teamsnap_instance, teamsnap.models.Team):
|
||||
benchcoach_model = Team
|
||||
d = {
|
||||
'name': teamsnap_instance.name,
|
||||
}
|
||||
|
||||
elif isinstance(teamsnap_instance, teamsnap.models.Location):
|
||||
benchcoach_model = Venue
|
||||
d = {
|
||||
'name': teamsnap_instance.name,
|
||||
}
|
||||
|
||||
elif isinstance(teamsnap_instance, teamsnap.models.Member):
|
||||
benchcoach_model = Player
|
||||
d = {
|
||||
'first_name': teamsnap_instance.first_name,
|
||||
'last_name': teamsnap_instance.last_name,
|
||||
'jersey_number': teamsnap_instance.jersey_number,
|
||||
}
|
||||
|
||||
elif isinstance(teamsnap_instance, teamsnap.models.Availability):
|
||||
benchcoach_model = Availability
|
||||
|
||||
translation = {
|
||||
teamsnap_instance.YES: Availability.YES,
|
||||
teamsnap_instance.NO: Availability.NO,
|
||||
teamsnap_instance.MAYBE: Availability.MAYBE
|
||||
}
|
||||
|
||||
d = {
|
||||
'available': translation.get(teamsnap_instance.status_code, Availability.UNKNOWN),
|
||||
'player': teamsnap_instance.member.benchcoach_object,
|
||||
'event': teamsnap_instance.event.benchcoach_object
|
||||
}
|
||||
|
||||
r = []
|
||||
|
||||
if teamsnap_instance.member.benchcoach_object:
|
||||
d['player'] = teamsnap_instance.member.benchcoach_object
|
||||
elif not teamsnap_instance.member.benchcoach_object:
|
||||
raise Availability.DoesNotExist
|
||||
|
||||
if teamsnap_instance.event.benchcoach_object:
|
||||
d['event'] = teamsnap_instance.event.benchcoach_object
|
||||
elif not teamsnap_instance.event.benchcoach_object:
|
||||
raise Event.DoesNotExist
|
||||
|
||||
else:
|
||||
raise ValueError
|
||||
|
||||
r=[]
|
||||
if teamsnap_instance.benchcoach_object:
|
||||
benchcoach_object = benchcoach_model.objects.filter(id=teamsnap_instance.benchcoach_object.id)
|
||||
benchcoach_object.update(**d)
|
||||
created = False
|
||||
r.append((benchcoach_object.first(), created))
|
||||
# elif not teamsnap_instance.benchcoach_object and create_if_doesnt_exist:
|
||||
elif not teamsnap_instance.benchcoach_object:
|
||||
raise django.db.models.Model.DoesNotExist
|
||||
|
||||
return r
|
||||
|
||||
def _find_counterpart(self, instance):
|
||||
instance_type = type(instance)
|
||||
if instance_type == Availability:
|
||||
counterpart_instance = instance.teamsnap_availability
|
||||
|
||||
elif instance_type == Player:
|
||||
counterpart_instance = instance.teamsnap_member
|
||||
|
||||
elif instance_type == Event:
|
||||
counterpart_instance = instance.teamsnap_event
|
||||
|
||||
elif instance_type == Venue:
|
||||
counterpart_instance = instance.teamsnap_location
|
||||
|
||||
elif instance_type == Team:
|
||||
if hasattr(instance, 'teamsnap_opponent'):
|
||||
counterpart_instance = instance.teamsnap_opponent
|
||||
elif hasattr(instance, 'teamsnap_team'):
|
||||
counterpart_instance = instance.teamsnap_team
|
||||
else:
|
||||
raise ValueError("instance doesn't seem to be an teamsnap opponent or a teamsnap team")
|
||||
|
||||
elif instance_type == Positioning:
|
||||
counterpart_instance = instance.teamsnap_lineupentry
|
||||
|
||||
if not counterpart_instance: raise Exception()
|
||||
|
||||
return counterpart_instance
|
||||
|
||||
def _fetch_new_data(self, instance):
|
||||
api_object = instance.ApiObject.get(client=self.client, id=instance.id)
|
||||
return api_object.data
|
||||
|
||||
def _fetch_sync(self, instance):
|
||||
r=[]
|
||||
counterpart_instance = self._find_counterpart(instance)
|
||||
r += self._update_teamsnapdb_from_teamsnapapi(counterpart_instance)
|
||||
r += self._update_teamsnapdb_to_benchcoachdb(instance, counterpart_instance)
|
||||
return r
|
||||
|
||||
def _sync_qs (self, qs, direction):
|
||||
if qs.model not in self.models:
|
||||
raise TypeError(f"Sync engine does not sync {qs.model} models")
|
||||
|
||||
r=[]
|
||||
|
||||
for instance in qs:
|
||||
r += self._sync_instance(instance, direction=direction)
|
||||
|
||||
return r
|
||||
|
||||
def _sync_instance(self, instance, direction, data=None):
|
||||
r=[]
|
||||
if direction == 'download':
|
||||
r += self._fetch_sync(instance)
|
||||
|
||||
elif direction == 'upload':
|
||||
raise NotImplementedError('Uploading not supported by this sync engine yet.')
|
||||
else:
|
||||
raise TypeError(f"Direction {direction} not supported. 'upload' or 'download' must be specified")
|
||||
|
||||
return r
|
||||
|
||||
|
||||
def sync(self, qs: django.db.models.QuerySet = None, instance: django.db.models.Model = None,
|
||||
direction='download') -> List[Tuple[django.db.models.Model, bool]]:
|
||||
if not qs and not instance:
|
||||
raise TypeError(f"sync requires either a QuerySet or model instance to be provided")
|
||||
if qs and instance:
|
||||
raise TypeError(f"sync requires either a QuerySet or model instance to be provided, but not both")
|
||||
elif qs:
|
||||
r = self._sync_qs(qs, direction)
|
||||
elif instance:
|
||||
r = self._sync_instance(instance, direction)
|
||||
|
||||
return r
|
||||
|
||||
def import_items(self, object_name):
|
||||
object_names = []
|
||||
if object_name == 'team':
|
||||
object_names += ['opponent', 'team']
|
||||
elif object_name == 'player':
|
||||
object_names = ['member']
|
||||
elif object_name == 'venue':
|
||||
object_name == ['location']
|
||||
elif object_name in [model.__name__.lower() for model in self.models]:
|
||||
object_names += [object_name]
|
||||
|
||||
if len(object_names) == 0:
|
||||
raise ValueError('no valid keyword provided')
|
||||
|
||||
for object_name in object_names:
|
||||
Object = {
|
||||
obj.__name__.lower(): obj
|
||||
for obj in
|
||||
[
|
||||
teamsnap.models.Availability,
|
||||
teamsnap.models.Event,
|
||||
# teamsnap.models.LineupEntry,
|
||||
teamsnap.models.Location,
|
||||
teamsnap.models.Member,
|
||||
teamsnap.models.Opponent,
|
||||
teamsnap.models.Team,
|
||||
# teamsnap.models.User
|
||||
]
|
||||
}.get(object_name)
|
||||
|
||||
ApiObject = {
|
||||
apiobj.__name__.lower(): apiobj
|
||||
for apiobj in
|
||||
[
|
||||
teamsnap.teamsnap.api.Availability,
|
||||
teamsnap.teamsnap.api.Event,
|
||||
# teamsnap.teamsnap.api.LineupEntry,
|
||||
teamsnap.teamsnap.api.Location,
|
||||
teamsnap.teamsnap.api.Member,
|
||||
teamsnap.teamsnap.api.Opponent,
|
||||
teamsnap.teamsnap.api.Team,
|
||||
# teamsnap.teamsnap.api.User
|
||||
]
|
||||
}.get(object_name)
|
||||
|
||||
pass
|
||||
|
||||
if not Object: raise KeyError(f"key {object_name} not found.")
|
||||
r = []
|
||||
|
||||
a = ApiObject.search(self.client, team_id=self.managed_teamsnap_team_id)
|
||||
for _a in a:
|
||||
response = self._update_or_create_from_teamsnapapi(Object, _a.data)
|
||||
r += response
|
||||
|
||||
return r
|
||||
@@ -1,25 +0,0 @@
|
||||
from django.test import TestCase
|
||||
import os
|
||||
|
||||
from benchcoach.utils.teamsnap_sync_engine import TeamsnapSyncEngine
|
||||
|
||||
import benchcoach.models
|
||||
|
||||
TEAMSNAP_TOKEN = os.environ['TEAMSNAP_TOKEN']
|
||||
TEAM_TEAMSNAP_ID = os.environ['TEAM_TEAMSNAP_ID']
|
||||
|
||||
syncengine = TeamsnapSyncEngine(managed_team_teamsnap_id=TEAM_TEAMSNAP_ID, teamsnap_token=TEAMSNAP_TOKEN)
|
||||
r=syncengine.import_items('event')
|
||||
pass
|
||||
|
||||
class TestEventModel(TestCase):
|
||||
fixtures = ['minimal']
|
||||
|
||||
def setUp(self):
|
||||
self.syncengine = TeamsnapSyncEngine(managed_team_teamsnap_id=TEAM_TEAMSNAP_ID, teamsnap_token=TEAMSNAP_TOKEN)
|
||||
|
||||
def test_all_models(self):
|
||||
for Model in self.syncengine.models:
|
||||
with self.subTest():
|
||||
self.syncengine.import_items(Model.__name__.lower())
|
||||
pass
|
||||
Reference in New Issue
Block a user