spradio-server-django/savepointradio/radio/behaviors.py
2018-01-04 16:28:59 -05:00

62 lines
1.8 KiB
Python

from django.db import models
from django.utils import timezone
from django.utils.translation import ugettext_lazy as _
class Disableable(models.Model):
"""
Mixin for models that can be disabled with a specified reason.
"""
disabled = models.BooleanField(_('disabled state'), default=False)
disabled_date = models.DateTimeField(_('disabled on'),
default=None,
blank=True,
null=True)
disabled_reason = models.TextField(_('reason for disabling'), blank=True)
class Meta:
abstract = True
def disable(self, reason=''):
self.disabled = True
self.disabled_date = timezone.now()
self.disabled_reason = reason
self.save()
def enable(self):
self.disabled = False
self.disabled_date = None
self.disabled_reason = ''
self.save()
def _is_enabled(self):
return not self.disabled
_is_enabled.boolean = True
is_enabled = property(_is_enabled)
class Publishable(models.Model):
"""
Mixin for models that can be published to restrict accessibility before an
appointed date/time.
"""
published_date = models.DateTimeField(_('published for listening'),
default=None,
blank=True,
null=True)
class Meta:
abstract = True
def publish(self, date=None):
if date is None:
date = timezone.now()
self.published_date = date
self.save()
def _is_published(self):
if self.published_date is not None:
return self.published_date < timezone.now()
return False
_is_published.boolean = True
is_published = property(_is_published)