auditbot/bot.py

import teambot,network,prefer,time,sys
from colorama import init, Fore, Style
ENDF = Style.RESET_ALL

ROOT = "/home/khuxkm/code/auditbot/"

opts = prefer.Preferences(ROOT+"prefs.json")

VERBOSE = opts.get("verbose")
COLORS = {"*":"cyan","!":"yellow","X":"red"}
def say_log(s,p="*",c=None):
	if c is None:
		c = COLORS.get(p,None)
	c = getattr(Fore,c.upper())
	print(c+"[{}] {}".format(p,s)+ENDF)

bbj = network.BBJ(port=opts.get("port",7099))
say_log("Logging in with username {} and password {}...".format(opts.get("username"),opts.get("password")))
if not bbj.set_credentials(opts.get("username"),opts.get("password")):
	say_log("Incorrect details provided","X")
	sys.exit(1)
if not opts.get("thread_id"):
	say_log("Creating thread...")
	thread = bbj.thread_create(opts.get("title"),opts.get("text"))
	opts.set("thread_id",thread["thread_id"])
say_log("...and BBJ is all set up!")

trigger = lambda x: "{}: ".format(x)
log = lambda x,e: e[len(trigger(x)):]

class AuditBot(teambot.Handler):
	def on_pubmsg(self,target,nick,text):
		nick = nick.split("|")[0]
		say_log("[{}] {}: {}".format(target,nick,text))
		if text.startswith(trigger(self._bot.bot_nick)):
			if nick in opts.get("sudoers"):
				say_log("{} is auditing \"{}\"".format(nick,log(self._bot.bot_nick,text)),"!","green")
				bbj.thread_reply(opts.get("thread_id"),opts.get("format").format(time.strftime("%Y-%m-%d %H:%M:%S"),nick,log(self._bot.bot_nick,text)))
#				say_log(repr(r))
				self.say(target,"{}: audited!".format(nick))
			else:
				say_log("{} is auditing \"{}\" but is not a sudoer!".format(nick,log(self._bot.bot_nick,text)),"X")
				self.say(target,"{}: you aren't a sudoer!".format(nick))

if __name__=="__main__":
	channels = "#sudoers".split()
	bot = teambot.TeamBot(channels,"auditor","localhost",chandler=AuditBot)
	say_log("Starting IRC bot...")
	bot.start()