benchcoach-django/teamsnap/models.py

from django.db import models

import benchcoach.models
import pyteamsnap.api
from django.utils.timezone import localtime

class TeamsnapBaseModel(models.Model):
   type = None
   id = models.CharField(max_length=50, unique=True, primary_key=True)
   created_at = models.DateTimeField(null=True)
   updated_at = models.DateTimeField(null=True)
   ApiObject = pyteamsnap.api.ApiObject

   class Meta:
      abstract = True

   def __str__(self):
      return f"TeamSnap {self.__class__.__name__} Object ({self.id})"

   @property
   def api_url(self):
      return "https://api.teamsnap.com/v3/{type}/{id}".format(type=self.type, id=self.id)

class Team(TeamsnapBaseModel):
   type = 'team'
   name = models.CharField(max_length=50, null=True)
   benchcoach_object = models.OneToOneField(
      benchcoach.models.Team,
      on_delete=models.CASCADE,
      related_name="teamsnap_team"
   )
   ApiObject = pyteamsnap.api.Team

class User(TeamsnapBaseModel):
   type = 'user'
   first_name = models.CharField(max_length=50, null=True)
   last_name = models.CharField(max_length = 50, null=True)
   email = models.EmailField(null=True)
   managed_teams = models.ManyToManyField(Team)
   ApiObject = pyteamsnap.api.User

   @classmethod
   def update_or_create_from_teamsnap_api(cls, teamsnap_data):
      fields = ['id', 'first_name', 'last_name', 'email']
      user_data = {k:teamsnap_data[k] for k in fields}
      managed_teams = []
      for managed_team_id in teamsnap_data['managed_team_ids']:
         obj, created = Team.objects.get_or_create(id=managed_team_id)
         managed_teams.append(obj)
         pass
      id = user_data.pop('id')
      user, created = cls.objects.update_or_create(id=id, defaults=user_data)
      user.managed_teams.add(*managed_teams)
      return (user, created)

class TeamsnapManagedObjectModel(TeamsnapBaseModel):
   team = models.ForeignKey(
      Team,
      verbose_name="managed by team",
      null=True,
      on_delete=models.CASCADE,
   )

   class Meta:
      abstract = True

   @property
   def url(self, endpoint='view'):
      return f"https://go.teamsnap.com/{self.team.id}/{self.type}/{endpoint}/{self.id}"

class Opponent(TeamsnapManagedObjectModel):
   type = 'opponent'
   name = models.CharField(max_length=50, null=True)
   benchcoach_object = models.OneToOneField(
      benchcoach.models.Team,
      on_delete=models.CASCADE,
      related_name="teamsnap_opponent"
   )
   ApiObject = pyteamsnap.api.Opponent

class Location(TeamsnapManagedObjectModel):
   type = 'location'
   name = models.CharField(max_length=50, null=True)
   benchcoach_object = models.OneToOneField(
      benchcoach.models.Venue,
      on_delete=models.CASCADE,
      related_name="teamsnap_location"
   )
   ApiObject = pyteamsnap.api.Location

class Member(TeamsnapManagedObjectModel):
   # url format is
   # f"https://go.teamsnap.com/{self.team.teamsnap_id}/roster/player/{self.teamsnap_id}"
   # f"https://go.teamsnap.com/{self.team.teamsnap_id}/roster/edit/{self.teamsnap_id}"
   type = 'member'
   name = models.CharField(max_length=50, null=True)
   benchcoach_object = models.OneToOneField(
      benchcoach.models.Player,
      on_delete=models.CASCADE,
      related_name="teamsnap_member"
   )
   first_name = models.CharField(max_length = 50, null=True)
   last_name = models.CharField(max_length = 50, null=True)
   jersey_number = models.IntegerField(null=True)
   is_non_player = models.BooleanField(null=True)
   ApiObject = pyteamsnap.api.Member

   def __str__(self):
      return f"{self.last_name}, {self.first_name} ({self.id})"

   @property
   def name(self):
      return f"{self.first_name} {self.last_name}"

