diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/IRCBot.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/src/IRCBot.py b/src/IRCBot.py index 8760f610..b1060dbf 100644 --- a/src/IRCBot.py +++ b/src/IRCBot.py @@ -36,7 +36,7 @@ class ListLambdaPollHook(PollHook.PollHook): class Bot(object): def __init__(self, directory, args, cache, config, database, events, - exports, log, modules, timers, lock_file): + exports, log, modules, timers): self.directory = directory self.args = args self.cache = cache @@ -68,11 +68,7 @@ class Bot(object): self._read_thread = None self._write_thread = None - self._poll_timeouts = [] # typing.List[PollHook] - self._poll_timeouts.append(self._timers) - self._poll_timeouts.append(self.cache) - self._poll_timeouts.append(lock_file) - + self._poll_timeouts = [] # typing.List[PollHook.PollHook] self._poll_timeouts.append(ListLambdaPollHook( lambda: self.servers.values(), lambda server: server.until_read_timeout())) @@ -84,6 +80,9 @@ class Bot(object): self._poll_timeouts.append(ListLambdaPollHook( lambda: self.servers.values(), self._throttle_timeout)) + def add_poll_hook(self, hook: PollHook.PollHook): + self._poll_timeouts.append(hook) + def _throttle_timeout(self, server: IRCServer.Server): if server.socket.waiting_throttled_send(): return server.socket.send_throttle_timeout() |