mygpo/directory/models.py

   1 from __future__ import unicode_literals
   2
   3 from django.core.cache import cache
   4
   5 from couchdbkit.ext.django.schema import *
   6
   7 from mygpo.podcasts.models import Podcast
   8 from mygpo.utils import iterate_together
   9 from mygpo.core.proxy import DocumentABCMeta
  10
  11
  12 class Category(Document):
  13
  14     __metaclass__ = DocumentABCMeta
  15
  16     label = StringProperty(required=True)
  17     updated = DateTimeProperty(required=True)
  18     spellings = StringListProperty()
  19     podcasts = ListProperty()
  20
  21
  22     def merge_podcasts(self, podcasts):
  23         """
  24         Merges some entries into the current category.
  25         """
  26
  27         key = lambda e: e.podcast
  28
  29         podcasts = sorted(podcasts, key=key)
  30         self.podcasts = sorted(self.podcasts, key=key)
  31
  32         new_entries = []
  33
  34         for e1, e2 in iterate_together([self.podcasts, podcasts], key):
  35             if e1 is None:
  36                 new_entries.append(e2)
  37
  38             elif e2 is None:
  39                 new_entries.append(e1)
  40
  41             else:
  42                 new_entries.append( max(e1, e2) )
  43
  44         self.podcasts = new_entries
  45
  46
  47     # called from within a template where we can't pass parameters
  48     def get_podcasts_more(self, start=0, end=40):
  49         return self.get_podcasts(start, end)
  50
  51
  52     def get_podcasts(self, start=0, end=10):
  53         cache_id = 'category-%s-%d-%d' % (self._id, start, end)
  54
  55         podcasts = cache.get(cache_id)
  56         if podcasts:
  57             return podcasts
  58
  59         ids = self.podcasts[start:end]
  60
  61         # TODO: this should not be needed anymore after migration
  62         if ids and not isinstance(ids[0], unicode):
  63             return []
  64
  65         podcasts = Podcast.objects.filter(id__in=ids)
  66         cache.set(cache_id, podcasts)
  67
  68         return podcasts
  69
  70
  71     def get_weight(self):
  72         return getattr(self, '_weight', len(self.podcasts))
  73
  74
  75     def get_tags(self):
  76         return self.spellings + [self.label]
  77
  78     def __repr__(self):
  79         return '%s (+%d variants)' % (self.label, len(self.spellings))
  80
  81
  82 class ExamplePodcasts(Document):
  83     podcast_ids  = StringListProperty()
  84     updated      = DateTimeProperty()