diff --git a/app/utils/results.py b/app/utils/results.py index 70a3696..2341c36 100644 --- a/app/utils/results.py +++ b/app/utils/results.py @@ -227,26 +227,28 @@ def add_ip_card(html_soup: BeautifulSoup, ip: str) -> BeautifulSoup: BeautifulSoup """ - # HTML IP card tag - ip_tag = html_soup.new_tag('div') - ip_tag['class'] = 'ZINbbc xpd O9g5cc uUPGi' + main_div = html_soup.select_one('#main') + if main_div: + # HTML IP card tag + ip_tag = html_soup.new_tag('div') + ip_tag['class'] = 'ZINbbc xpd O9g5cc uUPGi' - # For IP Address html tag - ip_address = html_soup.new_tag('div') - ip_address['class'] = 'kCrYT ip-address-div' - ip_address.string = ip + # For IP Address html tag + ip_address = html_soup.new_tag('div') + ip_address['class'] = 'kCrYT ip-address-div' + ip_address.string = ip - # Text below the IP address - ip_text = html_soup.new_tag('div') - ip_text.string = 'Your public IP address' - ip_text['class'] = 'kCrYT ip-text-div' + # Text below the IP address + ip_text = html_soup.new_tag('div') + ip_text.string = 'Your public IP address' + ip_text['class'] = 'kCrYT ip-text-div' - # Adding all the above html tags to the IP card - ip_tag.append(ip_address) - ip_tag.append(ip_text) + # Adding all the above html tags to the IP card + ip_tag.append(ip_address) + ip_tag.append(ip_text) - # Insert the element at the top of the result list - html_soup.select_one('#main').insert_before(ip_tag) + # Insert the element at the top of the result list + main_div.insert_before(ip_tag) return html_soup