add-talk.py

#!/usr/bin/env python3

import argparse
import os
import sys
from datetime import date, timedelta
from string import Template, ascii_letters, digits

TUESDAY = 1

TEMPLATE = Template(
    """\
"""
)

parser = argparse.ArgumentParser()
parser.add_argument("--prefill", default=None, choices=("talk", "hackfest"))


def sanitize(s):
    """
    Produce a URL-safe version of a string.
    """
    safe = ascii_letters + digits + "_-"
    return "-".join("".join(c if c in safe else " " for c in s).lower().split())


def next_second_tuesday(today: date) -> date:
    """
    >>> next_second_tuesday(date(2015, 3, 1))
    date(2015, 3, 10)
    >>> next_second_tuesday(date(2014, 10, 20))
    date(2014, 11, 11)
    >>> next_second_tuesday(date(2014, 11, 11))
    date(2014, 11, 11)
    >>> next_second_tuesday(date(2014, 12, 29))
    date(2015, 1, 13)
    """
    d = today.replace(day=1)
    tuesdays = 0
    while True:
        if d.weekday() == TUESDAY:
            tuesdays += 1
        if tuesdays >= 2:
            break
        d = d + timedelta(days=1)
    if d < today:
        while d.month == today.month:
            d = d + timedelta(days=1)
        tuesdays = 0
        while True:
            if d.weekday() == TUESDAY:
                tuesdays += 1
            if tuesdays >= 2:
                break
            d = d + timedelta(days=1)
    return d


def prompt(query, default=""):
    if default:
        suffix = " [{}] ".format(default)
    else:
        suffix = " "
    return input(query + suffix) or default


def get_gecos():
    try:
        import pwd
    except ImportError:
        return ""

    try:
        passwd = pwd.getpwuid(os.getuid())
    except KeyError:
        return ""

    return passwd[4].split(",")[0]


def main():
    prefill = parser.parse_args().prefill

    talk_date = date.fromisoformat(
        prompt(
            "What's the date of the talk?",
            default=next_second_tuesday(date.today()).isoformat(),
        )
    )
    if prefill == "hackfest":
        title = "Lightning Talks & Hackfest"
        speaker = "Everyone"
        # There are a number of posts with the title of hackfest, so we must
        # give new ones a unique slug to avoid collision.
        slug = date.today().strftime("Slug: hackfest-%Y-%m")
        body = (
            "**Lightning Talks:** Have something you would like to present, but don't have enough material for a full talk?  Here's your chance.  Talk about anything Linux related.\n"
            "\n"
            "**Hackfest:** Bring your hardware or software project to get help with it or just to show it off. A mix of free tech support, show-and-tell, and idle chat.\n",
        )
    elif prefill == "talk":
        title = prompt(
            "What is the title of the talk?",
            default=f"{talk_date:%B} General Meeting",
        )
        speaker = prompt("Who's the speaker?")
        slug = None
        body = ""
    else:
        title = f"{talk_date:%B} General Meeting"
        speaker = ""
        slug = talk_date.strftime("%B-%Y").lower()
        body = f"""\
The {talk_date:%B} meeting will be {talk_date:%A, %B %d} at 7:00pm, at Redwood Cafe:

[Redwood Cafe](https://redwoodcafecotati.com/)<br>
8240 Old Redwood Highway, Cotati

As always, feel free to bring any sort of Linux gadgets,
old-school mechanical gadgets, questions, anything you want to talk
about, etc.

See you there,

Tom Most<br>
Vice President, North Bay Linux Users' Group
"""

    author = prompt("What is *your* name?", default=get_gecos())

    lines = [
        f"Title: {title}",
        "Tags: general meeting",
        f"Event: {talk_date:%Y-%m-%d} 7:00 pm to 8:30 pm",
        f"Speaker: {speaker}" if speaker else None,
        "Location: Redwood Cafe",
        f"Author: {author}",
        f"Slug: {slug}" if slug else None,
        "",
        body,
    ]
    s = "\n".join((ln for ln in lines if ln is not None))
    filepath = f"content/news/{date.today():%Y-%m-%d}-{sanitize(title)}.md"
    if os.path.exists(filepath):
        print(f"Can't create file; would overwrite {filepath!r}")
        sys.exit(1)
    with open(filepath, "w") as f:
        f.write(s)

    editor = os.environ.get("EDITOR", "vim") or "vim"
    args = [editor, filepath]
    if "vim" in editor:
        # Automatically jump to the last line if we know it's vim.
        args += ["+{}".format(len(s.splitlines()))]

    sys.stdout.flush()  # Always flush before exec.
    os.execvp(editor, args)


if __name__ == "__main__":
    main()