From 9b44b6cd139f4e47bd57513a3af663cd2eecb571 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Mon, 11 Feb 2019 10:00:41 +0000
Subject: Shift socket.socket related logic to IRCSocket.py

---
 src/IRCBot.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'src/IRCBot.py')

diff --git a/src/IRCBot.py b/src/IRCBot.py
index 6caf720b..59b3837f 100644
--- a/src/IRCBot.py
+++ b/src/IRCBot.py
@@ -101,8 +101,9 @@ class Bot(object):
     def next_send(self) -> typing.Optional[float]:
         next = None
         for server in self.servers.values():
-            timeout = server.send_throttle_timeout()
-            if server.waiting_send() and (next == None or timeout < next):
+            timeout = server.socket.send_throttle_timeout()
+            if (server.socket.waiting_send() and
+                    (next == None or timeout < next)):
                 next = timeout
         return next
 
@@ -238,7 +239,8 @@ class Bot(object):
                         self.log.warn(
                             "Disconnected from %s, reconnecting in %d seconds",
                             [str(server), reconnect_delay])
-                elif server.waiting_send() and server.throttle_done():
+                elif (server.socket.waiting_send() and
+                        server.socket.throttle_done()):
                     self.register_both(server)
 
             for sock in list(self.other_sockets.values()):
-- 
cgit v1.3.1-10-gc9f91