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

   1 import time
   2 import urllib.request, urllib.error, urllib.parse
   3 from optparse import make_option
   4
   5 from mygpo.data import delicious
   6 from mygpo.maintenance.management.podcastcmd import PodcastCommand
   7
   8
   9 SOURCE = "delicious"
  10
  11
  12 class Command(PodcastCommand):
  13     """
  14     Adds tags from the webservice delicious.com to podcasts
  15
  16     Podcasts are specified either by URL or the --toplist and --random
  17     parameter. The delicious webservice is queried for the podcasts' websites.
  18     The returned tags are added to the podcasts for the 'delicious' source.
  19     """
  20
  21     def handle(self, *args, **options):
  22
  23         fetch_queue = self.get_podcasts()
  24
  25         for p in fetch_queue:
  26             if not p or not p.link:
  27                 continue
  28
  29             # we don't want to spam delicious
  30             time.sleep(1)
  31
  32             try:
  33                 f = urllib.request.urlopen(p.link)
  34             except urllib.error.HTTPError:
  35                 continue
  36
  37             tags = delicious.get_tags(f.url)
  38
  39             self.update(podcast=p, tags=tags)
  40
  41     def update(self, podcast, tags):
  42         podcast.tags[SOURCE] = tags
  43         podcast.save()