Add title to cover attribution.
[redakcja.git] / src / cover / management / commands / refresh_covers.py
index cc0ef31..5732563 100644 (file)
@@ -1,11 +1,7 @@
-# -*- coding: utf-8 -*-
-#
 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 import urllib2 as urllib
-from optparse import make_option
-
 from django.core.files.base import ContentFile
 from django.core.management import BaseCommand
 
@@ -14,34 +10,33 @@ from cover.utils import get_flickr_data, URLOpener, FlickrError
 
 
 class Command(BaseCommand):
-    option_list = BaseCommand.option_list + (
-        make_option('--from', dest='from_id', type=int, default=1),
-    )
+    def add_arguments(self, parser):
+        parser.add_argument('--from', dest='from_id', type=int, default=1)
 
     def handle(self, *args, **options):
         from_id = options.get('from_id', 1)
         images = Image.objects.filter(id__gte=from_id).exclude(book=None).order_by('id')
         images = images.filter(source_url__contains='flickr.com').exclude(download_url__endswith='_o.jpg')
         for image in images:
-            print image.id
+            print(image.id)
             try:
                 flickr_data = get_flickr_data(image.source_url)
-                print flickr_data
+                print(flickr_data)
             except FlickrError as e:
-                print 'Flickr analysis failed: %s' % e
+                print('Flickr analysis failed: %s' % e)
             else:
                 flickr_url = flickr_data['download_url']
                 if flickr_url != image.download_url:
                     same_url = Image.objects.filter(download_url=flickr_url)
                     if same_url:
-                        print 'Download url already present in image %s' % same_url.get().id
+                        print('Download url already present in image %s' % same_url.get().id)
                         continue
                 try:
                     t = URLOpener().open(flickr_url).read()
                 except urllib.URLError:
-                    print 'Broken download url'
+                    print('Broken download url')
                 except IOError:
-                    print 'Connection failed'
+                    print('Connection failed')
                 else:
                     image.download_url = flickr_url
                     image.file.save(image.file.name, ContentFile(t))