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.
main
MoistCat 2022-12-29 17:17:34 -05:00 committed by GitHub
parent 7041b43db9
commit 08aa1ab8f1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 4 deletions

4
.gitignore vendored
View File

@ -19,3 +19,7 @@ dist/
# env
whoogle.env
# vim
*~
*.swp

View File

@ -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=')