mygpo/directory/management/commands/set-example-podcasts.py

   1 from datetime import datetime
   2 import fileinput
   3
   4 from django.core.management.base import BaseCommand
   5
   6 from couchdbkit.exceptions import ResourceNotFound
   7
   8 from mygpo.podcasts.models import Podcast
   9 from mygpo.directory.models import ExamplePodcasts
  10
  11
  12 EXAMPLES_DOCID = 'example_podcasts'
  13
  14 class Command(BaseCommand):
  15
  16     def handle(self, *args, **options):
  17
  18         urls = list(map(str.strip, fileinput.input(args)))
  19
  20         try:
  21             examples = ExamplePodcasts.get(EXAMPLES_DOCID)
  22         except ResourceNotFound:
  23             examples = ExamplePodcasts()
  24             examples._id = EXAMPLES_DOCID
  25
  26         podcasts = Podcast.objects.filter(urls__url__in=urls)
  27         examples.podcast_ids = [podcast.id for podcast in podcasts]
  28         examples.updated = datetime.utcnow()
  29         examples.save()