nakidai
/
whoogle-search
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Only create ip card if main result div is found 

The ip address card that is created for searches like "my ip" only needs
to be created/inserted if a main result div id is found.

Fixes #735
main
Ben Busby 2022-04-26 15:18:29 -06:00
parent abc30d7da3
commit 62d7491936
No known key found for this signature in database
GPG Key ID: B9B7231E01D924A1
1 changed files with 18 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
app/utils/results.py
View File

@ -227,6 +227,8 @@ def add_ip_card(html_soup: BeautifulSoup, ip: str) -> BeautifulSoup:
BeautifulSoup BeautifulSoup
""" """
main_div = html_soup.select_one('#main')
if main_div:
# HTML IP card tag # HTML IP card tag
ip_tag = html_soup.new_tag('div') ip_tag = html_soup.new_tag('div')
ip_tag['class'] = 'ZINbbc xpd O9g5cc uUPGi' ip_tag['class'] = 'ZINbbc xpd O9g5cc uUPGi'
@ -246,7 +248,7 @@ def add_ip_card(html_soup: BeautifulSoup, ip: str) -> BeautifulSoup:
ip_tag.append(ip_text) ip_tag.append(ip_text)
# Insert the element at the top of the result list # Insert the element at the top of the result list
html_soup.select_one('#main').insert_before(ip_tag) main_div.insert_before(ip_tag)
return html_soup return html_soup