ircbot/tbot.py

import argparse
import re

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


class IRCBot(irc.IRCClient):
    @property
    def nickname(self):
        return self.factory.nickname

    def signedOn(self):
        for c in self.factory.channels:
            self.join(c)
        print('signed on as {}.'.format(self.nickname))

    def joined(self, channel):
        print('joined {}'.format(channel))

    def privmsg(self, user, channel, msg):
        username = user.split('!', 1)[0]
        if channel == self.nickname:
            self.msg(username, "whisper back ...")
        if self.nickname in msg:
            msg = self.factory.nick_stripper.sub('', msg)
            self.msg(channel, 'echo: {}'.format(msg))


class IRCBotFactory(protocol.ClientFactory):
    protocol = IRCBot

    def __init__(self, channels, nickname='bot'):
        self.channels = channels
        self.nickname = nickname
        self.nick_stripper = re.compile(r'{}\s*?\:?'.format(self.nickname))

    def clientConnectionLost(self, connector, reason):
        print('lost connection({}), reconnecting.'.format(reason))

    def clientConnectionFailed(self, connector, reason):
        print('could not connect: {}'.format(reason))


parser = argparse.ArgumentParser()
parser.add_argument('-H', '--host', type=str, default='localhost')
parser.add_argument('-p', '--port', type=int, default=6667)
parser.add_argument('-n', '--nick', type=str, default='bot')
parser.add_argument('channels', nargs='+')

args = parser.parse_args()

reactor.connectTCP(
    args.host, int(args.port),
    IRCBotFactory(
        ('#{}'.format(c) for c in args.channels),
        nickname=args.nick
    )
)
reactor.run()