blackout.py

#!/usr/bin/env python
"""
Black out all repeated words, preserving punctuation.

For NaNoGenMo 2022.
https://github.com/NaNoGenMo/2022/
"""

import argparse
import random
import re
import sys

all_words = set()


def is_word(thing):
    found = re.match(r"\w+", thing, re.UNICODE)
    return found


def meow_meow(line, converter_fun):
    """Meowify a line"""
    meowed = []
    # Break line into words and non-words (e.g. punctuation and space)
    things = re.findall(r"\w+|[^\w]", line, re.UNICODE)
    for thing in things:
        if is_word(thing):
            meowed.append(converter_fun(thing))
        else:
            meowed.append(thing)
    return "".join(meowed)


def blackout(word):
    if word.lower() in all_words:
        return len(word) * "█"
    all_words.add(word.lower())
    return word


def meow(word):
    """Meowify a word"""
    meowed = ""
    length = len(word)

    if length == 1:
        return capify("m", word)
    elif length == 2:
        return capify("me", word)
    elif length == 3:
        return capify("mew", word)
    elif length == 4:
        return capify("meow", word)

    # Words longer than four will have:
    #  * first letter M
    #  * last letter W
    #  * middle with a random number of Es, then some Os

    # Number of EOs:
    eeohs = length - len("m") - len("w")
    # Number of Es:
    ees = random.randrange(1, eeohs)
    # Number of Os:
    ohs = eeohs - ees

    meowed = "m" + ("e" * ees) + ("o" * ohs) + "w"
    return capify(meowed, word)


def capify(word, reference):
    """Make sure word has the same capitalisation as reference"""
    new_word = ""

    # First check whole word before char-by-char
    if reference.islower():
        return word.lower()
    elif reference.isupper():
        return word.upper()

    # Char-by-char checks
    for i, c in enumerate(reference):
        if c.isupper():
            new_word += word[i].upper()
        else:
            new_word += word[i]
    return new_word


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description="Replace all words with meows, preserving punctuation."
    )
    parser.add_argument(
        "infile",
        nargs="?",
        type=argparse.FileType("r"),
        default=sys.stdin,
        help="Input text",
    )
    parser.add_argument(
        "-t",
        "--translation",
        action="store_true",
        help="Output a line-by-line translation",
    )
    args = parser.parse_args()

    # for line in fileinput.input(openhook=fileinput.hook_encoded("utf-8")):
    for line in args.infile:
        line = line.rstrip()  # No BOM
        if args.translation:
            print()
            print(line)
        print(meow_meow(line, blackout))

# End of file