mygpo/categories/models.py

   1 from django.db import models
   2
   3 from mygpo.core.models import UpdateInfoModel
   4 from mygpo.podcasts.models import Podcast
   5
   6
   7 class Category(UpdateInfoModel):
   8     """ A category of podcasts """
   9
  10     title = models.CharField(max_length=1000, null=False, blank=False,
  11                              unique=True)
  12
  13     num_entries = models.IntegerField()
  14
  15     class Meta:
  16         verbose_name = 'Category'
  17         verbose_name_plural = 'Categories'
  18
  19         index_together = [
  20             ('modified', 'num_entries'),
  21         ]
  22
  23     def save(self, *args, **kwargs):
  24         self.num_entries = self.entries.count()
  25         super(Category, self).save(*args, **kwargs)
  26
  27     @property
  28     def podcasts(self):
  29         return self.entries.prefetch_related('podcast', 'podcast__slugs')
  30
  31     @property
  32     def clean_title(self):
  33         return self.title.replace('\n', ' ')
  34
  35
  36 class CategoryEntry(UpdateInfoModel,):
  37     """ A podcast in a category """
  38
  39     category = models.ForeignKey(Category, related_name='entries',
  40                                  on_delete=models.CASCADE)
  41
  42     podcast = models.ForeignKey(Podcast,
  43                                 on_delete=models.CASCADE)
  44
  45     class Meta:
  46         unique_together = [
  47             ('category', 'podcast'),
  48         ]
  49
  50         index_together = [
  51             ('category', 'modified'),
  52         ]
  53
  54
  55 class CategoryTag(models.Model):
  56
  57     tag = models.SlugField(unique=True)
  58
  59     category = models.ForeignKey(Category, related_name='tags',
  60                                  on_delete=models.CASCADE)