prevent adding new images with existing sources

[redakcja.git] / apps / cover / management / commands / refresh_covers.py

diff --git a/apps/cover/management/commands/refresh_covers.py b/apps/cover/management/commands/refresh_covers.py
index 854f3bb..2c78db9 100644 (file)
--- a/apps/cover/management/commands/refresh_covers.py
+++ b/apps/cover/management/commands/refresh_covers.py
@@ -22,9 +22,10 @@ class Command(BaseCommand):
         from_id = options.get('from_id', 1)
         for image in Image.objects.filter(id__gte=from_id).exclude(book=None).order_by('id'):
             print image.id
-            if 'flickr.com' in image.source_url:
+            if image.source_url and 'flickr.com' in image.source_url:
                 try:
                     flickr_data = get_flickr_data(image.source_url)
+                    print flickr_data
                 except FlickrError as e:
                     print 'Flickr analysis failed: %s' % e
                 else:
@@ -35,12 +36,12 @@ class Command(BaseCommand):
                             print 'Download url already present in image %s' % same_url.get().id
                             continue
                     try:
-                        t = URLOpener().open(image.download_url).read()
+                        t = URLOpener().open(flickr_url).read()
                     except urllib.URLError:
                         print 'Broken download url'
                     except IOError:
                         print 'Connection failed'
                     else:
-                        image.download_url = flickr_data['download_url']
+                        image.download_url = flickr_url
                         image.file.save(image.file.name, ContentFile(t))
                         image.save()