58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
from twisted.words.protocols import irc
|
|
from twisted.internet import reactor, protocol
|
|
|
|
from .bot import Bot
|
|
from .settings import logger
|
|
|
|
class IRCBot(irc.IRCClient):
|
|
"""An IRC bot"""
|
|
|
|
def connectionMade(self):
|
|
super(IRCBot, self).connectionMade()
|
|
logger.info('{name} is connected'.format(name=self.nickname))
|
|
self.join(self.factory.channel)
|
|
|
|
def connectionLost(self, reason):
|
|
super(IRCBot, self).connectionLost(reason)
|
|
logger.info('{name} is disconnected : {reason}'.format(
|
|
name=self.nickname,
|
|
reason = reason
|
|
))
|
|
|
|
def signedOn(self):
|
|
self.join(self.factory.channel)
|
|
|
|
def joined(self, channel):
|
|
logger.info(self.nickname + ' joined ' + self.factory.channel)
|
|
if self.factory.bot.on_join is not None:
|
|
self.say(self.factory.channel, self.factory.bot.on_join)
|
|
|
|
def privmsg(self, user, channel, msg):
|
|
results = self.factory.bot.get_reaction(user, channel, msg)
|
|
logger.debug(self.nickname + ' heard ' + msg)
|
|
if results:
|
|
logger.info(self.nickname + ' reacting to ' + msg)
|
|
for r in results:
|
|
self.say(self.factory.channel, r)
|
|
|
|
|
|
class IRCBotFactory(protocol.ClientFactory):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.channel = bot.channel
|
|
|
|
def clientConnectionLost(self, connector, reason):
|
|
connector.connect()
|
|
logger.info("Client connexion lost")
|
|
|
|
def clientConnectionFailed(self, connector, reason):
|
|
logger.info("Connection failed : " + reason)
|
|
reactor.stop()
|
|
|
|
def buildProtocol(self, addr):
|
|
p = IRCBot()
|
|
p.factory = self
|
|
p.nickname = self.bot.nickname
|
|
return p
|
|
|