qotnews/apiserver/feeds/tildes.py

151 lines
4.7 KiB
Python

import logging
logging.basicConfig(
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.DEBUG)
if __name__ == '__main__':
import sys
sys.path.insert(0,'.')
import requests
from bs4 import BeautifulSoup
from datetime import datetime
from utils import clean
# cache the topic groups to prevent redirects
group_lookup = {}
USER_AGENT = 'qotnews scraper (github:tannercollin)'
API_TOPSTORIES = lambda : 'https://tildes.net'
API_ITEM = lambda x : 'https://tildes.net/shortener/{}'.format(x)
SITE_LINK = lambda group, ref : 'https://tildes.net/{}/{}'.format(group, ref)
SITE_AUTHOR_LINK = lambda x : 'https://tildes.net/user/{}'.format(x)
def api(route):
try:
headers = {'User-Agent': USER_AGENT}
r = requests.get(route, headers=headers, timeout=5)
if r.status_code != 200:
raise Exception('Bad response code ' + str(r.status_code))
return r.text
except KeyboardInterrupt:
raise
except BaseException as e:
logging.critical('Problem hitting tildes website: {}'.format(str(e)))
return False
def feed():
html = api(API_TOPSTORIES())
if not html: return []
soup = BeautifulSoup(html, features='html.parser')
articles = soup.find('ol', class_='topic-listing').findAll('article')
return [x['id'].split('-')[1] for x in articles] or []
def unix(date_str):
return int(datetime.strptime(date_str, '%Y-%m-%dT%H:%M:%SZ').timestamp())
def comment(i):
i = i.article
if i.find('div', class_='is-comment-removed'):
return False
if i.find('div', class_='is-comment-deleted'):
return False
c = {}
lu = i.find('a', class_='link-user')
c['author'] = str(lu.string if lu else 'unknown user')
c['score'] = 1
c['date'] = unix(i.find('time')['datetime'])
c['text'] = clean(i.find('div', class_='comment-text').encode_contents().decode() or '')
ct = i.find('ol', class_='comment-tree')
c['comments'] = [comment(j) for j in ct.findAll('li', recursive=False)] if ct else []
c['comments'] = list(filter(bool, c['comments']))
return c
def story(ref):
if ref in group_lookup:
html = api(SITE_LINK(group_lookup[ref], ref))
else:
html = api(API_ITEM(ref))
if not html:
logging.info('Bad Tildes API response.')
return False
soup = BeautifulSoup(html, features='html.parser')
a = soup.find('article', class_='topic-full')
if a is None:
logging.info('Tildes <article> element not found.')
return False
h = a.find('header')
lu = h.find('a', class_='link-user')
error = a.find('div', class_='text-error')
if error:
if 'deleted' in error.string or 'removed' in error.string:
logging.info('Article was deleted or removed.')
return False
s = {}
s['author'] = str(lu.string if lu else 'unknown user')
s['author_link'] = SITE_AUTHOR_LINK(s['author'])
s['score'] = int(h.find('span', class_='topic-voting-votes').string)
s['date'] = unix(h.find('time')['datetime'])
s['title'] = str(h.h1.string)
s['group'] = str(soup.find('div', class_='site-header-context').a.string)
group_lookup[ref] = s['group']
s['link'] = SITE_LINK(s['group'], ref)
ud = a.find('div', class_='topic-full-link')
s['url'] = ud.a['href'] if ud else s['link']
sc = a.find('ol', id='comments')
s['comments'] = [comment(i) for i in sc.findAll('li', recursive=False)]
s['comments'] = list(filter(bool, s['comments']))
ch = a.find('header', class_='topic-comments-header')
s['num_comments'] = int(ch.h2.string.split(' ')[0]) if ch else 0
if s['group'].split('.')[0] not in [
'~arts',
'~comp',
'~creative',
'~design',
'~engineering',
'~finance',
'~science',
'~tech',
]:
logging.info('Group ({}) not in whitelist.'.format(s['group']))
return False
if s['score'] < 15 and s['num_comments'] < 10:
logging.info('Score ({}) or num comments ({}) below threshold.'.format(s['score'], s['num_comments']))
return False
td = a.find('div', class_='topic-full-text')
if td:
s['text'] = clean(td.encode_contents().decode() or '')
return s
# scratchpad so I can quickly develop the parser
if __name__ == '__main__':
print(feed())
#normal = story('gxt')
#print(normal)
#no_comments = story('gxr')
#print(no_comments)
#self_post = story('gsb')
#print(self_post)
#li_comment = story('gqx')
#print(li_comment)
#broken = story('q4y')
#print(broken)
# make sure there's no self-reference
#import copy
#for x in [normal, no_comments, self_post, li_comment]:
# _ = copy.deepcopy(x)