From 752e82941a64290c94637e234f15e619a9f9b760 Mon Sep 17 00:00:00 2001
From: anonusr <97858880+an0nusr@users.noreply.github.com>
Date: Wed, 2 Feb 2022 03:17:39 -0800
Subject: [PATCH] Fix Next button in gallery download. (#65)

In the current code, if the next button is not found, there is nothing
stopping the call to (null).parent

This adjusts the call to check if the button is found, and then
also checks if the parent is found.
---
 furaffinity-dl.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/furaffinity-dl.py b/furaffinity-dl.py
index 584256d..7d7c5a7 100755
--- a/furaffinity-dl.py
+++ b/furaffinity-dl.py
@@ -203,12 +203,12 @@ while True:
         sleep(args.interval)
 
     if args.category != "favorites":
-        next_button = s.find('button', class_='button standard', text="Next").parent
-        if next_button is None:
+        next_button = s.find('button', class_='button standard', text="Next")
+        if next_button is None or next_button.parent is None:
             print('Unable to find next button')
             break
 
-        page_num = next_button.attrs['action'].split('/')[-2]
+        page_num = next_button.parent.attrs['action'].split('/')[-2]
 
         print('Downloading page', page_num, page_url)
     else: