duck-booking-tool/booking/models.py

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

    @staticmethod
    def get_similar_comps(name):
        comps = __class__.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)
    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