dr.botzo/modules/MegaHAL.py
Brian S. Stephan 01d3c7c80c migrate some code that became pivotal to the bot into DrBotIRC.
this is a big change. DrBotIrc is now in charge of module loading
and unloading, aliases, and recursion. the Alias module is no more,
and a bunch of functionality was moved out of IrcAdmin, including
also config file saving, the sigint handler, and quitting the bot.

additionally, a lot of stuff got caught in the wake. dr.botzo.py
is simpler now, and lets DrBotIRC do the dynamic loading stuff.
Module.__init__ changed, modules no longer get modlist and instead
get a reference to the DrBotIRC object. IrcAdmin still has the same
exposed methods, but now calls out to DrBotIRC to achieve some of
them.

naturally, a recursion/alias rewrite was included with this change.
it is clearer now (i think), but probably brittle somewhere.
additionally, currently any module that has registered a pubmsg
handler can potentially fire more than once on one input (without
recursion). this may be the next thing to fix. do() may need to
be split, or maybe it's time to stop having modules deal with
pubmsg/privmsg entirely. need to decide.

WEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEEE
2011-01-07 17:38:26 -06:00

80 lines
2.6 KiB
Python

"""
MegaHAL - MegaHAL bot for IRC
Copyright (C) 2010 Brian S. Stephan
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
from ConfigParser import NoOptionError
import mh_python
import os
import re
from extlib import irclib
from Module import Module
class MegaHAL(Module):
"""Have an annoying, dumb, and hilarious chatterbot on IRC."""
def priority(self):
return 95
def __init__(self, irc, config, server):
"""Upon creation, open the MegaHAL brain and get ready for doing actual stuff."""
Module.__init__(self, irc, config, server)
mh_python.initbrain()
def save(self):
"""Sync the megahal brain."""
mh_python.cleanup()
def shutdown(self):
"""Sync the megahal brain."""
mh_python.cleanup()
def do(self, connection, event, nick, userhost, replypath, what, admin_unlocked):
"""Say something on IRC if the bot is being addressed, or at least learn what was said."""
# i think this megahal module is corrupting the object while walking it,
# so give it a stupid copy of the string. stupid stupid stupid.
line = ''.join(what)
reply = ""
append_nick = False
if re.search(connection.get_nickname(), line, re.IGNORECASE) is not None:
addressed_pattern = '^' + connection.get_nickname() + '[:,]\s+'
addressed_re = re.compile(addressed_pattern)
if addressed_re.match(what):
line = addressed_re.sub('', line)
append_nick = True
reply = mh_python.doreply(line)
elif re.search('^megahal$', line, re.IGNORECASE) is not None:
reply = mh_python.doreply('')
elif replypath is not None:
# only learn if the command is not an internal one
mh_python.learn(line)
if reply is not "":
if append_nick:
reply = nick + ": " + reply
return self.reply(connection, replypath, reply)
# vi:tabstop=4:expandtab:autoindent
# kate: indent-mode python;indent-width 4;replace-tabs on;