From db10084faa2b639480cb02636db67da1cc32c009 Mon Sep 17 00:00:00 2001
From: James Martindale <11380394+jkmartindale@users.noreply.github.com>
Date: Sun, 9 Aug 2020 14:57:01 -0600
Subject: [PATCH] Fix downloading multiple Favorites gallery pages

---
 furaffinity-dl.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/furaffinity-dl.py b/furaffinity-dl.py
index 2226c4e..85701d7 100755
--- a/furaffinity-dl.py
+++ b/furaffinity-dl.py
@@ -183,7 +183,18 @@ while True:
     for img in s.findAll('figure'):
         download(img.find('a').attrs.get('href'))
 
-    page_num += 1
+    # Favorites galleries use a weird timestamp system, so grab the next "page" from the Next button
+    if args.category == 'favorites':
+        next_button = s.find('a', class_='button standard right')
+        if next_button is None:
+            break
+
+        # URL looks something like /favorites/:username/:timestamp/next
+        # Splitting on the username is more robust to future URL changes
+        page_num = next_button.attrs['href'].split(args.username + '/')[-1]
+    else:
+        page_num += 1
+    
     print('Downloading page', page_num)
 
 print('Finished downloading')