koboldgen/koboldgen.py

import random as r
import argparse

beg = ["a","e","i","o","u","ba","be","bi","bo","bu","by","y","da","de","di","do","du","dy","fa","fi","fo","fe","fu","ga","ge","gi","go","gu","ka","ke","ki","ko","ku","ky","ma","me","mi","mo","mu","na","ne","ni","no","nu","pa","pe","pi","po","pu","ra","re","ri","ro","ru","ry","sa","se","si","so","su","ta","te","ti","to","tu","ty","wa","we","wi","wo","wy","za","ze","zi","zo","zu","zy"]
mid = beg + ["l","x","n","r"]

def gen_name(n=None):
    if n == None:
        lgt = r.randint(3,9)
    else:
        lgt = n
    morae = []
    while len("".join(morae)) < lgt:
        if len(morae) == 0:
            mora = r.choice(beg)
            morae.append(mora[0].upper() + mora[1:])
        else:
            mora = r.choice(mid)
            if morae[-1] == mora:
                mora = r.choice(mid)
            morae.append(mora)
    return "".join(morae)


def gen_stats(n):
    if n < 4:
        print("Too few stat points!")
        return [0,0,0,0]
    stats = [1,1,1,1]
    points = n - 4
    slots = [0,1,2,3]
    for _ in range(points):
        tgl = False
        while tgl == False:
            slt = r.choice(slots)
            if stats[slt] == 6:
                continue
            if stats[slt] == 5:
                if r.randint(0,1) == 1:
                    continue
            stats[slt] += 1
            tgl = True
    return stats

def gen_career():
    return r.choice(["Soldier/Guard","Pilot","Medic","Mechanic","Politician","Spellcaster","Performer","Historian","Spy","Cook","Cartographer"])

def gen_plot():
    loc1 = ["a friendly","a hostile","a derelict","an airless","a poison-filled/covered","an overgrown","a looted","a burning","a frozen","a haunted","an infested"]
    loc2 = ["asteroid","moon","space station","spaceship","ringworld","Dyson sphere","planet","Space Whale","pocket of folded space","time vortex","Reroll"]
    miss = ["to explore!","to loot everything not bolted down too securely","to find the last group of kobolds who came here","to find a rumored secret weapon","to find a way to break someone else’s secret weapon","to claim this place in the name of the Kobold Empire!","to make friends!","to rediscover lost technology","to find lost magical items","to find and defeat a powerful enemy","Reroll"]
    prob = ["an army of undead","a rival band of kobolds","a detachment from the Elf Armada","refugees with parasites. Big parasites","an artificial intelligence bent on multi-world domination","robot spiders","semi-intelligent metal eating slime","a living asteroid that intends to follow the kobolds home like the largest puppy","an old lich that wants everyone to stay off of their “lawn”","elder gods hailing from the dark spaces between the stars","Reroll"]
    cl1 = r.choice(loc1)
    cl2 = r.choice(loc2)
    cm = r.choice(miss)
    cp = r.choice(prob)
    if cl2 == "Reroll":
        cl2p = r.choice(loc2[:-1])
        if cl2p == "Asteroid":
            cl2 = "battlefield on an asteroid"
        else:
            cl2 = "battlefield on a " + cl2p
    if cm == "Reroll":
        cm = "to ...well, the kobolds weren't paying attention, so don't tell them, but they're supposed to " + r.choice(miss[:-1])
    if cp == "Reroll":
        cp = "a Floating Brain Monster and its minion, " + r.choice(prob[:-1])
    return [cl1, cl2, cm, cp]

# def gen_transpo():
#     name1 = ["Red","Orange","Yellow","Green","Blue","Violet","Dark","Light","Frenzied","Maniacal","Ancient"]
#     name2 = ["Moon","Comet","Star","Saber","World-Eater","Dancer","Looter","Phlogiston","Fireball","Mecha","Raptor"]
#     prop1 = ["is stealthy & unarmored","is speedy & unarmored","is maneuverable & unarmored","is always repairable","is self-repairing","is flamboyant & speedy","is slow & armored","is flamboyant & armored","is hard to maneuver & armored","has Too Many Weapons!","has a prototype hyperdrive"]
#     prop2 = ["has an annoying AI","has inconveniently crossed circuits","has an unpredictable power source","drifts to the right","is haunted","was recently 'found' so the kobolds are unused to it","is too cold","has a constant odd smell","its interior design... changes","its water pressure shifts between slow drip and power wash","it leaves a visible smoke trail"]
#     return [r.choice(name1), r.choice(name2), r.choice(prop1), r.choice(prop2)]

