mygpo/favorites/models.py

   1 from __future__ import unicode_literals
   2
   3 from django.db import models
   4 from django.conf import settings
   5
   6 from mygpo.core.models import UpdateInfoModel
   7 from mygpo.podcasts.models import Episode
   8
   9
  10 class FavoriteEpisode(UpdateInfoModel):
  11
  12     # the user that has a favorite episode
  13     user = models.ForeignKey(settings.AUTH_USER_MODEL,
  14                              on_delete=models.CASCADE)
  15
  16     # episode that is the user's favorite
  17     episode = models.ForeignKey(Episode, db_index=True,
  18                                 on_delete=models.PROTECT)
  19
  20     class Meta:
  21         unique_together = [
  22             ('user', 'episode'),
  23         ]
  24
  25     @classmethod
  26     def episodes_for_user(cls, user):
  27         """ Favorite episodes of the given user
  28
  29         Returns the episodes directly, not the FavoriteEpisode objects """
  30         return Episode.objects.filter(favoriteepisode__user=user)\
  31                               .select_related('podcast')\
  32                               .prefetch_related('slugs', 'podcast__slugs')