ModerateWinGuy/BDFR_Web
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add test for site downloader gallery

Browse Source
This commit is contained in:
Serene-Arc
2021-02-15 17:45:41 +10:00
committed by Ali Parlakci
parent 1b40b16970
commit bb85fb8934
2 changed files with 26 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
bulkredditdownloader/site_downloaders/gallery.py
View File

@@ -6,7 +6,7 @@ import logging
import requests
from praw.models import Submission
from bulkredditdownloader.errors import ResourceNotFound, NotADownloadableLinkError
from bulkredditdownloader.errors import NotADownloadableLinkError, ResourceNotFound
from bulkredditdownloader.site_downloaders.base_downloader import BaseDownloader
logger = logging.getLogger(__name__)
@@ -33,7 +33,7 @@ class Gallery(BaseDownloader):
except KeyError:
continue
return [self._download_album(images)]
return self._download_album(images)
@staticmethod
def _get_data(link: str) -> dict:
@@ -62,7 +62,6 @@ class Gallery(BaseDownloader):
def _download_album(self, images: dict):
out = []
for i, image in enumerate(images):
out.append(self._download_resource(image['url']))
for image_key in images.keys():
out.append(self._download_resource(images[image_key]['url']))
return out