process-content.py

#!/usr/bin/env python
import sys
import os
import json
import io
from lxml.html import soupparser
from PIL import Image

if len(sys.argv) != 2:
    print("usage: ./process-content.py pregen/<CONTENT_DIRECTORY>")
    exit(1)

pregenContentPath = sys.argv[1]
if not os.path.isdir(pregenContentPath):
    print("error: %s is not a content directory" % pregenContentPath)
    print("usage: ./process-content pregen/<CONTENT_DIRECTORY>")
    exit(1)

metadataPath = os.path.join(pregenContentPath, "metadata.json")
if not os.path.isfile(metadataPath):
    print("error: %s does not have a metadata.json file" % pregenContentPath)
    exit(1)

def hyphensToCamelcase(s):
    return "".join(map(lambda p: p.capitalize(), s.strip(os.path.sep).split("-")))

def areasToTypescript(areas):
    typescript = "[\n"
    areaLines = []
    for area in areas:
        areaLines.append("        { name: \"%s\", shape: \"%s\", coords: \"%s\" }" % (area["name"], area["shape"], area["coords"]))
    typescript += ",\n".join(areaLines)

    typescript += "\n    ]\n"
    return typescript

def adsToTypescript(ads):
    adLines = ["        " + json.dumps(ad) for ad in ads]
    typescript = ",\n".join(adLines)
    return typescript

contentName = pregenContentPath.split("pregen" + os.path.sep)[-1].strip(os.path.sep)

imageDir = "public/static/images"
contentImgPath = os.path.join(imageDir, "%s.jpg" % contentName)
thumbnailImgPath = os.path.join(imageDir, "%s_thumbnail.jpg" % contentName)
if not os.path.isfile(contentImgPath):
    print("error: %s content image file does not exist" % contentImgPath)
    exit(1)

# TODO: add other image-checking errors like resolution and file size maximum

im = Image.open(contentImgPath)
if im.width > 1920 or im.height > 1920:
    print("error: %s content file is too large, maximum of 1920x1920" % contentImgPath)
    exit(1)

if not os.path.isfile(thumbnailImgPath):
    print("error: %s content thumbnail file does not exist" % thumbnailImgPath)
    exit(1)

# create the thumbnail
THUMB_WIDTH = 300
THUMB_HEIGHT = 168
im = Image.open(thumbnailImgPath)
if im.width != THUMB_WIDTH or im.height != THUMB_HEIGHT:
    print("error: %s content thumbnail file must be %ix%i" % (thumbnailImgPath, THUMB_WIDTH, THUMB_HEIGHT))
    exit(1)

with open(metadataPath) as f:
    metadata = json.load(f)

    className = hyphensToCamelcase(contentName)

    areas = []
    for areaPath in os.listdir(pregenContentPath):
        if areaPath.endswith(".areas"):    
            areaName = os.path.splitext(areaPath)[0]
            print("processing areas file: " + areaPath)
            if ":" in areaName:
                print("error: area files like '%s' cannot contain a colon" % areaName)
                exit(1)
            with open(os.path.join(pregenContentPath, areaPath)) as a:
                # super hacky way to parse HTML
                contents = u"<areas>\n" + a.read() + "</areas>\n"
                tree = soupparser.parse(io.StringIO(contents))
                areasRoot = [ child for child in tree.getroot() ][0]
                for child in areasRoot:
                    areas.append({
                        "name" : areaName,
                        "shape" : child.attrib["shape"],
                        "coords" : child.attrib["coords"]
                    })
                
    print(areas)

    if "ads" in metadata:
        ads = metadata["ads"]
    else:
        ads = []

    typescript = """
import { Metadata } from '../Metadata'

export default {
    "title": "%s",
    "subpath": "%s",
    "creator": "%s",
    "creatorLink": "%s",
    "ads": [
%s
    ],
    "areas": %s
} as Metadata;
""" % (metadata["title"], contentName, metadata["creator"], metadata["creator-link"], adsToTypescript(ads), areasToTypescript(areas))

    print(typescript)

    processedPath = "src/autogenerated"

    typescriptPath = os.path.join(processedPath, "%s.ts" % className)
    with open(typescriptPath, "w") as o:
        o.write(typescript)

        print("*** ADD THIS LINE TO 'src/MetadataStore.ts' ***\n")
        print("import %s from './autogenerated/%s';" % (className, className))