apps/cover/management/commands/refresh_covers.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 import urllib2 as urllib
   7 from optparse import make_option
   8
   9 from django.core.files.base import ContentFile
  10 from django.core.management import BaseCommand
  11
  12 from cover.models import Image
  13 from cover.utils import get_flickr_data, URLOpener, FlickrError
  14
  15
  16 class Command(BaseCommand):
  17     option_list = BaseCommand.option_list + (
  18         make_option('--from', dest='from_id', type=int, default=1),
  19     )
  20
  21     def handle(self, *args, **options):
  22         from_id = options.get('from_id', 1)
  23         for image in Image.objects.filter(id__gte=from_id).exclude(book=None).order_by('id'):
  24             print image.id
  25             if image.source_url and 'flickr.com' in image.source_url:
  26                 try:
  27                     flickr_data = get_flickr_data(image.source_url)
  28                 except FlickrError as e:
  29                     print 'Flickr analysis failed: %s' % e
  30                 else:
  31                     flickr_url = flickr_data['download_url']
  32                     if flickr_url != image.download_url:
  33                         same_url = Image.objects.filter(download_url=flickr_url)
  34                         if same_url:
  35                             print 'Download url already present in image %s' % same_url.get().id
  36                             continue
  37                     try:
  38                         t = URLOpener().open(image.download_url).read()
  39                     except urllib.URLError:
  40                         print 'Broken download url'
  41                     except IOError:
  42                         print 'Connection failed'
  43                     else:
  44                         image.download_url = flickr_data['download_url']
  45                         image.file.save(image.file.name, ContentFile(t))
  46                         image.save()