api.py

import os
import redis
from flask import Flask, render_template, redirect, jsonify, url_for, abort
from json import loads, dumps
from util import json, jsonp, requires_auth, support_jsonp
from topstories import get_search_terms_by_probe
from forms import SpaceProbeForm
from dateutil import parser

app = Flask(__name__)
REDIS_URL = os.getenv('REDISTOGO_URL', 'redis://localhost:6379')
r_server = redis.StrictRedis.from_url(REDIS_URL)

app = Flask(__name__)

app.config['SECRET_KEY'] = os.getenv('SECRET_KEY')

@app.route('/')
@app.route('/news.json')
@support_jsonp
def topstories():
    topstories = loads(r_server.get('topstories'))

    # add the published_readable field to the bulk json feed
    for t in topstories:
        published = topstories[t]['published']
        published_readable = parser.parse(published).strftime('%B %e, %Y')
        topstories[t]['published_readable'] = published_readable
        
    return jsonify(topstories)


@app.route('/<probe_name>')
@support_jsonp
def topstories_single(probe_name):
    topstories = loads(r_server.get('topstories'))
    try:
        # pass along a human readable version of published as well
        post = topstories[probe_name.lower()]
        post['published_readable'] = parser.parse(post['published']).strftime('%B %e, %Y')
        return jsonify({"post":post})
    except KeyError:
        abort(404)


@app.route('/admin/<probe_name>/update.html', methods = ['POST'])
@requires_auth
def admin(probe_name):
    topstories = loads(r_server.get('topstories'))
    topstories_archive = loads(r_server.get('topstories_archive'))
    form = SpaceProbeForm()

    if form.validate():
        # archive the old
        if form.probe_name.data in topstories:
            topstories_archive[form.probe_name.data] = topstories[form.probe_name.data]
        # update the new
        topstories[form.probe_name.data] = {'title':form.title.data, 'link':form.url.data, 'published': form.date.data }
        # save to redis
        r_server.set('topstories', dumps(topstories))
        r_server.set('topstories_archive', dumps(topstories))

        # redirct to main page
        return redirect(url_for('admin_main', _anchor=form.probe_name.data))
    
    return render_template('admin_probe.html', form=form, probe_name=probe_name, post=topstories[probe_name])


@app.route('/admin/<probe_name>/')
@requires_auth
def admin_probe(probe_name):
    probe_name = ' '.join(probe_name.split('_'))
    topstories = loads(r_server.get('topstories'))
    try:
        post = topstories[probe_name]
    except KeyError:
        post = {}

    kwargs = {'probe_name': probe_name, 'post':post, 'form':SpaceProbeForm()}
    return render_template('admin_probe.html', **kwargs)


@app.route('/admin/')
def admin_main():
    spaceprobes = get_search_terms_by_probe()
    topstories = loads(r_server.get('topstories'))
    kwargs = {'topstories':topstories, 'spaceprobes': spaceprobes}
    return render_template('admin.html', **kwargs)


if __name__ == '__main__':
    app.debug = True
    app.run()