limetime/app/timer/models/timer.py

from django.db import models
from django.utils.timezone import now


class Timer(models.Model):

    STATE_ACTIVE = 1
    STATE_EXPIRED = 2

    STATE_CHOICES = (
        (STATE_ACTIVE, 'Active'),
        (STATE_EXPIRED, 'Expired'),
    )

    TYPE_SHIELD_REENFORCEMENT = 1
    TYPE_ARMOR_REENFORCEMENT = 2

    TYPE_CHOICES = (
        (TYPE_SHIELD_REENFORCEMENT, 'Shield Reenforcement'),
        (TYPE_ARMOR_REENFORCEMENT, 'Armor Reenforcement'),
    )

    location = models.ForeignKey('timer.Location', related_name='timers')
    expiration = models.DateTimeField('Timer Expiration')
    reenforcement_type = models.PositiveIntegerField('Timer Type', choices=TYPE_CHOICES)

    @property
    def state(self):
        if self.expiration <= now():
            return self.STATE_EXPIRED
        return self.STATE_ACTIVE

    def get_state_display(self):
        state = self.state
        for v, disp in self.STATE_CHOICES:
            if state == v:
                return disp

    class Meta:
        app_label = 'timer'
        ordering = ['-expiration']