mygpo/userfeeds/feeds.py

   1 from mygpo.api.models.users import EpisodeFavorite
   2
   3
   4 class FavoriteFeed():
   5
   6     def __init__(self, user):
   7         self.user = user
   8
   9     def title(self):
  10         return '%s\'s Favorite Episodes' % self.user.username
  11
  12     def get_episodes(self):
  13         return [x.episode for x in EpisodeFavorite.objects.filter(user=self.user).order_by('-created')]
  14
  15     def language(self):
  16         """
  17         If all of the feed's episodes have the same language, return it,
  18         otherwise return an empty string
  19         """
  20         l = list(set([x.language for x in self.get_episodes() if x.language]))
  21         if len(l) == 1:
  22             return l[0]
  23         else:
  24             return ''
  25