duck-booking-tool/booking/models.py
2015-10-19 17:09:16 +02:00

164 lines
4.9 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

# -*- 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