fix mistake.

This commit is contained in:
Jason Schwarzenberger 2020-11-12 17:15:29 +13:00
parent 9318627f1b
commit 64ced635cc

View File

@ -248,7 +248,7 @@ class Category(_Base):
self.tz = tz self.tz = tz
self.category_url = url self.category_url = url
def _filter_links(self, links, excludes=None): def _filter_links(self, links, category_url, excludes=None):
links = list(filter(None, [link if link.startswith(category_url) else None for link in links])) links = list(filter(None, [link if link.startswith(category_url) else None for link in links]))
links = list(filter(None, [link if link != category_url else None for link in links])) links = list(filter(None, [link if link != category_url else None for link in links]))
links = list(set(links)) links = list(set(links))
@ -264,7 +264,7 @@ class Category(_Base):
links = soup.find_all('a', href=True) links = soup.find_all('a', href=True)
links = [link.get('href') for link in links] links = [link.get('href') for link in links]
links = [f"{base_url}{link}" if link.startswith('/') else link for link in links] links = [f"{base_url}{link}" if link.startswith('/') else link for link in links]
links = self._filter_links(links, excludes) links = self._filter_links(links, category_url, excludes)
return links return links
def feed(self, excludes=None): def feed(self, excludes=None):