1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
|
# -*- coding: utf-8 -*-
"""
Django Extensions abstract base model classes.
"""
from django.db import models
from django.utils.timezone import now
from django.utils.translation import ugettext_lazy as _
from django_extensions.db.fields import (
AutoSlugField, CreationDateTimeField, ModificationDateTimeField,
)
class TimeStampedModel(models.Model):
""" TimeStampedModel
An abstract base class model that provides self-managed "created" and
"modified" fields.
"""
created = CreationDateTimeField(_('created'))
modified = ModificationDateTimeField(_('modified'))
def save(self, **kwargs):
self.update_modified = kwargs.pop('update_modified', getattr(self, 'update_modified', True))
super(TimeStampedModel, self).save(**kwargs)
class Meta:
get_latest_by = 'modified'
ordering = ('-modified', '-created',)
abstract = True
class TitleDescriptionModel(models.Model):
""" TitleDescriptionModel
An abstract base class model that provides title and description fields.
"""
title = models.CharField(_('title'), max_length=255)
description = models.TextField(_('description'), blank=True, null=True)
class Meta:
abstract = True
class TitleSlugDescriptionModel(TitleDescriptionModel):
""" TitleSlugDescriptionModel
An abstract base class model that provides title and description fields
and a self-managed "slug" field that populates from the title.
"""
slug = AutoSlugField(_('slug'), populate_from='title')
class Meta:
abstract = True
class ActivatorQuerySet(models.query.QuerySet):
""" ActivatorQuerySet
Query set that returns statused results
"""
def active(self):
""" Returns active query set """
return self.filter(status=ActivatorModel.ACTIVE_STATUS)
def inactive(self):
""" Returns inactive query set """
return self.filter(status=ActivatorModel.INACTIVE_STATUS)
class ActivatorModelManager(models.Manager):
""" ActivatorModelManager
Manager to return instances of ActivatorModel: SomeModel.objects.active() / .inactive()
"""
def get_queryset(self):
""" Use ActivatorQuerySet for all results """
return ActivatorQuerySet(model=self.model, using=self._db)
def active(self):
""" Returns active instances of ActivatorModel: SomeModel.objects.active(),
proxy to ActivatorQuerySet.active """
return self.get_queryset().active()
def inactive(self):
""" Returns inactive instances of ActivatorModel: SomeModel.objects.inactive(),
proxy to ActivatorQuerySet.inactive """
return self.get_queryset().inactive()
class ActivatorModel(models.Model):
""" ActivatorModel
An abstract base class model that provides activate and deactivate fields.
"""
INACTIVE_STATUS, ACTIVE_STATUS = range(2)
STATUS_CHOICES = (
(INACTIVE_STATUS, _('Inactive')),
(ACTIVE_STATUS, _('Active')),
)
status = models.IntegerField(_('status'), choices=STATUS_CHOICES, default=ACTIVE_STATUS)
activate_date = models.DateTimeField(blank=True, null=True, help_text=_('keep empty for an immediate activation'))
deactivate_date = models.DateTimeField(blank=True, null=True, help_text=_('keep empty for indefinite activation'))
objects = ActivatorModelManager()
class Meta:
ordering = ('status', '-activate_date',)
abstract = True
def save(self, *args, **kwargs):
if not self.activate_date:
self.activate_date = now()
super(ActivatorModel, self).save(*args, **kwargs)
|