mygpo/search/util.py

   1 from django.db.models import Count
   2 from mygpo.api.models import Subscription, Podcast
   3 from mygpo.data.models import PodcastTag
   4
   5
   6 def podcast_group_entry(group, subscriber_count=None):
   7     from mygpo.search.models import SearchEntry
   8
   9     if not subscriber_count:
  10         subscriber_count = Subscription.objects.filter(podcast__group=group).values('user').distinct().count()
  11
  12     entry = SearchEntry()
  13     entry.text = group.title
  14     entry.obj_type = 'podcast_group'
  15     entry.obj_id = group.id
  16
  17     podcasts = Podcast.objects.filter(group=group)
  18     entry.tags = tag_string(PodcastTag.objects.filter(podcast__in=podcasts))
  19
  20     entry.priority = subscriber_count
  21
  22     return entry
  23
  24
  25 def podcast_entry(podcast, subscriber_count=None):
  26     from mygpo.search.models import SearchEntry
  27
  28     if not subscriber_count:
  29         subscriber_count = Subscription.objects.filter(podcast=podcast).values('user').distinct().count()
  30
  31     entry = SearchEntry()
  32     entry.text = podcast.title
  33     entry.obj_type = 'podcast'
  34     entry.obj_id = podcast.id
  35
  36     entry.tags = tag_string(PodcastTag.objects.filter(podcast=podcast))
  37
  38     entry.priority = subscriber_count
  39
  40     return entry
  41
  42
  43 def tag_string(tags, max_length=200):
  44     """
  45     returns a string of the most-assigned tags
  46
  47     tags is expected to be a PodcastTag QuerySet
  48     """
  49     tags = PodcastTag.objects.top_tags(tags)
  50     return ','.join([t['tag'] for t in tags])[:max_length]
  51