sotdbot/bot.py

import teambot, sys

USAGE = dict(admin="Stop trying to use this command!",sotd="{}sotd <link> <name>")
PUBLIC_COMMANDS = ["sotd"]

SOTDS = dict()

def get_title(link):
	return link # TODO: actually make this get the title of a link.

class SOTDBot(teambot.Handler,teambot.CommandHandlerMixin):
	def __init__(self,*args):
		self.prefix = "!"
		self.botop = "khuxkm"
		super(SOTDBot,self).__init__(*args)
	def handle_command(self,target,nick,text):
		if not text.startswith(self.prefix):
			return
		self.target = target if target!=self._bot.nick else nick
		self.sourcenick = nick
		args = text[len(self.prefix):].split()
		cmd = args.pop(0)
		if hasattr(self,"do_"+cmd.lower()):
			try:
				getattr(self,"do_"+cmd.lower())(*args)
			except TypeError as e:
				self.say(self.target,"{}: Usage: {}".format(nick,USAGE[command.lower()].format(self.prefix)))
		elif text in ("!botlist","!rollcall"):
			self.do_botlist()

	def do_admin(self,*args):
		if self.sourcenick!=self.botop:
			self.say(target,self.sourcenick+": You can't tell me what to do!")
			return
		args = list(args)
		subcmd = args.pop(0)
		if subcmd=="down":
			self._bot.die("Bye bye!")
		elif subcmd=="prefix":
			self.prefix = args[0]

	def do_botlist(self):
		self.say(self.target,"Maintainer: khuxkm | commands: {}".format(", ".join("{}{}".format(self.prefix,cmd) for cmd in PUBLIC_COMMANDS)))

	def do_sotd(self,link,*name):
		if name:
			name = " ".join(name)
		else:
			name = get_title(link)
		SOTDS[self.sourcenick] = (name,link)
		self.say(self.target,self.sourcenick+": Your SOTD has been set to {}.".format(name))

if __name__=="__main__":
	channels = "#bots".split()
	bot = teambot.TeamBot(channels,"sotdbot","localhost",chandler=SOTDBot)
	bot.start()