spacepaste

Paste details

reply | raw

  1.  
  2. from __future__ import print_function
  4. # twisted imports
  5. from twisted.words.protocols import irc
  6. from twisted.internet import reactor, protocol
  7. from twisted.python import log
  9. # system imports
  10. import time, sys
  13. class MessageLogger:
  14. """
  15. An independent logger class (because separation of application
  16. and protocol logic is a good thing).
  17. """
  18. def __init__(self, file):
  19. self.file = file
  21. def log(self, message):
  22. """Write a message to the file."""
  23. timestamp = time.strftime("[%H:%M:%S]", time.localtime(time.time()))
  24. self.file.write('%s %s\n' % (timestamp, message))
  25. self.file.flush()
  27. def close(self):
  28. self.file.close()
  31. class LogBot(irc.IRCClient):
  32. """A logging IRC bot."""
  34. nickname = "twistedbot"
  36. def connectionMade(self):
  37. irc.IRCClient.connectionMade(self)
  38. self.logger = MessageLogger(open(self.factory.filename, "a"))
  39. self.logger.log("[connected at %s]" %
  40. time.asctime(time.localtime(time.time())))
  42. def connectionLost(self, reason):
  43. irc.IRCClient.connectionLost(self, reason)
  44. self.logger.log("[disconnected at %s]" %
  45. time.asctime(time.localtime(time.time())))
  46. self.logger.close()
  49. # callbacks for events
  51. def signedOn(self):
  52. """Called when bot has successfully signed on to server."""
  53. self.join(self.factory.channel)
  55. def joined(self, channel):
  56. """This will get called when the bot joins the channel."""
  57. self.logger.log("[I have joined %s]" % channel)
  59. def privmsg(self, user, channel, msg):
  60. """This will get called when the bot receives a message."""
  61. user = user.split('!', 1)[0]
  62. self.logger.log("<%s> %s" % (user, msg))
  64. # Check to see if they're sending me a private message
  65. if channel == self.nickname:
  66. msg = "It isn't nice to whisper! Play nice with the group."
  67. #run function
  68. runfunction(self)
  69. self.msg(user, msg)
  70. return
  71. # Otherwise check to see if it is a message directed at me
  72. if msg.startswith(self.nickname + ":"):
  73. msg = "%s: I am a log bot" % user
  74. self.msg(channel, msg)
  75. self.logger.log("<%s> %s" % (self.nickname, msg))
  77. def runfuntion(self):
  78. print("-----Ran function")
  80. def action(self, user, channel, msg):
  81. """This will get called when the bot sees someone do an action."""
  82. user = user.split('!', 1)[0]
  83. self.logger.log("* %s %s" % (user, msg))
  85. # irc callbacks
  87. def irc_NICK(self, prefix, params):
  88. """Called when an IRC user changes their nickname."""
  89. old_nick = prefix.split('!')[0]
  90. new_nick = params[0]
  91. self.logger.log("%s is now known as %s" % (old_nick, new_nick))
  94. # For fun, override the method that determines how a nickname is changed on
  95. # collisions. The default method appends an underscore.
  96. def alterCollidedNick(self, nickname):
  97. """
  98. Generate an altered version of a nickname that caused a collision in an
  99. effort to create an unused related name for subsequent registration.
  100. """
  101. return nickname + '^'
  105. class LogBotFactory(protocol.ClientFactory):
  106. """A factory for LogBots.
  108. A new protocol instance will be created each time we connect to the server.
  109. """
  111. def __init__(self, channel, filename):
  112. self.channel = channel
  113. self.filename = filename
  115. def buildProtocol(self, addr):
  116. p = LogBot()
  117. p.factory = self
  118. return p
  120. def clientConnectionLost(self, connector, reason):
  121. """If we get disconnected, reconnect to server."""
  122. connector.connect()
  124. def clientConnectionFailed(self, connector, reason):
  125. print("connection failed:", reason)
  126. reactor.stop()
  129. if __name__ == '__main__':
  130. # initialize logging
  131. log.startLogging(sys.stdout)
  133. # create factory protocol and application
  134. f = LogBotFactory(sys.argv[1], sys.argv[2])
  136. # connect factory to this host and port
  137. reactor.connectTCP("irc.freenode.net", 6667, f)
  139. # run bot
  140. reactor.run()
  141.  
Powered by spacepaste. Paste removal.