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     @property
  36     def tag(self):
  37         return self.tags.first().tag
  38
  39
  40 class CategoryEntry(UpdateInfoModel,):
  41     """ A podcast in a category """
  42
  43     category = models.ForeignKey(Category, related_name='entries',
  44                                  on_delete=models.CASCADE)
  45
  46     podcast = models.ForeignKey(Podcast,
  47                                 on_delete=models.CASCADE)
  48
  49     class Meta:
  50         unique_together = [
  51             ('category', 'podcast'),
  52         ]
  53
  54         index_together = [
  55             ('category', 'modified'),
  56         ]
  57
  58
  59 class CategoryTag(models.Model):
  60
  61     tag = models.SlugField(unique=True)
  62
  63     category = models.ForeignKey(Category, related_name='tags',
  64                                  on_delete=models.CASCADE)