qotnews/apiserver/server.py

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, stream_with_context, Response
from werkzeug.exceptions import NotFound
from flask_cors import CORS

import gevent
from gevent import monkey
from gevent.pywsgi import WSGIServer

monkey.patch_all()

archive.init()

CACHE_LENGTH = 150
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', {})

    # 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 = {}

def get_story(sid):
    if sid in news_cache:
        return news_cache[sid]
    else:
        return archive.get_story(sid)

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='')
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 = [copy.copy(x) for x in front_page if 'title' in x and x['title']]
    front_page = front_page[:60]
    for story in front_page:
        story.pop('text', None)
        story.pop('comments', None)

    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/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)

@flask_app.route('/api/<sid>')
def story(sid):
    story = get_story(sid)
    return dict(story=story) if story else abort(404)

@flask_app.route('/')
@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')

@flask_app.route('/<sid>', strict_slashes=False)
@flask_app.route('/<sid>/c', strict_slashes=False)
def static_story(sid):
    try:
        return flask_app.send_static_file(sid)
    except NotFound:
        pass

    story = get_story(sid)
    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 or urlparse(story['link']).hostname or ''
    url = url.replace('www.', '')

    return render_template('index.html',
            title=story['title'],
            url=url,
            description=description)

http_server = WSGIServer(('', 33842), flask_app)

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)
            else:
                logging.info('Skipping update - no story #' + str(news_index+1))

            gevent.sleep(6)

            news_index += 1
            if news_index == CACHE_LENGTH: news_index = 0

    except KeyboardInterrupt:
        logging.info('Ending feed thread...')
    except ValueError as e:
        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...')
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