X-Git-Url: https://jfr.im/git/erebus.git/blobdiff_plain/f5f2b59293a7b8adcac9491390bc88e32c535547..b4e3e62e5e48277e34e40da4f9b79b4291fabce3:/modules/help.py

diff --git a/modules/help.py b/modules/help.py
index fe3340c..770a0fb 100644
--- a/modules/help.py
+++ b/modules/help.py
@@ -94,28 +94,45 @@ def _mkhelp(level, func):
 	return lines
 
 def _genhelp(bot, user, chan, realtarget, *args):
-	try:
-		filepath = bot.parent.cfg.get('help', 'path', default='./help/%d.txt')
-		for level in range(-1, 101):
+	module = None
+	minlevel = -1
+	maxlevel = 100
+	filepath = bot.parent.cfg.get('help', 'path', default='./help/%d.txt')
+	for arg in args:
+		if arg.startswith("@"):
+			module = arg[1:]
+		elif arg.startswith("#") and user.glevel >= lib.ADMIN:
+			minlevel = maxlevel = int(arg[1:])
+		else:
+			filepath = arg
+			if minlevel != maxlevel:
+				minlevel = maxlevel
+	print "%r %r %r %r" % (module, minlevel, maxlevel, filepath)
+	for level in range(minlevel, maxlevel+1):
+		if '%d' in filepath:
 			filename = filepath % (level)
-			fo = open(filename, 'w')
-			lines = []
-			for func in helps.itervalues():
-				lines += _mkhelp(level, func)
-			for line in sorted(lines):
-				fo.write(str(line)+"\n")
-	except Exception as e:
-		return e
+		else:
+			filename = filepath
+		fo = open(filename, 'w')
+		lines = []
+		for func in helps.itervalues():
+			if module is not None and func.module != module:
+					continue
+			lines += _mkhelp(level, func)
+		for line in sorted(lines):
+			fo.write(str(line)+"\n")
+		fo.close()
 	return True
 
 @lib.hook(glevel=1, needchan=False)
-@lib.help(None, "generates help file", "default path: ./help/<level>.txt", "config as: [help]", "path = ./help/%d.txt")
+@lib.help("[@<module>] [#<level>] [file]", "generates help file", "arguments are all optional and may be specified in any order", "default file: ./help/<level>.txt", "config as: [help]", "path = ./help/%d.txt")
 def genhelp(bot, user, chan, realtarget, *args):
-	ret = _genhelp(bot, user, chan, realtarget, *args)
-	if not isinstance(ret, BaseException):
-		bot.msg(user, "Help written.")
-	else:
-		bot.msg(user, "Failed writing help. %s" % (ret))
+	try:
+		_genhelp(bot, user, chan, realtarget, *args)
+	except Exception as e:
+		bot.msg(user, "Failed writing help. %s" % (e))
+		return
+	bot.msg(user, "Help written.")
 
 @lib.hook(needchan=False)
 @lib.help("<command>", "describes a command")
@@ -137,7 +154,10 @@ def help(bot, user, chan, realtarget, *args):
 @lib.help(None, "provides command list")
 def showcommands(bot, user, chan, realtarget, *args):
 	if bool(int(bot.parent.cfg.get('help', 'autogen', default=0))):
-		_genhelp(bot, user, chan, realtarget, *args)
+		try:
+			_genhelp(bot, user, chan, realtarget, *args)
+		except: pass
+
 	url = bot.parent.cfg.get('help', 'url', default=None)
 	if url is None:
 		try:
@@ -162,7 +182,7 @@ def help(bot, user, chan, realtarget, *args):
 		for line in sorted(lines):
 			bot.slowmsg(user, str(line))
 		bot.slowmsg(user, "End of command listing.")
-	elif args[0][0] == "@":
+	elif args[0].startswith("@"):
 		lines = []
 		mod = args[0][1:].lower()
 		for func in helps.itervalues():