diff --git a/bulkredditdownloader/site_downloaders/erome.py b/bulkredditdownloader/site_downloaders/erome.py index bb4373b..6d6e76d 100644 --- a/bulkredditdownloader/site_downloaders/erome.py +++ b/bulkredditdownloader/site_downloaders/erome.py @@ -25,23 +25,12 @@ class Erome(BaseDownloader): if not links: raise NotADownloadableLinkError('Erome parser could not find any links') - if len(links) == 1: - link = links.pop() - link = self._validate_url(link) - return [Resource(self.post, link)] - - else: - out = [] - for i, link in enumerate(links): - link = self._validate_url(link) - out.append(Resource(self.post, link)) - return out - - @staticmethod - def _validate_url(image): - if not re.match(r'https?://.*', image): - image = "https://" + image - return image + out = [] + for link in links: + if not re.match(r'https?://.*', link): + link = 'https://' + link + out.append(Resource(self.post, link)) + return out @staticmethod def _get_links(url: str) -> set[str]: diff --git a/bulkredditdownloader/tests/downloaders/test_erome.py b/bulkredditdownloader/tests/downloaders/test_erome.py index b6d1d54..9500cf8 100644 --- a/bulkredditdownloader/tests/downloaders/test_erome.py +++ b/bulkredditdownloader/tests/downloaders/test_erome.py @@ -10,25 +10,14 @@ from bulkredditdownloader.site_downloaders.erome import Erome @pytest.mark.online @pytest.mark.parametrize(('test_url', 'expected_urls'), ( - ('https://www.erome.com/a/vqtPuLXh', ( - 'https://s6.erome.com/365/vqtPuLXh/KH2qBT99.jpg', - 'https://s6.erome.com/365/vqtPuLXh/KH2qBT99_480p.mp4', - ) - ), + ('https://www.erome.com/a/vqtPuLXh', ('https://s6.erome.com/365/vqtPuLXh/KH2qBT99_480p.mp4',)), ('https://www.erome.com/a/ORhX0FZz', - ('https://s4.erome.com/355/ORhX0FZz/9IYQocM9.jpg', - 'https://s4.erome.com/355/ORhX0FZz/9IYQocM9_480p.mp4', - 'https://s4.erome.com/355/ORhX0FZz/9eEDc8xm.jpg', + ('https://s4.erome.com/355/ORhX0FZz/9IYQocM9_480p.mp4', 'https://s4.erome.com/355/ORhX0FZz/9eEDc8xm_480p.mp4', - 'https://s4.erome.com/355/ORhX0FZz/EvApC7Rp.jpg', 'https://s4.erome.com/355/ORhX0FZz/EvApC7Rp_480p.mp4', - 'https://s4.erome.com/355/ORhX0FZz/LruobtMs.jpg', 'https://s4.erome.com/355/ORhX0FZz/LruobtMs_480p.mp4', - 'https://s4.erome.com/355/ORhX0FZz/TJNmSUU5.jpg', 'https://s4.erome.com/355/ORhX0FZz/TJNmSUU5_480p.mp4', - 'https://s4.erome.com/355/ORhX0FZz/X11Skh6Z.jpg', 'https://s4.erome.com/355/ORhX0FZz/X11Skh6Z_480p.mp4', - 'https://s4.erome.com/355/ORhX0FZz/bjlTkpn7.jpg', 'https://s4.erome.com/355/ORhX0FZz/bjlTkpn7_480p.mp4') ), ))