forked from tanner/qotnews
avoid duplicate articles listed on the category page
This commit is contained in:
parent
16b59f6c67
commit
9fff1b9e46
|
@ -16,6 +16,7 @@ from utils import clean
|
||||||
|
|
||||||
USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:77.0) Gecko/20100101 Firefox/77.0'
|
USER_AGENT = 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:77.0) Gecko/20100101 Firefox/77.0'
|
||||||
|
|
||||||
|
|
||||||
def unix(date_str):
|
def unix(date_str):
|
||||||
date_tzfix = date_str
|
date_tzfix = date_str
|
||||||
if ":" == date_tzfix[-3]:
|
if ":" == date_tzfix[-3]:
|
||||||
|
@ -33,6 +34,7 @@ def unix(date_str):
|
||||||
pass
|
pass
|
||||||
return 0
|
return 0
|
||||||
|
|
||||||
|
|
||||||
def xml(route, ref=None):
|
def xml(route, ref=None):
|
||||||
try:
|
try:
|
||||||
headers = {'User-Agent': USER_AGENT, 'X-Forwarded-For': '66.249.66.1'}
|
headers = {'User-Agent': USER_AGENT, 'X-Forwarded-For': '66.249.66.1'}
|
||||||
|
@ -46,6 +48,7 @@ def xml(route, ref=None):
|
||||||
logging.error('Problem hitting URL: {}'.format(str(e)))
|
logging.error('Problem hitting URL: {}'.format(str(e)))
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
|
||||||
def parse_extruct(s, data):
|
def parse_extruct(s, data):
|
||||||
for rdfa in data['rdfa']:
|
for rdfa in data['rdfa']:
|
||||||
for key, props in rdfa.items():
|
for key, props in rdfa.items():
|
||||||
|
@ -94,6 +97,7 @@ def parse_extruct(s, data):
|
||||||
|
|
||||||
return s
|
return s
|
||||||
|
|
||||||
|
|
||||||
class Sitemap:
|
class Sitemap:
|
||||||
def __init__(self, url):
|
def __init__(self, url):
|
||||||
self.sitemap_url = url
|
self.sitemap_url = url
|
||||||
|
@ -104,43 +108,8 @@ class Sitemap:
|
||||||
soup = BeautifulSoup(markup, features='lxml')
|
soup = BeautifulSoup(markup, features='lxml')
|
||||||
articles = soup.find('urlset').findAll('url')
|
articles = soup.find('urlset').findAll('url')
|
||||||
articles = list(filter(None, [a if a.find('lastmod') is not None else None for a in articles]))
|
articles = list(filter(None, [a if a.find('lastmod') is not None else None for a in articles]))
|
||||||
return [x.find('loc').text for x in articles] or []
|
links = [x.find('loc').text for x in articles] or []
|
||||||
|
links = list(set(links))
|
||||||
def story(self, ref):
|
|
||||||
markup = xml(lambda x: ref)
|
|
||||||
if not markup:
|
|
||||||
return False
|
|
||||||
|
|
||||||
s = {}
|
|
||||||
s['author_link'] = ''
|
|
||||||
s['score'] = 0
|
|
||||||
s['comments'] = []
|
|
||||||
s['num_comments'] = 0
|
|
||||||
s['link'] = ref
|
|
||||||
s['url'] = ref
|
|
||||||
s['date'] = 0
|
|
||||||
|
|
||||||
data = extruct.extract(markup)
|
|
||||||
s = parse_extruct(s, data)
|
|
||||||
|
|
||||||
if not s['date']:
|
|
||||||
return False
|
|
||||||
return s
|
|
||||||
|
|
||||||
class Category:
|
|
||||||
def __init__(self, url):
|
|
||||||
self.category_url = url
|
|
||||||
self.base_url = '/'.join(url.split('/')[:3])
|
|
||||||
|
|
||||||
def feed(self):
|
|
||||||
markup = xml(lambda x: self.category_url)
|
|
||||||
if not markup: return []
|
|
||||||
soup = BeautifulSoup(markup, features='html.parser')
|
|
||||||
links = soup.find_all('a', href=True)
|
|
||||||
links = [link.get('href') for link in links]
|
|
||||||
links = [f"{self.base_url}{link}" if link.startswith('/') else link for link in links]
|
|
||||||
links = list(filter(None, [link if link.startswith(self.category_url) else None for link in links]))
|
|
||||||
links = list(filter(None, [link if link != self.category_url else None for link in links]))
|
|
||||||
return links
|
return links
|
||||||
|
|
||||||
def story(self, ref):
|
def story(self, ref):
|
||||||
|
@ -164,6 +133,47 @@ class Category:
|
||||||
return False
|
return False
|
||||||
return s
|
return s
|
||||||
|
|
||||||
|
|
||||||
|
class Category:
|
||||||
|
def __init__(self, url):
|
||||||
|
self.category_url = url
|
||||||
|
self.base_url = '/'.join(url.split('/')[:3])
|
||||||
|
|
||||||
|
def feed(self):
|
||||||
|
markup = xml(lambda x: self.category_url)
|
||||||
|
if not markup: return []
|
||||||
|
soup = BeautifulSoup(markup, features='html.parser')
|
||||||
|
links = soup.find_all('a', href=True)
|
||||||
|
links = [link.get('href') for link in links]
|
||||||
|
links = [f"{self.base_url}{link}" if link.startswith('/') else link for link in links]
|
||||||
|
links = list(filter(None, [link if link.startswith(self.category_url) else None for link in links]))
|
||||||
|
links = list(filter(None, [link if link != self.category_url else None for link in links]))
|
||||||
|
links = list(set(links))
|
||||||
|
|
||||||
|
return links
|
||||||
|
|
||||||
|
def story(self, ref):
|
||||||
|
markup = xml(lambda x: ref)
|
||||||
|
if not markup:
|
||||||
|
return False
|
||||||
|
|
||||||
|
s = {}
|
||||||
|
s['author_link'] = ''
|
||||||
|
s['score'] = 0
|
||||||
|
s['comments'] = []
|
||||||
|
s['num_comments'] = 0
|
||||||
|
s['link'] = ref
|
||||||
|
s['url'] = ref
|
||||||
|
s['date'] = 0
|
||||||
|
|
||||||
|
data = extruct.extract(markup)
|
||||||
|
s = parse_extruct(s, data)
|
||||||
|
|
||||||
|
if not s['date']:
|
||||||
|
return False
|
||||||
|
return s
|
||||||
|
|
||||||
|
|
||||||
# scratchpad so I can quickly develop the parser
|
# scratchpad so I can quickly develop the parser
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
print("Sitemap: Stuff")
|
print("Sitemap: Stuff")
|
||||||
|
|
Loading…
Reference in New Issue
Block a user