From b8db724ed298e8b2ab9e463496e1f5e184158ac6 Mon Sep 17 00:00:00 2001
From: khuxkm fbexl <khuxkm@ttm.sh>
Date: Sun, 3 Nov 2019 14:01:49 -0500
Subject: [PATCH] Add generate list

---
 commands/generate.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/commands/generate.py b/commands/generate.py
index a7af9f0..f831a6b 100644
--- a/commands/generate.py
+++ b/commands/generate.py
@@ -5,7 +5,7 @@ NAME = re.compile(r'\$([^\s]+)')
 SUB = r"#\1#"
 replace_name = lambda x: NAME.sub(SUB,x)
 
-GENERATORS = [
+GENERATORS_LIST = [
 	"book_fantasy",
 	"book_horror",
 	"book_hp",
@@ -29,8 +29,10 @@ def get_generator(name):
 
 GENERATORS = utils.LazyLoadedDict(get_generator)
 
-@plugin.command("generate")
+@plugin.command("generate","<{}>".format("|".join(GENERATORS_LIST)))
 def generate(bot,channel,nick,generator):
+	if generator not in GENERATORS_LIST:
+		bot.say(channel,"{}: Invalid generator!".format(nick))
 	res = GENERATORS[generator]()
 	if generator!="idea":
 		res = titlecase.titlecase(res)