class Event(TeamsnapManagedObjectModel):
   # url is
   # f"https://go.teamsnap.com/{self.team.teamsnap_id}/schedule/view_game/{self.teamsnap_id}"
   # f"https://go.teamsnap.com/{self.team.teamsnap_id}/schedule/edit_game/{self.teamsnap_id}"
   type = 'event'
   benchcoach_object = models.OneToOneField(
      benchcoach.models.Event,
      on_delete=models.CASCADE,
      related_name="teamsnap_event"
   )
   label = models.CharField(max_length = 50, null=True)
   start_date = models.DateTimeField(null=True)
   opponent = models.ForeignKey(Opponent, null=True, on_delete=models.CASCADE, related_name="opponent")
   location = models.ForeignKey(Location, null=True, on_delete=models.CASCADE)
   formatted_title = models.CharField(max_length = 50, null=True)
   points_for_opponent = models.PositiveSmallIntegerField(null=True)
   points_for_team = models.PositiveSmallIntegerField(null=True)
   is_game = models.BooleanField(null=True)
   game_type = models.CharField(max_length = 50, null=True)
   ApiObject = pyteamsnap.api.Event

   @property
   def csv_event_title(self)->str:
      '''
      TeamSnap has a title format that it uses for the csv export which includes the date and time
      :return: formatted string
      '''
      # activate(zone)
      start_date = localtime(self.start_date)
      return f"{self.formatted_title} {start_date:%m/%d/%Y}" + f"{start_date:%-I:%M %p}".rjust(9)

   def __str__(self):
      return f"{self.formatted_title} ({self.id})"

class Availability(TeamsnapManagedObjectModel):
   type='availability'
   YES = 1
   NO = 0
   MAYBE = 2
   UNKNOWN = None
   status_codes = [
      (YES, 'Yes'),
      (NO, 'No'),
      (MAYBE, 'Maybe'),
      (UNKNOWN, 'Unknown')
   ]
   event = models.ForeignKey(Event, null=True, on_delete=models.CASCADE)
   member = models.ForeignKey(Member, null=True, on_delete=models.CASCADE)
   benchcoach_object = models.OneToOneField(
      benchcoach.models.Availability,
      on_delete=models.CASCADE,
      related_name="teamsnap_availability"
   )
   status_code = models.SmallIntegerField(choices=status_codes, null=True, blank=True, default=None)
   ApiObject = pyteamsnap.api.Availability

   def __str__(self):
      return f"{self.member} - {self.event} ({self.id})"

   class Meta:
      verbose_name_plural = "availabilities"

class LineupEntry(TeamsnapManagedObjectModel):
   member = models.ForeignKey(Member, on_delete=models.CASCADE)
   event = models.ForeignKey(Event, on_delete=models.CASCADE)
   positions = [
      (11, 'EH'),
      (1, 'P'),
      (2, 'C'),
      (3, '1B'),
      (4, '2B'),
      (5, '3B'),
      (6, 'SS'),
      (7, 'LF'),
      (8, 'CF'),
      (9, 'RF'),
      (10,'DH')
   ]
   benchcoach_object = models.OneToOneField(
      benchcoach.models.Positioning,
      on_delete=models.CASCADE,
      related_name="teamsnap_lineupentry"
   )
   label = models.PositiveSmallIntegerField(choices=positions, default=None, null=True, blank=True)
   sequence = models.PositiveSmallIntegerField(default=0, null=True, blank=True)
   ApiObject = pyteamsnap.api.EventLineupEntry

   @classmethod
   def update_or_create_from_teamsnap_api(cls, teamsnap_data):
      fields = [
         'id',
         'created_at',
         'updated_at',
         'label',
         'sequence'
      ]
      lineup_entry_data = {k: teamsnap_data[k] for k in fields}
      member, created = Member.objects.get_or_create(id=teamsnap_data['member_id'])
      team, created = Team.objects.get_or_create(id=teamsnap_data['team_id'])
      event, created = Event.objects.get_or_create(id=teamsnap_data['event_id'])
      id = lineup_entry_data.pop('id')
      lineup_entry, created = cls.objects.update_or_create(id=id, defaults=lineup_entry_data)
      lineup_entry.team = team
      lineup_entry.event = event
      lineup_entry.member = member
      lineup_entry.save()
      return (lineup_entry, created)