93 lines
2.8 KiB
Python
Executable file
93 lines
2.8 KiB
Python
Executable file
# 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.
|
|
"""
|
|
name = self.templates[template or random.choice(self.default_templates)]
|
|
|
|
# get a list of all name parts we need
|
|
name_parts = TEMPLATE_RE.findall(name)
|
|
|
|
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."
|
|
|
|
# clean up the input
|
|
inp = inp.strip().lower()
|
|
|
|
# get a list of available name generators
|
|
files = os.listdir(GEN_DIR)
|
|
all_modules = []
|
|
for i in files:
|
|
if os.path.splitext(i)[1] == ".json":
|
|
all_modules.append(os.path.splitext(i)[0])
|
|
all_modules.sort()
|
|
|
|
# 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:
|
|
try:
|
|
generator = get_generator(f.read())
|
|
except ValueError as error:
|
|
return "Unable to read name file: {}".format(error)
|
|
|
|
# time to generate some names
|
|
name_list = generator.generate_names(10)
|
|
|
|
# and finally return the final message :D
|
|
return "Some names to ponder: {}.".format(get_text_list(name_list, 'and'))
|