spitouthtml.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-

# This script takes a Python dictionary of newspaper stories
# and turns them into HTML files, including an index.html page.

import codecs
import parsearticles
import pystache
from os import path
from dateutil.parser import parse

# An individual "article" or "story" is a dictionary with
# headline
# date
# body


class HtmlOutput(object):
    def __init__(self, story, article_index):
        self.story = story
        self.article_index = article_index
        self.dateindex = [x["date"] for x in self.article_index]
        self.datemarker = self.dateindex.index(self.story["date"])

    def headline(self):
        return self.story["headline"]

    def pubdate(self):
        self.dateobject = parse(self.story["date"])
        return english_from_date(self.dateobject)

    def verbiage(self):
        return self.story["body"]

    def nextstoryhed(self):
        if self.datemarker == len(self.article_index) - 1:
            return "Last story"
        else:
            return "Next: " + self.article_index[self.datemarker+1]["headline"]

    def nextstorydateslug(self):
        if self.datemarker == len(self.article_index) - 1:
            return ""
        else:
            self.nextdate = self.article_index[self.datemarker+1]["date"]
            self.dateobject = parse(self.nextdate)
            return filename_from_date(self.dateobject)

    def prevstoryhed(self):
        if self.datemarker == 0:
            return "First story"
        else:
            return "Previous: " + self.article_index[self.datemarker-1]["headline"]

    def prevstorydateslug(self):
        if self.datemarker == 0:
            return ""
        else:
            self.prevdate = self.article_index[self.datemarker-1]["date"]
            self.dateobject = parse(self.prevdate)
            return filename_from_date(self.dateobject)


class IndexOutput(object):
    def __init__(self, article_index):
        self.article_index = article_index

    def stories(self):

        # This would be better as a Pystache partial or
        # iteration if I could figure that out.

        toc = ""

        for storydesc in self.article_index:
            dateobject = parse(storydesc["date"])
            dateURL = filename_from_date(dateobject)
            datestring = english_from_date(dateobject)
            toc += '<li><a href="' + dateURL + '">' + storydesc["headline"] + "</a>, " + datestring + "</li>"
        return toc

def filename_from_date(dateobject):
    return dateobject.strftime("%d-%B-%Y") + ".html"

def english_from_date(dateobject):
    return dateobject.strftime("%A, %d %B %Y")

def htmlize_story(story, article_index):

    '''use pystache to turn each storydict into HTML to output,
    take HTML string and write it to a file'''

    storyoutput = HtmlOutput(story, article_index)
    dateobject = parse(story["date"])
    dateslug = filename_from_date(dateobject)
    renderer = pystache.Renderer()

    with codecs.open(dateslug, encoding='utf-8', mode='w') as f:
        f.write(renderer.render(storyoutput))

def htmlize_article_index(article_index):

    '''use pystache to turn the list into HTML to output,
    take HTML string and write it to a file '''

    indexoutput = IndexOutput(article_index)
    renderer = pystache.Renderer()

    with codecs.open("index.html", encoding='utf-8', mode='w') as f:
        f.write(renderer.render(indexoutput))


if __name__ == "__main__":
    (article_list, article_index) = parsearticles.parse_files(parsearticles.ARCHIVEFILES, [])
    htmlize_article_index(article_index)
    for article in article_list:
        htmlize_story(article, article_index)