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

   1 from django.core.management.base import BaseCommand
   2 from optparse import make_option
   3 from mygpo.api.models import Podcast
   4 from mygpo.data.models import RelatedPodcast
   5 from mygpo.data.podcast import calc_similar_podcasts
   6
   7 class Command(BaseCommand):
   8
   9     option_list = BaseCommand.option_list + (
  10         make_option('--max', action='store', type='int', dest='max', default=15, help="Maximum number of similar podcasts to calculate for each podcast."),
  11         )
  12
  13
  14     def handle(self, *args, **options):
  15
  16         max_related = options.get('max')
  17
  18         podcasts = Podcast.objects.all().order_by('id').only('id')
  19
  20         for podcast in podcasts.iterator():
  21             print podcast.id, podcast.title
  22
  23             l = calc_similar_podcasts(podcast)
  24
  25             RelatedPodcast.objects.filter(ref_podcast=podcast).delete()
  26             for (p, count) in l[:max_related]:
  27                 RelatedPodcast.objects.create(ref_podcast=podcast, rel_podcast=p, priority=count)
  28