68 lines
1.8 KiB
Python
Executable File
68 lines
1.8 KiB
Python
Executable File
#!/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()
|
|
|