models.py

# -*- coding: utf-8 -*-

from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save, pre_save, m2m_changed
from django.utils.translation import ugettext as _
from tastypie.models import create_api_key


class Person(models.Model):
    user = models.OneToOneField(User, blank=True, null=True, related_name='handball_profile')

    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)
    address = models.CharField(max_length=50, blank=True)
    city = models.CharField(max_length=50, blank=True)
    zip_code = models.IntegerField(null=True, blank=True)
    birthday = models.DateField(null=True, blank=True)
    pass_number = models.IntegerField(unique=True, null=True, blank=True)
    gender = models.CharField(max_length=10, choices=(('male', _('male')), ('female', _('female'))), default='male')
    mobile_number = models.CharField(max_length=20, blank=True)

    def __unicode__(self):
        return self.first_name + ' ' + self.last_name


class Team(models.Model):
    name = models.CharField(max_length=50)

    players = models.ManyToManyField('Person', blank=True, related_name='teams', through='TeamPlayerRelation')
    coaches = models.ManyToManyField('Person', blank=True, related_name='teams_coached', through='TeamCoachRelation')
    # league = models.ForeignKey('League', related_name='league', blank=True)
    club = models.ForeignKey('Club', related_name='teams')
    managers = models.ManyToManyField('Person', blank=True, related_name='teams_managed')

    def __unicode__(self):
        return self.club.name + ' ' + self.name


class Club(models.Model):
    name = models.CharField(max_length=50)

    home_site = models.ForeignKey('Site', blank=True, null=True)
    district = models.ForeignKey('District', related_name='clubs')
    members = models.ManyToManyField('Person', related_name='clubs', blank=True, through='ClubMemberRelation')
    managers = models.ManyToManyField('Person', blank=True, related_name='clubs_managed')

    def __unicode__(self):
        return self.name


class League(models.Model):
    name = models.CharField(max_length=50)
    gender = models.CharField(max_length=10, choices=(('male', _('male')), ('female', _('female'))))
    age_group = models.CharField(max_length=20, choices=(('adults', _('adults')), ('juniors', _('juniors')), ('kids', _('kids'))))

    union = models.ForeignKey('Union', related_name='leagues', blank=True)
    district = models.ForeignKey('District', related_name='leagues', blank=True)
    managers = models.ManyToManyField('Person', blank=True, related_name='leagues_managed')

    def __unicode__(self):
        return '{0} {1} {2}'.format(self.name, self.gender, self.age_group)


class LeagueTemplate(models.Model):
    name = models.CharField(max_length=50)
    gender = models.CharField(max_length=10, choices=(('male', _('male')), ('female', _('female'))))
    age_group = models.CharField(max_length=20, choices=(('adults', _('adults')), ('juniors', _('juniors')), ('kids', _('kids'))))

    def __unicode__(self):
        return '{0} {1} {2}'.format(self.name, self.gender, self.age_group)


class Group(models.Model):
    name = models.CharField(max_length=50)

    union = models.ForeignKey('Union', related_name='groups', blank=True)
    district = models.ForeignKey('District', related_name='groups', blank=True)
    league = models.ForeignKey('League', related_name='groups', blank=True)
    teams = models.ManyToManyField('Team', related_name='groups', blank=True)

    def __unicode__(self):
        return '{0}, {1}, {2}, {3}'.format(self.name, self.league.name, self.league.gender, self.league.age_group, self.league.district.name)


class District(models.Model):
    name = models.CharField(max_length=50)

    union = models.ForeignKey('Union', related_name='districts')
    managers = models.ManyToManyField('Person', blank=True, related_name='districts_managed')

    def __unicode__(self):
        return self.name


class Union(models.Model):
    name = models.CharField(max_length=50)

    managers = models.ManyToManyField('Person', blank=True, related_name='unions_managed')

    def __unicode__(self):
        return self.name


class Game(models.Model):
    number = models.IntegerField(unique=True, blank=True, null=True)
    start = models.DateTimeField()
    score_home = models.IntegerField()
    score_away = models.IntegerField()

    home = models.ForeignKey('Team', related_name='games_home')
    away = models.ForeignKey('Team', related_name='games_away')
    referee = models.ForeignKey('Person', related_name='games_as_referee')
    timer = models.ForeignKey('Person', related_name='games_as_timer')
    secretary = models.ForeignKey('Person', related_name='games_as_secretary')
    supervisor = models.ForeignKey('Person', related_name='games_as_supervisor')
    winner = models.ForeignKey('Team', related_name='games_won', blank=True, null=True)
    group = models.ForeignKey('Group', related_name='games')
    game_type = models.ForeignKey('GameType')
    site = models.ForeignKey('Site')
    players = models.ManyToManyField('Person', through='GamePlayerRelation')

    def __unicode__(self):
        return '{0}/{1}/{2}: {3} {4} vs. {5} {6}'.format(self.start.year, self.start.month, self.start.day, self.home.club.name, self.home.name, self.away.club.name, self.away.name)


