X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/29d7fc838e192b6a18dd31d81fe9ab1bc37b6cbb..d8990e1ab64fe3bf453d06fab8e983d25f2df2c4:/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 854f3bbe..cc0ef31c 100644 --- a/apps/cover/management/commands/refresh_covers.py +++ b/apps/cover/management/commands/refresh_covers.py @@ -20,27 +20,29 @@ class Command(BaseCommand): def handle(self, *args, **options): from_id = options.get('from_id', 1) - for image in Image.objects.filter(id__gte=from_id).exclude(book=None).order_by('id'): + 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 - if '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: + 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 + continue try: - flickr_data = get_flickr_data(image.source_url) - except FlickrError as e: - print 'Flickr analysis failed: %s' % e + t = URLOpener().open(flickr_url).read() + except urllib.URLError: + print 'Broken download url' + except IOError: + print 'Connection failed' 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 - continue - try: - t = URLOpener().open(image.download_url).read() - except urllib.URLError: - print 'Broken download url' - except IOError: - print 'Connection failed' - else: - image.download_url = flickr_data['download_url'] - image.file.save(image.file.name, ContentFile(t)) - image.save() + image.download_url = flickr_url + image.file.save(image.file.name, ContentFile(t)) + image.save()