38 lines
956 B
Python
38 lines
956 B
Python
import plugin, utils, tracery, titlecase, re
|
|
from tracery.modifiers import base_english
|
|
|
|
NAME = re.compile(r'\$([^\s]+)')
|
|
SUB = r"#\1#"
|
|
replace_name = lambda x: NAME.sub(SUB,x)
|
|
|
|
GENERATORS = [
|
|
"book_fantasy",
|
|
"book_horror",
|
|
"book_hp",
|
|
"book",
|
|
"book_rom",
|
|
"book_sf",
|
|
"char",
|
|
"idea",
|
|
"land",
|
|
"place"
|
|
]
|
|
GENERATOR_URL = "https://github.com/cwarwicker/discord-WriterBot/raw/master/assets/json/en/gen_{}.json"
|
|
def get_generator(name):
|
|
generator = utils.get(GENERATOR_URL.format(name))
|
|
out = dict()
|
|
out.update(generator["names"])
|
|
out["origin"] = [replace_name(x) for x in generator["formats"]]
|
|
ret = tracery.Grammar(out)
|
|
ret.add_modifiers(base_english)
|
|
return (lambda: ret.flatten("#origin#"))
|
|
|
|
GENERATORS = utils.LazyLoadedDict(get_generator)
|
|
|
|
@plugin.command("generate")
|
|
def generate(bot,channel,nick,generator):
|
|
res = GENERATORS[generator]()
|
|
if generator!="idea":
|
|
res = titlecase.titlecase(res)
|
|
bot.say(channel,"{}: {}".format(nick,res))
|