From d2821fe67b56a270ca89c8e99b3739f7f1062f7d Mon Sep 17 00:00:00 2001 From: YadominJinta Date: Thu, 21 Oct 2021 01:48:02 +0800 Subject: [PATCH] fix: test not pass --- app/filter.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/app/filter.py b/app/filter.py index 37cb957..43ec3f5 100644 --- a/app/filter.py +++ b/app/filter.py @@ -50,11 +50,12 @@ class Filter: self.nojs = config['nojs'] if 'nojs' 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.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.user_key = user_key self.main_divs = ResultSet('') self._elements = 0 - self.config = config def __getitem__(self, name): return getattr(self, name) @@ -139,9 +140,9 @@ class Filter: def remove_block_titles(self) -> None: if not self.main_divs: return - if self.config.block_title == '': + if self.block_title == '': 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)]: block_divs = [_ for _ in div.find_all('h3', recursive=True) if block_title.search(_.text) is not None] @@ -150,9 +151,9 @@ class Filter: def remove_block_url(self) -> None: if not self.main_divs: return - if self.config.block_url == '': + if self.block_url == '': 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)]: block_divs = [_ for _ in div.find_all('a', recursive=True) if block_url.search(_.attrs['href']) is not None]