Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

store and expose reconnection timers on IRCBot.reconnections

Browse source
This commit is contained in:
jesopo 2019-09-01 08:38:22 +01:00
parent 85efd75518
commit 19d8f0b497
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/IRCBot.py
View file

@ -40,6 +40,7 @@ class Bot(object):
self.start_time = time.time() self.start_time = time.time()
self.running = False self.running = False
self.servers = {} self.servers = {}
self.reconnections = {}
self._event_queue = queue.Queue() # type: typing.Queue[TriggerEvent] self._event_queue = queue.Queue() # type: typing.Queue[TriggerEvent]
@ -217,6 +218,8 @@ class Bot(object):
if not self.reconnect(event["server_id"], if not self.reconnect(event["server_id"],
event.get("connection_params", None)): event.get("connection_params", None)):
event["timer"].redo() event["timer"].redo()
else:
del self.reconnections[event["server_id"]]
def reconnect(self, server_id: int, connection_params: typing.Optional[ def reconnect(self, server_id: int, connection_params: typing.Optional[
utils.irc.IRCConnectionParameters]=None) -> bool: utils.irc.IRCConnectionParameters]=None) -> bool:
args = {} # type: typing.Dict[str, str] args = {} # type: typing.Dict[str, str]
@ -387,8 +390,11 @@ class Bot(object):
if not self.get_server_by_id(server.id): if not self.get_server_by_id(server.id):
reconnect_delay = self.config.get("reconnect-delay", 10) reconnect_delay = self.config.get("reconnect-delay", 10)
self._timers.add("reconnect", reconnect_delay,
timer = self._timers.add("reconnect", reconnect_delay,
server_id=server.id) server_id=server.id)
self.reconnections[server.id] = timer
self.log.warn( self.log.warn(
"Disconnected from %s, reconnecting in %d seconds", "Disconnected from %s, reconnecting in %d seconds",
[str(server), reconnect_delay]) [str(server), reconnect_delay])