Python / Twisted IRC bot日志记录问题
嘿。 我最近才开始使用Python,而我的朋友建议使用Twisted作为创建此IRC bot的手段,因为它会简单得多。 这是我到目前为止的代码(这是基于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 %sn' % (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()
这个问题是机器人只记录频道消息,但我也希望它记录频道连接和部分。 什么是最简单的方法来完成这一点? 谢谢。
定义连接和零件的方法并记录它们:
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文档和源代码将来可能对您有用。
文件似乎很清楚。 像对privmsg
所做的那样覆盖userJoined
和userLeft
方法。