110 lines
3.5 KiB
Python
110 lines
3.5 KiB
Python
from django.db import models
|
||
from django.contrib.auth.models import User
|
||
from django.utils import timezone
|
||
from django.conf import settings
|
||
|
||
from .ducklevel import minutes_to_level
|
||
|
||
class Species(models.Model):
|
||
"""Model to hold the Ducks’ species"""
|
||
|
||
name = models.CharField(max_length = 40, unique = True)
|
||
|
||
def __str__(self):
|
||
return self.name
|
||
|
||
class Location(models.Model):
|
||
"""Model to hold the possible locations of the Ducks"""
|
||
|
||
name = models.CharField(max_length = 50)
|
||
|
||
def __str__(self):
|
||
return self.name
|
||
|
||
class Competence(models.Model):
|
||
"""Model to hold Duck competences"""
|
||
|
||
name = models.CharField(max_length = 255, unique = True)
|
||
added_at = models.DateTimeField(default = timezone.now)
|
||
added_by = models.ForeignKey(User)
|
||
|
||
def __str__(self):
|
||
return self.name
|
||
|
||
class Duck(models.Model):
|
||
"""Model to hold Duck data"""
|
||
|
||
name = models.CharField(max_length = 80, null = True, blank = True)
|
||
color = models.CharField(max_length = 6)
|
||
species = models.ForeignKey(Species)
|
||
location = models.ForeignKey(Location)
|
||
comps = models.ManyToManyField(Competence, through = 'DuckCompetence')
|
||
donated_by = models.ForeignKey(User)
|
||
donated_at = models.DateTimeField(default = timezone.now)
|
||
adopted_by = models.ForeignKey(User, related_name = 'adopted_ducks', null = True)
|
||
adopted_at = models.DateTimeField(null = True)
|
||
bookings = models.ManyToManyField(User, through = 'Booking', related_name = '+')
|
||
on_holiday_since = models.DateTimeField(null = True)
|
||
on_holiday_until = models.DateTimeField(null = True)
|
||
|
||
def __str__(self):
|
||
if self.name == None or self.name == '':
|
||
return 'Unnamed :('
|
||
|
||
return self.name
|
||
|
||
def age(self):
|
||
seconds_d = timezone.now() - self.donated_at
|
||
seconds = seconds_d.total_seconds()
|
||
|
||
if seconds < 0:
|
||
return -1
|
||
|
||
return seconds
|
||
|
||
class DuckName(models.Model):
|
||
"""Model to hold name suggestions for Ducks"""
|
||
|
||
duck = models.ForeignKey(Duck)
|
||
name = models.CharField(max_length = 60)
|
||
suggested_by = models.ForeignKey(User)
|
||
suggested_at = models.DateTimeField(default = timezone.now)
|
||
closed_by = models.ForeignKey(User, related_name = '+')
|
||
closed_at = models.DateTimeField(null = True)
|
||
|
||
class DuckNameVote(models.Model):
|
||
"""Model to hold votes to Duck names"""
|
||
|
||
duck_name = models.ForeignKey(DuckName)
|
||
vote_timestamp = models.DateTimeField(default = timezone.now)
|
||
voter = models.ForeignKey(User)
|
||
upvote = models.BooleanField(default = True)
|
||
|
||
class DuckCompetence(models.Model):
|
||
"""Duck competence governor table"""
|
||
|
||
duck = models.ForeignKey(Duck)
|
||
comp = models.ForeignKey(Competence)
|
||
up_minutes = models.IntegerField(default = 0)
|
||
down_minutes = models.IntegerField(default = 0)
|
||
|
||
def level(self):
|
||
return minutes_to_level(self.up_minutes, self.down_minutes)
|
||
|
||
class Meta:
|
||
unique_together = ('duck', 'comp')
|
||
|
||
class Booking(models.Model):
|
||
"""Duck booking governor table"""
|
||
|
||
duck = models.ForeignKey(Duck)
|
||
user = models.ForeignKey(User)
|
||
comp_req = models.ForeignKey(Competence)
|
||
start_ts = models.DateTimeField(default = timezone.now)
|
||
end_ts = models.DateTimeField(null = True, blank = True)
|
||
successful = models.BooleanField(default = True)
|
||
|
||
@staticmethod
|
||
def total_booking_time():
|
||
return Booking.objects.filter(start_ts__isnull = False, end_ts__isnull = False).extra(select = {'amount': 'sum(strftime(%s, end_ts) - strftime(%s, start_ts))'}, select_params = ('%s', '%s'))[0].amount
|