forked from tanner/qotnews
159 lines
4.5 KiB
Python
159 lines
4.5 KiB
Python
import logging
|
|
logging.basicConfig(
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
|
level=logging.INFO)
|
|
|
|
import copy
|
|
import threading
|
|
import time
|
|
import shelve
|
|
from urllib.parse import urlparse
|
|
|
|
import archive
|
|
import feed
|
|
from utils import gen_rand_id
|
|
|
|
from flask import abort, Flask, request, render_template
|
|
from werkzeug.exceptions import NotFound
|
|
from flask_cors import CORS
|
|
|
|
archive.init()
|
|
|
|
CACHE_LENGTH = 300
|
|
DATA_FILE = 'data/data'
|
|
|
|
news_index = 0
|
|
|
|
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', {})
|
|
|
|
def get_story(id):
|
|
if id in news_cache:
|
|
return news_cache[id]
|
|
else:
|
|
return archive.get_story(id)
|
|
|
|
build_folder = '../webclient/build'
|
|
flask_app = Flask(__name__, template_folder=build_folder, static_folder=build_folder, static_url_path='')
|
|
cors = CORS(flask_app)
|
|
|
|
@flask_app.route('/api')
|
|
def api():
|
|
front_page = [news_cache[news_ref_to_id[ref]] for ref in news_list]
|
|
front_page = [x for x in front_page if 'title' in x and x['title']]
|
|
front_page = front_page[:100]
|
|
to_remove = ['text', 'comments']
|
|
front_page = [{k:v for k,v in s.items() if k not in to_remove} for s in front_page]
|
|
|
|
return {'stories': front_page}
|
|
|
|
@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}
|
|
|
|
@flask_app.route('/api/<id>')
|
|
def story(id):
|
|
story = get_story(id)
|
|
return dict(story=story) if story else abort(404)
|
|
|
|
@flask_app.route('/')
|
|
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')
|
|
|
|
@flask_app.route('/<id>', strict_slashes=False)
|
|
@flask_app.route('/<id>/c', strict_slashes=False)
|
|
def static_story(id):
|
|
try:
|
|
return flask_app.send_static_file(id)
|
|
except NotFound:
|
|
pass
|
|
|
|
story = get_story(id)
|
|
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)
|
|
url = urlparse(story['url']).hostname.replace('www.', '')
|
|
|
|
return render_template('index.html',
|
|
title=story['title'],
|
|
url=url,
|
|
description=description)
|
|
|
|
print('Starting Flask...')
|
|
web_thread = threading.Thread(target=flask_app.run, kwargs={'port': 33842})
|
|
web_thread.setDaemon(True)
|
|
web_thread.start()
|
|
|
|
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))
|
|
|
|
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)
|
|
|
|
time.sleep(3)
|
|
|
|
news_index += 1
|
|
if news_index == CACHE_LENGTH: news_index = 0
|
|
|
|
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
|