qotnews/apiserver/server.py

223 lines
6.7 KiB
Python
Raw Normal View History

2019-08-23 08:23:48 +00:00
import logging
logging.basicConfig(
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO)
2020-10-24 22:30:00 +00:00
import gevent
from gevent import monkey
monkey.patch_all()
from gevent.pywsgi import WSGIServer
2019-08-23 08:23:48 +00:00
import copy
2020-06-26 21:28:39 +00:00
import json
2019-08-23 08:23:48 +00:00
import threading
2020-06-25 23:38:16 +00:00
import traceback
2019-08-23 08:23:48 +00:00
import time
from urllib.parse import urlparse, parse_qs
2019-08-23 08:23:48 +00:00
import settings
2020-06-26 21:28:39 +00:00
import database
import search
2019-08-24 08:49:11 +00:00
import feed
from utils import gen_rand_id
2019-08-24 08:49:11 +00:00
2019-11-08 05:55:30 +00:00
from flask import abort, Flask, request, render_template, stream_with_context, Response
from werkzeug.exceptions import NotFound
2019-08-23 08:23:48 +00:00
from flask_cors import CORS
2020-06-26 21:28:39 +00:00
database.init()
search.init()
2019-08-23 08:23:48 +00:00
news_index = 0
2019-08-24 05:07:16 +00:00
2019-11-08 05:55:30 +00:00
def new_id():
nid = gen_rand_id()
2020-06-26 21:28:39 +00:00
while database.get_story(nid):
2019-11-08 05:55:30 +00:00
nid = gen_rand_id()
return nid
build_folder = '../webclient/build'
flask_app = Flask(__name__, template_folder=build_folder, static_folder=build_folder, static_url_path='')
2019-08-23 08:23:48 +00:00
cors = CORS(flask_app)
@flask_app.route('/api')
def api():
skip = request.args.get('skip', 0)
limit = request.args.get('limit', settings.FEED_LENGTH)
stories = database.get_stories(limit, skip)
2020-06-26 21:28:39 +00:00
# hacky nested json
res = Response('{"stories":[' + ','.join(stories) + ']}')
res.headers['content-type'] = 'application/json'
return res
2019-08-23 08:23:48 +00:00
2019-10-12 05:32:17 +00:00
@flask_app.route('/api/search', strict_slashes=False)
def apisearch():
2020-06-26 21:28:39 +00:00
q = request.args.get('q', '')
if len(q) >= 3:
results = search.search(q)
2019-10-12 05:32:17 +00:00
else:
results = '[]'
res = Response(results)
2022-03-05 21:33:07 +00:00
res.headers['content-type'] = 'application/json'
return res
2019-10-12 05:32:17 +00:00
2019-11-08 05:55:30 +00:00
@flask_app.route('/api/submit', methods=['POST'], strict_slashes=False)
def submit():
try:
url = request.form['url']
nid = new_id()
2021-09-06 00:21:05 +00:00
logging.info('Manual submission: ' + url)
parse = urlparse(url)
if 'news.ycombinator.com' in parse.hostname:
source = 'hackernews'
ref = parse_qs(parse.query)['id'][0]
elif 'tildes.net' in parse.hostname and '~' in url:
source = 'tildes'
ref = parse.path.split('/')[2]
2020-12-12 05:26:33 +00:00
elif 'lobste.rs' in parse.hostname and '/s/' in url:
source = 'lobsters'
ref = parse.path.split('/')[2]
elif 'reddit.com' in parse.hostname and 'comments' in url:
source = 'reddit'
ref = parse.path.split('/')[4]
elif 'news.t0.vc' in parse.hostname:
raise Exception('Invalid article')
else:
source = 'manual'
ref = url
existing = database.get_story_by_ref(ref)
if existing:
return {'nid': existing.sid}
else:
story = dict(id=nid, ref=ref, source=source)
valid = feed.update_story(story, is_manual=True)
if valid:
database.put_story(story)
search.put_story(story)
return {'nid': nid}
else:
raise Exception('Invalid article')
except BaseException as e:
logging.error('Problem with article submission: {} - {}'.format(e.__class__.__name__, str(e)))
2020-06-25 23:38:16 +00:00
print(traceback.format_exc())
2019-11-08 05:55:30 +00:00
abort(400)
2019-10-15 20:36:20 +00:00
@flask_app.route('/api/<sid>')
def story(sid):
2020-06-26 21:28:39 +00:00
story = database.get_story(sid)
if story:
# hacky nested json
res = Response('{"story":' + story.full_json + '}')
res.headers['content-type'] = 'application/json'
return res
else:
return abort(404)
2019-08-24 05:07:16 +00:00
@flask_app.route('/')
2019-10-15 20:36:20 +00:00
@flask_app.route('/search')
def index():
return render_template('index.html',
2022-06-30 00:05:15 +00:00
title='QotNews',
2022-06-29 23:20:53 +00:00
url='news.t0.vc',
description='Hacker News, Reddit, Lobsters, and Tildes articles rendered in reader mode',
robots='index',
)
2019-10-15 20:36:20 +00:00
@flask_app.route('/<sid>', strict_slashes=False)
@flask_app.route('/<sid>/c', strict_slashes=False)
def static_story(sid):
try:
2019-10-15 20:36:20 +00:00
return flask_app.send_static_file(sid)
except NotFound:
pass
2020-06-26 21:28:39 +00:00
story = database.get_story(sid)
2019-10-12 05:32:17 +00:00
if not story: return abort(404)
2020-06-26 21:28:39 +00:00
story = json.loads(story.full_json)
score = story['score']
num_comments = story['num_comments']
source = story['source']
description = '{} point{}, {} comment{} on {}'.format(
score, 's' if score != 1 else '',
num_comments, 's' if num_comments != 1 else '',
source)
2019-11-07 22:10:08 +00:00
url = urlparse(story['url']).hostname or urlparse(story['link']).hostname or ''
url = url.replace('www.', '')
2019-08-24 05:07:16 +00:00
return render_template('index.html',
2022-06-30 00:05:15 +00:00
title=story['title'] + ' | QotNews',
2022-06-29 23:20:53 +00:00
url=url,
description=description,
robots='noindex',
)
2019-08-23 08:23:48 +00:00
2020-08-14 03:57:18 +00:00
http_server = WSGIServer(('', 33842), flask_app)
2019-08-24 05:07:16 +00:00
2019-11-08 02:37:57 +00:00
def feed_thread():
global news_index
try:
while True:
# onboard new stories
if news_index == 0:
2020-06-26 21:28:39 +00:00
for ref, source in feed.list():
if database.get_story_by_ref(ref):
continue
2020-06-26 21:28:39 +00:00
try:
nid = new_id()
logging.info('Adding ref: {}, id: {}, source: {}'.format(ref, nid, source))
database.put_ref(ref, nid, source)
2020-06-26 21:28:39 +00:00
except database.IntegrityError:
logging.info('Already have ID / ref, skipping.')
2020-06-26 21:28:39 +00:00
continue
2019-11-08 02:37:57 +00:00
ref_list = database.get_reflist(settings.FEED_LENGTH)
2019-11-08 02:37:57 +00:00
# update current stories
2020-06-26 21:28:39 +00:00
if news_index < len(ref_list):
item = ref_list[news_index]
try:
story_json = database.get_story(item['sid']).full_json
story = json.loads(story_json)
except AttributeError:
story = dict(id=item['sid'], ref=item['ref'], source=item['source'])
logging.info('Updating {} story: {}, index: {}'.format(story['source'], story['ref'], news_index))
2020-11-02 00:13:43 +00:00
2020-06-26 21:28:39 +00:00
valid = feed.update_story(story)
2019-11-08 02:37:57 +00:00
if valid:
2020-06-26 21:28:39 +00:00
database.put_story(story)
search.put_story(story)
2019-11-08 02:37:57 +00:00
else:
database.del_ref(item['ref'])
logging.info('Removed ref {}'.format(item['ref']))
2020-07-04 22:39:38 +00:00
else:
2020-11-02 00:13:43 +00:00
logging.info('Skipping index: ' + str(news_index))
2019-11-08 02:37:57 +00:00
2020-07-04 22:39:38 +00:00
gevent.sleep(6)
2019-11-08 02:37:57 +00:00
news_index += 1
if news_index == settings.FEED_LENGTH: news_index = 0
2019-11-08 02:37:57 +00:00
2019-11-08 05:55:30 +00:00
except KeyboardInterrupt:
logging.info('Ending feed thread...')
except ValueError as e:
2021-09-06 00:21:05 +00:00
logging.critical('feed_thread error: {} {}'.format(e.__class__.__name__, e))
2019-11-08 02:37:57 +00:00
http_server.stop()
logging.info('Starting Feed thread...')
2019-11-08 02:37:57 +00:00
gevent.spawn(feed_thread)
logging.info('Starting HTTP thread...')
2019-11-08 02:37:57 +00:00
try:
http_server.serve_forever()
except KeyboardInterrupt:
logging.info('Exiting...')