mygpo/episodestates/models.py

   1 from django.db import models
   2 from django.conf import settings
   3
   4 from mygpo.podcasts.models import Episode
   5 from mygpo.history.models import EpisodeHistoryEntry
   6
   7
   8 class EpisodeState(models.Model):
   9     """ The latest status of an episode for a user """
  10
  11     user = models.ForeignKey(settings.AUTH_USER_MODEL,
  12                              on_delete=models.CASCADE)
  13
  14     episode = models.ForeignKey(Episode, on_delete=models.CASCADE)
  15
  16     # the latest action
  17     action = models.CharField(
  18         max_length=max(map(len, [action for action, name
  19                                  in EpisodeHistoryEntry.EPISODE_ACTIONS])),
  20         choices=EpisodeHistoryEntry.EPISODE_ACTIONS,
  21     )
  22
  23     # the timestamp at which the event happened (provided by the client)
  24     timestamp = models.DateTimeField()
  25
  26     class Meta:
  27         unique_together = [
  28             ('user', 'episode')
  29         ]
  30
  31     @classmethod
  32     def dict_for_user(cls, user, episodes=None):
  33         """ The state of the users episode as a {episode: state} dict """
  34         states = cls.objects.filter(user=user)
  35
  36         if episodes is not None:
  37             states = states.filter(episode__in=episodes)
  38
  39         return dict(states.values_list('episode', 'action'))