class GameType(models.Model):
    name = models.CharField(max_length=20)

    def __unicode__(self):
        return self.name


class Site(models.Model):
    # name = models.CharField(max_length=50)
    address = models.CharField(max_length=50)
    city = models.CharField(max_length=50)
    zip_code = models.IntegerField()
    number = models.IntegerField(unique=True, blank=True, null=True)

    def __unicode__(self):
        return '{0}, {1} {2} (#{3})'.format(self.address, self.zip_code, self.city, self.number)


class GamePlayerRelation(models.Model):
    player = models.ForeignKey('Person')
    game = models.ForeignKey('Game')

    team = models.ForeignKey('Team')
    shirt_number = models.IntegerField(blank=True, null=True)
    # goals = models.IntegerField()
    # warnings = models.IntegerField()
    # penalties = models.IntegerField()
    # disqualifications = models.IntegerField()
    # team_penalties = models.IntegerField()
    # penalty_shots_hit = models.IntegerField()
    # penalty_shots_miss = models.IntegerField()


class ClubMemberRelation(models.Model):
    member = models.ForeignKey('Person')
    club = models.ForeignKey('Club')

    primary = models.BooleanField()
    member_confirmed = models.BooleanField(default=False)
    manager_confirmed = models.BooleanField(default=False)


class TeamPlayerRelation(models.Model):
    player = models.ForeignKey('Person')
    team = models.ForeignKey('Team')

    player_confirmed = models.BooleanField(default=False)
    manager_confirmed = models.BooleanField(default=False)


class TeamCoachRelation(models.Model):
    coach = models.ForeignKey('Person')
    team = models.ForeignKey('Team')

    member_confirmed = models.BooleanField(default=False)
    manager_confirmed = models.BooleanField(default=False)


class Event(models.Model):
    time = models.IntegerField()

    person = models.ForeignKey('Person')
    event_type = models.ForeignKey('EventType')
    game = models.ForeignKey('Game', related_name='events')
    team = models.ForeignKey('Team')


class EventType(models.Model):
    name = models.CharField(max_length=20)

    def __unicode__(self):
        return self.name


def create_default_leagues(sender, instance, created, **kwargs):
    # Create defaults leagues for District after creation
    if created:
        templates = LeagueTemplate.objects.all()

        for template in templates:
            League.objects.create(name=template.name, gender=template.gender, age_group=template.age_group, district=instance)


def set_union_by_district(sender, instance, **kwargs):
    # If district is set, set according union
    if instance.district:
        instance.union = instance.district.union


# def set_club_by_team(sender, instance, action, reverse, model, pk_set, **kwargs):
#     if action == 'post_add':
#         for pk in pk_set:
#             player = model.objects.get(pk=pk)
#             # player.clubs.add(instance.club)
#             ClubMemberRelation.objects.create(member=player, club=instance)
#             player.save()

def set_club_by_team(sender, instance, created, **kwargs):
    try:
        ClubMemberRelation.objects.get(member=instance.player, club=instance.team.club)
    except ClubMemberRelation.DoesNotExist:
        ClubMemberRelation.objects.create(member=instance.player, club=instance.team.club,
            manager_confirmed=instance.manager_confirmed, member_confirmed=instance.player_confirmed)


def add_player_to_team(sender, instance, created, **kwargs):
    try:
        TeamPlayerRelation.objects.get(team=instance.team, player=instance.player)
    except TeamPlayerRelation.DoesNotExist:
        TeamPlayerRelation.objects.create(team=instance.team, player=instance.player, manager_confirmed=True, player_confirmed=True)

# Create API key for a new user
post_save.connect(create_api_key, sender=User)
post_save.connect(create_default_leagues, sender=District)
pre_save.connect(set_union_by_district, sender=League)
post_save.connect(set_club_by_team, sender=TeamPlayerRelation)
post_save.connect(add_player_to_team, sender=GamePlayerRelation)
# m2m_changed.connect(set_club_by_team, sender=Team.players.through)