63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
from django.db import models
|
|
from django import forms
|
|
import datetime
|
|
from django.utils import timezone
|
|
|
|
def current_day(ttz=None):
|
|
if ttz is None:
|
|
return (datetime.datetime.now(tz=timezone.utc)).date()
|
|
return datetime.datetime.utcnow().astimezone(pytz.timezone(ttz)).date()
|
|
|
|
# Create your models here.
|
|
class Athlete(models.Model):
|
|
first_name = models.CharField(max_length=200)
|
|
last_name = models.CharField(max_length=200)
|
|
club = models.CharField(max_length=200)
|
|
trueskill_mu = models.FloatField(default=25.)
|
|
trueskill_sigma = models.FloatField(default=8.333)
|
|
birth_year = models.IntegerField(default=1972)
|
|
gender = models.CharField(max_length=200, choices=(('m','M'),('f','F')))
|
|
|
|
class Meta:
|
|
unique_together = ('first_name','last_name','birth_year')
|
|
|
|
def __str__(self):
|
|
return u'{f} {l}'.format(f = self.first_name, l=self.last_name)
|
|
|
|
def get_absolute_url(self):
|
|
return "/athlete/%i/" % self.id
|
|
|
|
class athleteForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Athlete
|
|
fields = ['first_name','last_name','club','birth_year']
|
|
|
|
class Crew(models.Model):
|
|
athletes = models.ManyToManyField(Athlete)
|
|
name = models.CharField(max_length=200)
|
|
|
|
def __str__(self):
|
|
return u'{n}'.format(n=self.name)
|
|
|
|
class Race(models.Model):
|
|
name = models.CharField(max_length=200)
|
|
resulturl = models.URLField(null=True)
|
|
date = models.DateField(default=current_day)
|
|
resultlist = models.ManyToManyField(Crew,through='Result')
|
|
crew_size = models.IntegerField(default=1)
|
|
verified = models.BooleanField(default=False)
|
|
|
|
class Meta:
|
|
unique_together = ('date','name')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class Result(models.Model):
|
|
crew = models.ForeignKey(Crew, on_delete=models.CASCADE)
|
|
race = models.ForeignKey(Race, on_delete=models.CASCADE)
|
|
order = models.PositiveIntegerField()
|
|
|
|
class Meta:
|
|
unique_together = ('crew','race','order')
|