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

Add tests for downloader Erome

Browse Source
This commit is contained in:
Serene-Arc
2021-02-27 10:35:43 +10:00
committed by Ali Parlakci
parent 2a0dd4f6ac
commit 125b78a348
2 changed files with 57 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
bulkredditdownloader/site_downloaders/erome.py
View File

@@ -23,13 +23,13 @@ class Erome(BaseDownloader):
def find_resources(self, authenticator: Optional[SiteAuthenticator] = None) -> list[Resource]:
try:
images = self._get_links(self.post.url)
images = set(self._get_links(self.post.url))
except urllib.error.HTTPError:
raise NotADownloadableLinkError("Not a downloadable link")
if len(images) == 1:
image = images[0]
image = images.pop()
if not re.match(r'https?://.*', image):
image = "https://" + image
return [Resource(self.post, image)]
@@ -39,7 +39,7 @@ class Erome(BaseDownloader):
for i, image in enumerate(images):
if not re.match(r'https?://.*', image):
image = "https://" + image
out.append(Resource(self.post, image))
out.append(Resource(self.post, image))
return out
@staticmethod