mygpo/maintenance/management/podcastcmd.py

   1 from itertools import islice, chain
   2 from optparse import make_option
   3 import random
   4
   5 from django.core.management.base import BaseCommand
   6
   7 from mygpo.podcasts.models import Podcast
   8
   9
  10 class PodcastCommand(BaseCommand):
  11     """ command that operates on a list of podcasts specified by parameters """
  12
  13
  14     def add_arguments(self, parser):
  15         parser.add_argument('--toplist', action='store_true', dest='toplist',
  16             default=False, help="Update all entries from the Toplist."),
  17
  18         parser.add_argument('--update-new', action='store_true', dest='new',
  19             default=False, help="Update all podcasts with new Episodes"),
  20
  21         parser.add_argument('--max', action='store', dest='max', type='int',
  22             default=0, help="Set how many feeds should be updated at maximum"),
  23
  24         parser.add_argument('--random', action='store_true', dest='random',
  25             default=False, help="Update random podcasts, best used with --max"),
  26
  27         parser.add_argument('--next', action='store_true', dest='next',
  28             default=False, help="Podcasts that are due to be updated next"),
  29
  30
  31     def get_podcasts(self, *args, **options):
  32         return chain.from_iterable(self._get_podcasts(*args, **options))
  33
  34
  35     def _get_podcasts(self, *args, **options):
  36
  37         max_podcasts = options.get('max')
  38
  39         if options.get('toplist'):
  40             yield (p.url for p in self.get_toplist(max_podcasts))
  41
  42         if options.get('new'):
  43             query = Podcast.objects.filter(episode__title__isnull=True,
  44                                            episode__outdated=False)
  45             podcasts = query.distinct('id')[:max_podcasts]
  46             random.shuffle(podcasts)
  47             yield (p.url for p in podcasts)
  48
  49         if options.get('random'):
  50             podcasts = random_podcasts()
  51             yield (p.url for p in podcasts)
  52
  53         if options.get('next'):
  54             podcasts = Podcast.objects.all().order_by_next_update()[:max_podcasts]
  55             yield (p.url for p in podcasts)
  56
  57
  58         if args:
  59             yield args
  60
  61         if options.get('urls'):
  62             yield options.get('urls')
  63
  64         if not args and not options.get('toplist') and not options.get('new') \
  65                     and not options.get('random')  and not options.get('next'):
  66             query = Podcast.objects.order_by('last_update')
  67             podcasts = query.select_related('urls')[:max_podcasts]
  68             yield (p.url for p in podcasts)
  69
  70
  71     def get_toplist(self, max_podcasts=100):
  72         return Podcast.objects.all().toplist()[:max_podcasts]
  73
  74
  75 def random_podcasts():
  76     while True:
  77         yield Podcast.objects.all().random().first()