generate-sharing-images.py

import drawBot as db
import frontmatter
import glob
import os


def saveSharingImage(outpath, title, authors, categories, theme):
    themecols = {
        "blue": [(0.0, 0.13, 0.25, 1.0), (0.95, 0.98, 1.0, 1.0), (0.68, 0.8, 0.86, 1.0)],
        "red": [(0.18, 0.0, 0.06, 1.0), (1.0, 0.96, 0.95, 1.0), (0.85, 0.67, 0.65, 1.0)],
        "sand": [(0.188,0.173, 0.051), (0.98, 0.97, 0.94), (0.84, 0.8, 0.63, 1.0)]
    }
    DR = "Design Regression".upper()
    w, h = 1200, 600
    mainpad = 40

    # get theme colours
    textcol = themecols[theme][0]
    bgcol = themecols[theme][2]
    db.newPage(w, h)
    # background
    db.fill(*bgcol)
    db.stroke(None)
    db.rect(0, 0, w, h)
    # logo
    with db.savedState():
        db.translate(mainpad, h - mainpad - 24)
        db.fill(*textcol)
        db.font("AdapterPEDisplay-Eb")
        db.openTypeFeatures(ss14=True)
        db.fontSize(24)
        db.tracking(1)
        db.text(DR, (0, 0))
        tw, _ = db.textSize(DR)
        db.lineCap("round")
        db.stroke(*textcol)
        db.strokeWidth(4)
        db.translate(0, -10)
        db.line((2, 0), (tw - 4, 0))
    # categories & authors
    padx, pady = 30, 17
    with db.savedState():
        db.translate(mainpad, h / 2 + pady)
        db.fill(*textcol)
        db.font("AdapterMonoPE-Sb")
        db.fontSize(14)
        db.tracking(1)
        tw1, th1 = db.textSize(categories)
        tw2, _ = db.textSize(authors)
        db.text(categories, (padx, 3))
        db.text(authors, (3 * padx + tw1, 3))
        # frame
        db.fill(None)
        db.stroke(*textcol)
        db.strokeWidth(1)
        boxh= th1 + 2 * pady
        db.translate(0, -pady)
        db.rect(0, 0, tw1 + tw2 + 4 * padx, boxh)
        db.translate(2 * padx + tw1, 0)
        db.line((0, 0), (0, boxh))   
    # title
    with db.savedState():
        db.translate(mainpad, mainpad)
        db.fill(*textcol)
        db.font("AdapterPEDisplay-Bl")
        db.fontSize(56)
        #db.tracking(-0.05)
        db.fill(*textcol)
        db.textBox(title, (0, 0, w - 2 * mainpad, h / 2.6))
    dirs = os.path.dirname(outpath)
    print(dirs)
    if not os.path.exists(dirs):
        os.makedirs(dirs)
        print("-----", dirs)
    db.saveImage(outpath)
    print("Saved to:", outpath)


# run this for each post
for path in glob.glob("_posts/*.md"):
    # get post meta
    print("Processing:", path)
    with open(path) as f:
        post = frontmatter.load(f)
    if "published" in post and not post["published"]:
        continue
    if "sharing_image" in post:
        outpath = post["sharing_image"][1:]  # hack: remove initial /
    else:
        name, _ = os.path.splitext(os.path.basename(path))
        outpath = "assets/%s/sharing_image.png" % name
    if "title" in post:
        title = post["title"]
    else:
        title = "Untitle"
    if "authors" in post:
        authors = (", ".join(post["authors"])).upper()
    else:
        authors = "ANONYMOUS"
    if "categories" in post:
        categories = (", ".join(post["categories"])).upper()
    else:
        categories = "POST"
    if "theme" in post:
        theme = post["theme"]
    else:
        theme = "blue"
    saveSharingImage(
        outpath, title, authors, categories, theme
    )