diff options
Diffstat (limited to 'gallery_dl/extractor/nsfwalbum.py')
| -rw-r--r-- | gallery_dl/extractor/nsfwalbum.py | 24 |
1 files changed, 20 insertions, 4 deletions
diff --git a/gallery_dl/extractor/nsfwalbum.py b/gallery_dl/extractor/nsfwalbum.py index 9c4d686..be736d1 100644 --- a/gallery_dl/extractor/nsfwalbum.py +++ b/gallery_dl/extractor/nsfwalbum.py @@ -43,11 +43,27 @@ class NsfwalbumAlbumExtractor(GalleryExtractor): def images(self, page): iframe = self.root + "/iframe_image.php?id=" backend = self.root + "/backend.php" + retries = self._retries + for image_id in text.extract_iter(page, 'data-img-id="', '"'): - spirit = self._annihilate(text.extract(self.request( - iframe + image_id).text, 'giraffe.annihilate("', '"')[0]) - params = {"spirit": spirit, "photo": image_id} - data = self.request(backend, params=params).json() + spirit = None + tries = 0 + + while tries <= retries: + try: + if not spirit: + spirit = self._annihilate(text.extract( + self.request(iframe + image_id).text, + 'giraffe.annihilate("', '"')[0]) + params = {"spirit": spirit, "photo": image_id} + data = self.request(backend, params=params).json() + break + except Exception: + tries += 1 + else: + self.log.warning("Unable to fetch image %s", image_id) + continue + yield data[0], { "id" : text.parse_int(image_id), "width" : text.parse_int(data[1]), |