164 lines
4.9 KiB
Python
164 lines
4.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
from django.db import models
|
||
from django.contrib.auth.models import User
|
||
from django.utils import timezone
|
||
from django.conf import settings
|
||
|
||
from fuzzywuzzy import fuzz
|
||
|
||
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
|
||
|
||
@classmethod
|
||
def get_similar_comps(cls, name):
|
||
comps = cls.objects.values_list('name', flat=True)
|
||
ret = ()
|
||
|
||
for c in comps:
|
||
r = fuzz.ratio(name.lower(), c.lower())
|
||
|
||
# This ratio is subject to change
|
||
if r > settings.MIN_FUZZY_SIMILARITY:
|
||
ret = ret + (c,)
|
||
|
||
return ret
|
||
|
||
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, blank=True)
|
||
adopted_at = models.DateTimeField(null=True, blank=True)
|
||
bookings = models.ManyToManyField(User, through='Booking', related_name='+')
|
||
on_holiday_since = models.DateTimeField(null=True, blank=True)
|
||
on_holiday_until = models.DateTimeField(null=True, blank=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()
|
||
|
||
def booked_by(self):
|
||
l = self.booking_set.filter(end_ts=None)
|
||
|
||
if len(l) == 0:
|
||
return None
|
||
|
||
if len(l) > 1:
|
||
raise RuntimeError(u"Duck is booked more than once!")
|
||
|
||
return l[0].user
|
||
|
||
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, related_name='competences')
|
||
comp = models.ForeignKey(Competence, related_name='ducks')
|
||
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)
|
||
|
||
@classmethod
|
||
def total_booking_time(cls):
|
||
return cls.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
|
||
|
||
@classmethod
|
||
def duck_booking_time(cls, duck):
|
||
return cls.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
|