mygpo/directory/models.py

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