class Character:
    def __init__(self):
        self.name = ""
        self.career = ""
        self.stats = []

    def generate(self):
        self.gen_name()
        self.gen_stats()
        self.gen_career()

    def gen_name(self):
        self.name = gen_name()

    def gen_stats(self):
        self.stats = gen_stats(12)

    def gen_career(self):
        self.career = gen_career()

    def print_name(self):
        print(f"Name: {self.name} (Kobold {self.career})")

    def print(self):
        self.print_name()
        print(f"Order: {self.stats[0]}")
        print(f"Chaos: {self.stats[1]}")
        print(f"Brain: {self.stats[2]}")
        print(f"Body: {self.stats[3]}")

class Ship:
    def __init__(self):
        self.name1 = r.choice(["Red","Orange","Yellow","Green","Blue","Violet","Dark","Light","Frenzied","Maniacal","Ancient"])
        self.name2 = r.choice(["Moon","Comet","Star","Saber","World-Eater","Dancer","Looter","Phlogiston","Fireball","Mecha","Raptor"])
        self.gqual = r.choice(["is stealthy & unarmored","is speedy & unarmored","is maneuverable & unarmored","is always repairable","is self-repairing","is flamboyant & speedy","is slow & armored","is flamboyant & armored","is hard to maneuver & armored","has Too Many Weapons!","has a prototype hyperdrive"])
        self.bqual = r.choice(["has an annoying AI","has inconveniently crossed circuits","has an unpredictable power source","drifts to the right","is haunted","was recently 'found' so the kobolds are unused to it","is too cold","has a constant odd smell","its interior design... changes","its water pressure shifts between slow drip and power wash","it leaves a visible smoke trail"])
        self.fullname = f"{self.name1} {self.name2}"

    def print(self):
        print(f"The {self.fullname} {self.gqual}, but {self.bqual}.")

class Campaign:
    def __init__(self, n=None):
        n = 6 if n == None else n
        self.ship = Ship()
        self.params = gen_plot()
        self.characters = []
        for _ in range(n):
            c = Character()
            c.generate()
            self.characters.append(c)
        self.art = "an" if self.params[1][0] in ["A","E","I","O","U"] else "a"

    def print_params(self):
        print(f"The Kobolds of the {self.ship.fullname}")
        print(f"have been sent out to {self.art} {self.params[0]} {self.params[1]}")
        print(f"in order {self.params[2]}")
        print(f"but they're challenged by {self.params[3]}!")
        print()
        self.ship.print()

    def print_chars(self):
        print("The kobolds:")
        for k in self.characters:
            k.print()

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    group = parser.add_mutually_exclusive_group()
    group.add_argument("-c", "--campaign", help="print a full campaign block with N kobolds (default 6)", nargs="?", const=6, type=int, metavar="N")
    group.add_argument("-k", "--kobolds", help="print N kobolds", type=int, nargs="?", const=1, default=1, metavar="N")
    group.add_argument("-n", "--names", help="print N kobolds without stat blocks", nargs="?", const=1, type=int, metavar="N")
    group.add_argument("-p", "--params", help="print only the parameters of a campaign", action="store_true")
    group.add_argument("-s", "--ship", help="print only the ship name and description", action="store_true")
    args = parser.parse_args()
    # print(args)
    if args.campaign:
        cmp = Campaign(args.campaign)
        cmp.print_params()
        cmp.print_chars()
    elif args.params:
        cmp = Campaign()
        cmp.print_params()
    elif args.ship:
        ship = Ship()
        ship.print()
    elif args.names:
        for _ in range(args.names):
            c = Character()
            c.generate()
            c.print_name()
    else:
        for _ in range(args.kobolds):
            c = Character()
            c.generate()
            c.print()