Awesome IRC bot
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

67 lines
1.8 KiB

#!/usr/bin/env python
"""
.. module:: PingResponder
:synopsis: Module to repsond to irc server PING requests
.. moduleauthor:: Dave Pedu <dave@davepedu.com>
"""
from time import time, sleep
from threading import Thread
from pyircbot.modulebase import ModuleBase, hook
class PingResponder(ModuleBase):
def __init__(self, bot, moduleName):
ModuleBase.__init__(self, bot, moduleName)
self.timer = PingRespondTimer(self)
@hook("PING")
def pingrespond(self, msg, cmd):
"""Respond to the PING command"""
# got a ping? send it right back
self.bot.act_PONG(msg.trailing)
self.log.info("%s Responded to a ping: %s" % (self.bot.get_nick(), msg.trailing))
@hook("_RECV", "_SEND")
def resettimer(self, msg, cmd):
"""Resets the connection failure timer"""
self.timer.reset()
def ondisable(self):
self.timer.disable()
class PingRespondTimer(Thread):
"""
Tracks last ping from server, and reconnects if over a threshold
"""
def __init__(self, master):
Thread.__init__(self)
self.daemon = True
self.alive = True
self.master = master
self.reset()
self.start()
def reset(self):
"""
Reset the internal ping timeout counter
"""
self.lastping = time()
def disable(self):
"""
Allow the thread to die
"""
self.alive = False
def run(self):
while self.alive:
sleep(5)
if time() - self.lastping > self.master.config.get("activity_timeout", 300):
self.master.log.info("No activity in %s seconds. Reconnecting" % str(time() - self.lastping))
self.master.bot.kill("Ping timeout", forever=False)
self.reset()