choice.py

# vim:set ts=8 sts=8 sw=8 tw=80 noet cc=80:

import re
from utils import oneof

class Choice(object):
	def __init__(self, config):
		self.config = config

	def choice(self, s):
		r = r'^((hm+)?\.*\s+|.*:\s+)?([\sa-zA-ZäöüÄÖÜß-]+)\soder\s([\sa-zA-ZäöüÄÖÜß-]+)\?'
		match = re.match(r, s)
		if match:
			return [ match.group(3), match.group(4) ]
		r = r'((hm+)?\.*\s+|.*:\s+)?([\sa-zA-ZäöüÄÖÜß-]+(,\s[\sa-zA-ZäöüAÖÜß-]+)+,?\soder\s[\sa-zA-ZäöüÄÖÜß-]+)\?'
		match = re.match(r, s)
		if match:
			data = [ x.strip() for x in match.group(3).split(",") ]
			fields = data[:-1] + [ x.strip()
					for x in data[-1].split("oder ")
					if len(x.strip()) > 0 ]
			return fields
		return None

	def __call__(self, msg, nick, send_message):
		if not self.config.intrusive:
			return False
		if msg.lower().startswith("entweder"):
			return False
		choices = self.choice(msg.strip())
		if choices:
			send_message(oneof(choices))
			return True
		return False