app.py

#!/usr/bin/env python

import app_config
import argparse
import json
import re
import static
import string

from PIL import Image
from flask import Flask, make_response, render_template
from render_utils import make_context, smarty_filter, urlencode_filter

app = Flask(__name__)

app.add_template_filter(smarty_filter, name='smarty')
app.add_template_filter(urlencode_filter, name='urlencode')

def _title_sorter(book):
    title = book['title']
    if title.startswith('The'):
        title = book['title'][4:]
    return title

@app.route('/')
def index():
    """
    The index page.
    """
    # Set up standard page context.
    context = make_context()

    with open('data/featured.json') as f:
        context['featured'] = json.load(f)

    # Read the books JSON into the page.
    with open('www/static-data/books.json', 'rb') as readfile:
        context['books_js'] = readfile.read()
        books = json.loads(context['books_js'])
        books_text_only = books[:]
        books_text_only = sorted(books, key=_title_sorter)

    for book in books:
        if not book['text']:
            book['teaser'] = None
        else:
            book['teaser'] = _make_teaser(book)

    context['books'] = books
    context['books_text_only'] = books_text_only

    return render_template('index.html', **context)

@app.route('/seamus')
def seamus():
    """
    Preview for Seamus page
    """
    context = make_context()

    # Read the books JSON into the page.
    with open('www/static-data/books.json', 'rb') as readfile:
        books_data = json.load(readfile)
        books = sorted(books_data, key=lambda k: k['title'])

    # Harvest long tag names
    for book in books:
        tag_list = []
        for tag in book['tags']:
            tag_list.append(context['COPY']['tags'][tag]['value'])
        book['tag_list'] = tag_list

    context['books'] = books

    return render_template('seamus-preview.html', **context)


def _make_teaser(book):
    """
    Calculate a teaser
    """
    tag_stripper = re.compile(r'<.*?>')

    try:
        img = Image.open('www/assets/cover/%s.jpg' % book['slug'])
        width, height = img.size

        # Poor man's packing algorithm. How much text will fit?
        chars = height / 25 * 7
    except IOError:
        chars = 140

    text = tag_stripper.sub('', book['text'])

    if len(text) <= chars:
        return text

    i = chars

    # Walk back to last full word
    while text[i] != ' ':
        i -= 1

    # Like strip, but decrements the counter
    if text.endswith(' '):
        i -= 1

    # Kill trailing punctuation
    exclude = set(string.punctuation)
    if text[i-1] in exclude:
        i -= 1

    return '&#8220;' + text[:i] + ' ...&#8221;'

@app.route('/share/<slug>.html')
def share(slug):
    featured_book = None
    context = make_context()
    with open('www/static-data/books.json', 'rb') as f:
        books = json.load(f)
        for book in books:
            if book.get('slug') == slug:
                featured_book = book
                break

    if not featured_book:
        return 404

    featured_book['teaser'] = _make_teaser(featured_book)
    featured_book['thumb'] = "%sassets/cover/%s.jpg" % (context['SHARE_URL'], featured_book['slug'])

    context['twitter_handle'] = 'nprbooks'
    context['book'] = featured_book

    return make_response(render_template('share.html', **context))

app.register_blueprint(static.static)

# Boilerplate
if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--port')
    args = parser.parse_args()
    server_port = 8000

    if args.port:
        server_port = int(args.port)

    app.run(host='0.0.0.0', port=server_port, debug=app_config.DEBUG)