arrrrrBot.py

# encoding=utf-8

from config import CHATROOM_PRESENCE
from errbot import BotPlugin, botcmd
from errbot.utils import get_sender_username


class ArrrrrBot(BotPlugin):
	@botcmd
	def slap(self, mess, args):
		"""Let the bot slap people in the chatroom in an piratelike manner"""
		return "{0} slaps {1} around a bit with a large trout.".format(get_sender_username(mess), args)

	@botcmd
	def peek(self, mess, args):
		"""Print a saved one-liner"""
		argss = args.split(' ', 1)
		if argss[0] in self:
			return self[argss[0]]
		else:
			return "Nope."

	@botcmd
	def p(self, mess, args):
		"""Short alias for !peek"""
		return self.peek(mess, args)

	@botcmd
	def rm(self, mess, args):
		"""Alias for removing something from the one-liner memory"""
		argss = args.split(' ')
		for i in range(0, len(argss)):
			self.poke(mess, argss[i])
		return "...Done."

	@botcmd
	def poke(self, mess, args):
		"""Save a one-liner : !poke name content"""
		argss = args.split(' ', 1)
		name = argss[0]
		if len(argss) == 1:
			if name in self:
				del self[name]
			else:
				return "Nope."
		else:
			content = argss[1]
			if len(content) > 80 or '\n' in content:
				return "Nope."
			else:
				self[name] = content
				return "Got it."

	@botcmd
	def peekpoke(self, mess, args):
		"""List all saved one-liners (private message only)"""
		result = ""
		for name in self.keys():
			result += "{0}: \t {1}\n".format(name, self[name])
		return result

	@botcmd(admin_only=True)
	def hiersprichtgott(self, mess, args):
		self.send(CHATROOM_PRESENCE[0], args, message_type='groupchat')

	def callback_message(self, conn, mess):
		firstword = mess.getBody()
		try:
			firstword = firstword[:firstword.index(' ')]
		except ValueError:
			pass
		if firstword in self:
			self.send(mess.getFrom(), self[firstword], message_type=mess.getType())