From 80baab8de7e64e731eb300ade7afbc4474126976 Mon Sep 17 00:00:00 2001
From: Serene-Arc <serenical@gmail.com>
Date: Tue, 14 Sep 2021 13:47:46 +1000
Subject: [PATCH] Fix bug with different Vidble links

---
 bdfr/site_downloaders/vidble.py       |  8 +++++-
 tests/site_downloaders/test_vidble.py | 36 ++++++++++++++++-----------
 2 files changed, 28 insertions(+), 16 deletions(-)

diff --git a/bdfr/site_downloaders/vidble.py b/bdfr/site_downloaders/vidble.py
index 2f8f4f4..5cea0cb 100644
--- a/bdfr/site_downloaders/vidble.py
+++ b/bdfr/site_downloaders/vidble.py
@@ -22,7 +22,10 @@ class Vidble(BaseDownloader):
         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]
@@ -30,6 +33,9 @@ class Vidble(BaseDownloader):
 
     @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
index 1617bf1..0c5ebb2 100644
--- a/tests/site_downloaders/test_vidble.py
+++ b/tests/site_downloaders/test_vidble.py
@@ -33,6 +33,9 @@ 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)
@@ -40,21 +43,24 @@ def test_get_links(test_url: str, expected: set[str]):
 
 
 @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()