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 def dpx(self): all_time = Booking.total_booking_time() duck_time = Booking.duck_booking_time(self) if (all_time == None) or (duck_time == None): return 0 return Booking.duck_booking_time(self) / Booking.total_booking_time() 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 @staticmethod def duck_booking_time(duck): return Booking.objects.filter(start_ts__isnull = False, end_ts__isnull = False, duck = duck).extra(select = {'amount': 'sum(strftime(%s, end_ts) - strftime(%s, start_ts))'}, select_params = ('%s', '%s'))[0].amount