mygpo/data/management/commands/update-related-podcasts.py

   1 from optparse import make_option
   2 from operator import itemgetter
   3
   4 from django.core.management.base import BaseCommand
   5
   6 from mygpo.decorators import repeat_on_conflict
   7 from mygpo.core.models import Podcast
   8 from mygpo.data.podcast import calc_similar_podcasts
   9 from mygpo.utils import progress
  10 from mygpo.db.couchdb.podcast import podcast_count, all_podcasts
  11
  12
  13 class Command(BaseCommand):
  14
  15     option_list = BaseCommand.option_list + (
  16         make_option('--max', action='store', type='int', dest='max', default=15, help="Maximum number of similar podcasts to calculate for each podcast."),
  17         )
  18
  19
  20     def handle(self, *args, **options):
  21
  22         get_podcast = itemgetter(0)
  23
  24         max_related = options.get('max')
  25
  26         podcasts = all_podcasts()
  27         total = podcast_count()
  28
  29         for (n, podcast) in enumerate(podcasts):
  30
  31             l = calc_similar_podcasts(podcast)[:max_related]
  32
  33             related = map(get_podcast, l)
  34
  35             @repeat_on_conflict(['podcast'])
  36             def _update(podcast, related):
  37                 podcast.related_podcasts = related
  38                 podcast.save()
  39
  40             _update(podcast=podcast, related=related)
  41
  42             progress(n+1, total)