qotnews/apiserver/server.py

210 lines
6.0 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)
import copy
import threading
import time
2019-08-24 05:07:16 +00:00
import shelve
from urllib.parse import urlparse
2019-08-23 08:23:48 +00:00
2019-10-12 05:32:17 +00:00
import archive
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
2019-11-08 02:37:57 +00:00
import gevent
from gevent import monkey
from gevent.pywsgi import WSGIServer
monkey.patch_all()
2019-10-12 05:32:17 +00:00
archive.init()
2019-12-01 22:18:14 +00:00
CACHE_LENGTH = 150
2019-08-24 05:07:16 +00:00
DATA_FILE = 'data/data'
2019-08-23 08:23:48 +00:00
news_index = 0
2019-08-24 05:07:16 +00:00
with shelve.open(DATA_FILE) as db:
logging.info('Reading caches from disk...')
news_list = db.get('news_list', [])
news_ref_to_id = db.get('news_ref_to_id', {})
news_cache = db.get('news_cache', {})
2019-08-23 08:23:48 +00:00
2019-11-08 05:55:30 +00:00
# clean cache if broken
try:
for ref in news_list:
nid = news_ref_to_id[ref]
_ = news_cache[nid]
except KeyError as e:
logging.error('Unable to find key: ' + str(e))
logging.info('Clearing caches...')
news_list = []
news_ref_to_id = {}
news_cache = {}
2019-10-15 20:36:20 +00:00
def get_story(sid):
if sid in news_cache:
return news_cache[sid]
2019-10-12 05:32:17 +00:00
else:
2019-10-15 20:36:20 +00:00
return archive.get_story(sid)
2019-11-08 05:55:30 +00:00
def new_id():
nid = gen_rand_id()
while nid in news_cache or archive.get_story(nid):
nid = gen_rand_id()
return nid
def remove_ref(old_ref):
while old_ref in news_list:
news_list.remove(old_ref)
old_story = news_cache.pop(news_ref_to_id[old_ref])
old_id = news_ref_to_id.pop(old_ref)
logging.info('Removed ref {} id {}.'.format(old_ref, old_id))
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():
2019-08-23 08:23:48 +00:00
front_page = [news_cache[news_ref_to_id[ref]] for ref in news_list]
2019-10-12 23:41:31 +00:00
front_page = [copy.copy(x) for x in front_page if 'title' in x and x['title']]
2019-11-08 21:50:33 +00:00
front_page = front_page[:60]
2019-10-12 23:41:31 +00:00
for story in front_page:
story.pop('text', None)
story.pop('comments', None)
2019-10-12 05:32:17 +00:00
2019-08-23 08:23:48 +00:00
return {'stories': front_page}
2019-10-12 05:32:17 +00:00
@flask_app.route('/api/search', strict_slashes=False)
def search():
search = request.args.get('q', '')
if len(search) >= 3:
res = archive.search(search)
else:
res = []
return {'results': res}
2019-11-08 05:55:30 +00:00
@flask_app.route('/api/submit', methods=['POST'], strict_slashes=False)
def submit():
url = request.form['url']
nid = new_id()
news_story = dict(id=nid, ref=url, source='manual')
news_cache[nid] = news_story
valid = feed.update_story(news_story)
if valid:
archive.update(news_story)
return {'nid': nid}
else:
news_cache.pop(nid, '')
abort(400)
2019-10-15 20:36:20 +00:00
@flask_app.route('/api/<sid>')
def story(sid):
story = get_story(sid)
2019-10-12 05:32:17 +00:00
return dict(story=story) if story else 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',
title='Feed',
url='news.t0.vc',
description='Reddit, Hacker News, and Tildes combined, then pre-rendered in reader mode')
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
2019-10-15 20:36:20 +00:00
story = get_story(sid)
2019-10-12 05:32:17 +00:00
if not story: return abort(404)
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',
title=story['title'],
url=url,
description=description)
2019-08-23 08:23:48 +00:00
2019-11-08 02:37:57 +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:
feed_list = feed.list()
new_items = [(ref, source) for ref, source in feed_list if ref not in news_list]
for ref, source in new_items:
news_list.insert(0, ref)
nid = new_id()
news_ref_to_id[ref] = nid
news_cache[nid] = dict(id=nid, ref=ref, source=source)
if len(new_items):
logging.info('Added {} new refs.'.format(len(new_items)))
# drop old ones
while len(news_list) > CACHE_LENGTH:
old_ref = news_list[-1]
remove_ref(old_ref)
# update current stories
if news_index < len(news_list):
update_ref = news_list[news_index]
update_id = news_ref_to_id[update_ref]
news_story = news_cache[update_id]
valid = feed.update_story(news_story)
if valid:
archive.update(news_story)
else:
remove_ref(update_ref)
2019-11-08 21:50:33 +00:00
else:
logging.info('Skipping update - no story #' + str(news_index+1))
2019-11-08 02:37:57 +00:00
2019-12-01 22:18:14 +00:00
gevent.sleep(6)
2019-11-08 02:37:57 +00:00
news_index += 1
if news_index == CACHE_LENGTH: news_index = 0
2019-11-08 05:55:30 +00:00
except KeyboardInterrupt:
logging.info('Ending feed thread...')
except ValueError as e:
2019-11-08 02:37:57 +00:00
logging.error('feed_thread error: {} {}'.format(e.__class__.__name__, e))
http_server.stop()
print('Starting Feed thread...')
gevent.spawn(feed_thread)
print('Starting HTTP thread...')
try:
http_server.serve_forever()
except KeyboardInterrupt:
logging.info('Exiting...')
2019-08-24 05:07:16 +00:00
finally:
with shelve.open(DATA_FILE) as db:
logging.info('Writing caches to disk...')
db['news_list'] = news_list
db['news_ref_to_id'] = news_ref_to_id
db['news_cache'] = news_cache