Merge pull request #521 from Serene-Arc/bug_fix_518

Fix bug with different Vidble links
Serene-Arc · Sep 14, 2021 · e004ccd · e004ccd
2 parents 668fe80 + 80baab8
commit e004ccd
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 16 deletions.
diff --git a/bdfr/site_downloaders/vidble.py b/bdfr/site_downloaders/vidble.py
@@ -22,14 +22,20 @@ def __init__(self, post: Submission):
         super().__init__(post)
 
     def find_resources(self, authenticator: Optional[SiteAuthenticator] = None) -> list[Resource]:
-        res = self.get_links(self.post.url)
+        try:
+            res = self.get_links(self.post.url)
+        except AttributeError:
+            raise SiteDownloaderError(f'Could not read page at {self.post.url}')
         if not res:
             raise SiteDownloaderError(rf'No resources found at {self.post.url}')
         res = [Resource(self.post, r, Resource.retry_download(r)) for r in res]
         return res
 
     @staticmethod
     def get_links(url: str) -> set[str]:
+        if not re.search(r'vidble.com/(show/|album/|watch\?v)', url):
+            url = re.sub(r'/(\w*?)$', r'/show/\1', url)
+
         page = requests.get(url)
         soup = bs4.BeautifulSoup(page.text, 'html.parser')
         content_div = soup.find('div', attrs={'id': 'ContentPlaceHolder1_divContent'})

diff --git a/tests/site_downloaders/test_vidble.py b/tests/site_downloaders/test_vidble.py
@@ -33,28 +33,34 @@ def test_change_med_url(test_url: str, expected: str):
     ('https://vidble.com/watch?v=0q4nWakqM6kzQWxlePD8N62Dsflev0N9', {
         'https://www.vidble.com/0q4nWakqM6kzQWxlePD8N62Dsflev0N9.mp4',
     }),
+    ('https://www.vidble.com/pHuwWkOcEb', {
+        'https://www.vidble.com/pHuwWkOcEb.jpg',
+    }),
 ))
 def test_get_links(test_url: str, expected: set[str]):
     results = Vidble.get_links(test_url)
     assert results == expected
 
 
 @pytest.mark.parametrize(('test_url', 'expected_hashes'), (
-        ('https://www.vidble.com/show/UxsvAssYe5', {
-            '0ef2f8e0e0b45936d2fb3e6fbdf67e28',
-        }),
-        ('https://vidble.com/show/RDFbznUvcN', {
-            'c2dd30a71e32369c50eed86f86efff58',
-        }),
-        ('https://vidble.com/album/h0jTLs6B', {
-            '3b3cba02e01c91f9858a95240b942c71',
-            'dd6ecf5fc9e936f9fb614eb6a0537f99',
-            'b31a942cd8cdda218ed547bbc04c3a27',
-            '6f77c570b451eef4222804bd52267481',
-        }),
-        ('https://vidble.com/watch?v=0q4nWakqM6kzQWxlePD8N62Dsflev0N9', {
-            'cebe9d5f24dba3b0443e5097f160ca83',
-        }),
+    ('https://www.vidble.com/show/UxsvAssYe5', {
+        '0ef2f8e0e0b45936d2fb3e6fbdf67e28',
+    }),
+    ('https://vidble.com/show/RDFbznUvcN', {
+        'c2dd30a71e32369c50eed86f86efff58',
+    }),
+    ('https://vidble.com/album/h0jTLs6B', {
+        '3b3cba02e01c91f9858a95240b942c71',
+        'dd6ecf5fc9e936f9fb614eb6a0537f99',
+        'b31a942cd8cdda218ed547bbc04c3a27',
+        '6f77c570b451eef4222804bd52267481',
+    }),
+    ('https://vidble.com/watch?v=0q4nWakqM6kzQWxlePD8N62Dsflev0N9', {
+        'cebe9d5f24dba3b0443e5097f160ca83',
+    }),
+    ('https://www.vidble.com/pHuwWkOcEb', {
+        '585f486dd0b2f23a57bddbd5bf185bc7',
+    }),
 ))
 def test_find_resources(test_url: str, expected_hashes: set[str]):
     mock_download = Mock()