LockBot.py

from twisted.words.protocols import irc
from twisted.internet import protocol

from LockBotBrain import LockBotBrain
import Logger

class LockBot(irc.IRCClient):
    def __init__(self):
        self.brain = None
        self.logger = Logger.Logger()

    def _get_nickname(self):
        return self.factory.nickname
    nickname = property(_get_nickname)

    def _get_password(self):
        return self.factory.password
    password = property(_get_password)

    def _get_dbdir(self):
        return self.factory.dbdir
    dbdir = property(_get_dbdir)

    def signedOn(self):
        self.logger.info("Signed on as %s." % (self.nickname))
        self.brain = LockBotBrain(self.nickname, self.dbdir)
        self.join(self.factory.channel)

    def joined(self, channel):
        self.logger.info("Joined %s." % (channel))

    def privmsg(self, user, channel, msg):
        self.logger.debug("received: %s" % msg)

        responses = self.brain.processPrivMsg(user, channel, msg)
        for channel, message in responses:
            self.msg(channel, message)

class LockBotFactory(protocol.ClientFactory):
    protocol = LockBot

    def __init__(self, channel, nickname, dbdir, password=None):
        self.channel = channel
        self.nickname = nickname
        self.password = password
        self.dbdir = dbdir
        self.logger = Logger.Logger()

    def clientConnectionLost(self, connector, reason):
        self.logger.critical("Lost connection (%s), reconnecting." % (reason,))
        connector.connect()

    def clientConnectionFailed(self, connector, reason):
        self.logger.critical("Could not connect: %s" % (reason,))