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

fix: test not pass

Browse Source
This commit is contained in:
YadominJinta 2021-10-21 01:48:02 +08:00
parent fd49221df9
commit d2821fe67b
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app/filter.py
View File

@ -50,11 +50,12 @@ class Filter:
self.nojs = config['nojs'] if 'nojs' in config else False self.nojs = config['nojs'] if 'nojs' in config else False
self.new_tab = config['new_tab'] if 'new_tab' in config else False self.new_tab = config['new_tab'] if 'new_tab' in config else False
self.alt_redirect = config['alts'] if 'alts' in config else False self.alt_redirect = config['alts'] if 'alts' in config else False
self.block_title = config['block_title'] if 'block_title' in config else ''
self.block_url = config['block_url'] if 'block_url' in config else ''
self.mobile = mobile self.mobile = mobile
self.user_key = user_key self.user_key = user_key
self.main_divs = ResultSet('') self.main_divs = ResultSet('')
self._elements = 0 self._elements = 0
self.config = config
def __getitem__(self, name): def __getitem__(self, name):
return getattr(self, name) return getattr(self, name)
@ -139,9 +140,9 @@ class Filter:
def remove_block_titles(self) -> None: def remove_block_titles(self) -> None:
if not self.main_divs: if not self.main_divs:
return return
if self.config.block_title == '': if self.block_title == '':
return return
block_title = re.compile(self.config.block_title) block_title = re.compile(self.block_title)
for div in [_ for _ in self.main_divs.find_all('div', recursive=True)]: for div in [_ for _ in self.main_divs.find_all('div', recursive=True)]:
block_divs = [_ for _ in div.find_all('h3', recursive=True) block_divs = [_ for _ in div.find_all('h3', recursive=True)
if block_title.search(_.text) is not None] if block_title.search(_.text) is not None]
@ -150,9 +151,9 @@ class Filter:
def remove_block_url(self) -> None: def remove_block_url(self) -> None:
if not self.main_divs: if not self.main_divs:
return return
if self.config.block_url == '': if self.block_url == '':
return return
block_url = re.compile(self.config.block_url) block_url = re.compile(self.block_url)
for div in [_ for _ in self.main_divs.find_all('div', recursive=True)]: for div in [_ for _ in self.main_divs.find_all('div', recursive=True)]:
block_divs = [_ for _ in div.find_all('a', recursive=True) block_divs = [_ for _ in div.find_all('a', recursive=True)
if block_url.search(_.attrs['href']) is not None] if block_url.search(_.attrs['href']) is not None]