Clean "Show more results" of all site blocks (#646)

main
DUO Labs 2022-02-08 12:57:00 -05:00 committed by GitHub
parent 11099f7b1d
commit 502067addc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 1 deletions

View File

@ -119,9 +119,20 @@ class Filter:
header = soup.find('header')
if header:
header.decompose()
self.remove_site_blocks(soup)
return soup
def remove_site_blocks(self, soup) -> None:
if not self.config.block:
return
search_string = ' '.join(['-site:' +
_ for _ in self.config.block.split(',')])
selected = soup.body.findAll(text=re.compile(search_string))
for result in selected:
result.string.replace_with(result.string.replace(
search_string, ''))
def remove_ads(self) -> None:
"""Removes ads found in the list of search result divs