From 08aa1ab8f157bc7fbe59f270b2105b7540c02972 Mon Sep 17 00:00:00 2001 From: MoistCat <82777796+Moist-Cat@users.noreply.github.com> Date: Thu, 29 Dec 2022 17:17:34 -0500 Subject: [PATCH] Handle missing result div in filter (#911) Changed "find_all()[0]" for find; which yields only one result. Added check to ensure result_div exists before searching for results. --- .gitignore | 4 ++++ app/filter.py | 10 ++++++---- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/.gitignore b/.gitignore index 02b4e03..181094f 100644 --- a/.gitignore +++ b/.gitignore @@ -19,3 +19,7 @@ dist/ # env whoogle.env + +# vim +*~ +*.swp diff --git a/app/filter.py b/app/filter.py index 92e5a9d..1fee1dc 100644 --- a/app/filter.py +++ b/app/filter.py @@ -610,13 +610,15 @@ class Filter: # get some tags that are unchanged between mobile and pc versions cor_suggested = soup.find_all('table', attrs={'class': "By0U9"}) - next_pages = soup.find_all('table', attrs={'class': "uZgmoc"})[0] + next_pages = soup.find('table', attrs={'class': "uZgmoc"}) results = [] # find results div - results_div = soup.find_all('div', attrs={'class': "nQvrDb"})[0] - # find all the results - results_all = results_div.find_all('div', attrs={'class': "lIMUZd"}) + results_div = soup.find('div', attrs={'class': "nQvrDb"}) + # find all the results (if any) + results_all = [] + if results_div: + results_all = results_div.find_all('div', attrs={'class': "lIMUZd"}) for item in results_all: urls = item.find('a')['href'].split('&imgrefurl=')