mygpo/data/management/commands/feed-downloader.py

   1 from itertools import islice
   2 import traceback
   3 from optparse import make_option
   4
   5 from restkit.errors import RequestFailed
   6
   7 from mygpo.maintenance.management.podcastcmd import PodcastCommand
   8 from mygpo.data.feeddownloader import PodcastUpdater
   9
  10 try:
  11     from gevent import monkey
  12     monkey.patch_all()
  13 except ImportError:
  14     pass
  15
  16 import socket
  17 socket.setdefaulttimeout(300)
  18
  19
  20 class Command(PodcastCommand):
  21
  22     option_list = PodcastCommand.option_list + (
  23         make_option('--list-only', action='store_true', dest='list',
  24             default=False, help="Don't update anything, just list podcasts "),
  25         )
  26
  27
  28     def handle(self, *args, **options):
  29
  30         queue = self.get_podcasts(*args, **options)
  31
  32         max_podcasts = options.get('max')
  33         if max_podcasts:
  34             queue = islice(queue, 0, max_podcasts)
  35
  36         if options.get('list'):
  37             for podcast in queue:
  38                 print podcast
  39
  40         else:
  41             print 'Updating podcasts...'
  42
  43             updater = PodcastUpdater()
  44             for podcast in updater.update_queue(queue):
  45                 print podcast