New name generator :)

plugins/namegen.py

@@ -1,46 +1,87 @@
 # Plugin by Lukeroge
 from util import hook
-from util import molecular
 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 [modules] -- Generates some names using the chosen modules. " \
-    "'namegen list' will display a list of all modules."
+    "namegen [generator] -- Generates some names using the chosen generator. " \
+    "'namegen list' will display a list of all generators."
 
-    gen = molecular.Molecule()
+    # 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])
 
-    # get a list of available modules
-    all_modules = gen.list_modules()
-
-    # command to return a list of all available modules
+    # command to return a list of all available generators
     if inp == "list":
-        message = "Available modules: "
+        message = "Available generators: "
         message += get_text_list(all_modules, 'and')
         notice(message)
         return
 
     if inp:
-        # split the input into a list of modules
-        selected_modules = inp.split(' ')
+        selected_module = inp.split(' ')[0]
     else:
         # make some generic fantasy names
-        selected_modules = ["fantasy"]
+        selected_module = ["fantasy"]
 
-    # loop over the "selected_modules" list, and load any valid modules
-    for module in selected_modules:
-        if module in all_modules:
-            gen.load(module.encode('ascii'))
+    # check if the selected module is valid
+    if not selected_module in all_modules:
+        return "Invalid name generator :("
 
-    # lets try making a name to see if any modules actually got loaded
-    if not gen.name():
-        return "No valid modules specified :("
+    # 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 and put them in a list
-    name_list = []
-    for i in xrange(10):
-        name_list.append(gen.name())
+    # 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')