Проблема входа в бот Python / Twisted IRC - PullRequest
Новая
       6

Проблема входа в бот Python / Twisted IRC

0 голосов
/ 01 февраля 2011

Эй. Я только недавно начал использовать Python, и мой друг предложил использовать Twisted в качестве средства для создания этого IRC-бота, так как это будет намного проще. Вот код, который у меня есть (он основан на logbot.py, ха-ха) 

from twisted.internet import reactor, protocol
from twisted.words.protocols import irc
from twisted.python import log
import sys, time

class MessageLogger:
    def __init__(self, file):
        self.file = file

    def log(self, message):
        timestamp = time.strftime("[%H:%M:%S]", time.localtime(time.time()))
        self.file.write('%s %s\n' % (timestamp, message))
        self.file.flush()

    def close(self):
        self.file.close()

class IRCProtocol(irc.IRCClient):
    nickname = "3n7rar3"
    def connectionMade(self):
        irc.IRCClient.connectionMade(self)
        self.logger = MessageLogger(open(self.factory.filename, "a"))

    def signedOn(self):
        print 'Success! Connection established.'
        self.join(self.factory.channels)
        print 'Joined channel', self.factory.channels

    def privmsg(self, user, channel, msg):
        user = user.split('!', 1)[0]
        self.logger.log("<%s> %s" % (user, msg))

class IRCFactory(protocol.ClientFactory):
    protocol = IRCProtocol
    channels = "#testing"

    def __init__(self, channel, filename):
        self.channel = channel
        self.filename = filename

    def clientConnectionFailed(self, connector, reason):
        print "Connection failed because of %s" % reason
        reactor.stop()

    def clientConnectionLost(self, connector, reason):
        print "Connection lost: %s" % reason
        connector.connect()

if __name__ == "__main__":
    log.startLogging(sys.stdout)
    host, port = "irc.freenode.net", 6667
    fact = IRCFactory(sys.argv[1],sys.argv[2])
    reactor.connectTCP(host, port, fact)
    reactor.run()

Проблема в том, что бот регистрирует только сообщения канала, но я также хотел бы, чтобы он регистрировал соединения каналов и их части. Какой самый простой способ сделать это? Спасибо.

Ответы [ 2 ]

2 голосов
/ 01 февраля 2011

Определите методы для соединений и деталей и войдите в них тоже: 

def userJoined(self, user, channel):
    log.msg('%s has joined %s' % (user, channel))

def userLeft(self, user, channel):
    log.msg('%s has left %s' % (user, channel))

def userQuit(self, user, quitMessage):
    log.msg('%s has quit. Reason: %s' % (user, quitMessage))

Документация API и исходный код , вероятно, пригодятся вам в будущем.

1 голос
/ 01 февраля 2011

Документация мне кажется довольно понятной.Переопределите методы userJoined и userLeft, как вы сделали для privmsg.

...