CloudBot/plugins/namegen.py

# Plugin by Lukeroge
from util import hook
from util.text import get_text_list
import json, random, re, os

TEMPLATE_RE = re.compile(r"\{(.+?)\}")
GEN_DIR = "./plugins/data/name_files/"

def get_generator(_json):
    data = json.loads(_json)
    return NameGenerator(data["name"], data["templates"],
        data["default_templates"], data["parts"])

class NameGenerator(object):
    def __init__(self, name, templates, default_templates, parts):
        self.name = name
        self.templates = templates
        self.default_templates = default_templates
        self.parts = parts

    def generate_name(self, template=None):
        """
        Generates one name using the specified templates. 
        If no templates are specified, use a random template from the default_templates list.
        """
        template = self.templates[template or random.choice(self.default_templates)]

        # get a list of all name parts we need
        name_parts = TEMPLATE_RE.findall(template)

        name = template

        for name_part in name_parts:
            part = random.choice(self.parts[name_part])
            name = name.replace("{%s}" % name_part, part)

        return name


    def generate_names(self, amount, template=None):
        names = []
        for i in xrange(amount):
            names.append(self.generate_name())
        return names


    def get_template(self, template):
        return self.templates[template]


@hook.command(autohelp=False)
def namegen(inp, notice=None):
    "namegen [generator] -- Generates some names using the chosen generator. " \
    "'namegen list' will display a list of all generators."

    # get a list of available name generators
    files = os.listdir(GEN_DIR)
    all_modules = []
    for i in files:
        all_modules.append(os.path.splitext(i)[0])

    # command to return a list of all available generators
    if inp == "list":
        message = "Available generators: "
        message += get_text_list(all_modules, 'and')
        notice(message)
        return

    if inp:
        selected_module = inp.split(' ')[0]
    else:
        # make some generic fantasy names
        selected_module = ["fantasy"]

    # check if the selected module is valid
    if not selected_module in all_modules:
        return "Invalid name generator :("

    # load the name generator
    with open(os.path.join(GEN_DIR, "{}.json".format(selected_module))) as f:
        generator = get_generator(f.read())

    # time to generate some names
    name_list = generator.generate_names(10)

    # and finally return the final message :D
    return "Some names to ponder: %s." % get_text_list(name_list, 'and')