summaryrefslogtreecommitdiff
path: root/main.py
blob: f1ab46bdc3ef100d9fe4160846c58d00d8977281 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#!/usr/bin/env python3
import argparse
import sys

parser = argparse.ArgumentParser(prog="D&D&D&D", description="Dungeons & Dragons & D...")
parser.add_argument("word_num", default=None, nargs="?", type=int)
args = parser.parse_args()
word_num = args.word_num

WORDLIST_FILE_PATH = "nouns.txt"

with open(WORDLIST_FILE_PATH, "r") as wordlist_file:
    valid_words = set(
        [
            word[0].upper() + word.rstrip()[1:]
            for word in wordlist_file.readlines()
            if word.startswith("d")
        ]
    )

sys.stdout.write("Dungeons & Dragons")
for word in valid_words:
    if word_num is not None:
        if word_num <= 0:
            break
        word_num -= 1
    sys.stdout.write(" & " + word)
print()