Shut down event loop after disconnecting.

Author: shizmob

pydle/client.py

@@ -45,6 +45,7 @@ def __init__(self, nickname, fallback_nicknames=[], username=None, realname=None
         self.username = username or nickname.lower()
         self.realname = realname or nickname
         self.eventloop = None
+        self.own_eventloop = True
         self._reset_connection_attributes()
         self._reset_attributes()
 
@@ -95,6 +96,7 @@ def connect(self, hostname=None, port=None, reconnect=False, eventloop=None, **k
             self.eventloop = eventloop
         elif not self.eventloop:
             self.eventloop = async.EventLoop()
+        self.own_eventloop = not eventloop
 
         # Reset attributes and connect.
         if not reconnect:
@@ -121,6 +123,10 @@ def disconnect(self, expected=True):
             # Callback.
             self.on_disconnect(expected)
 
+            # Shut down event loop.
+            if expected and self.own_eventloop:
+                self.connection.stop()
+
             # Reset any attributes.
             self._reset_attributes()
 

pydle/connection.py

@@ -223,6 +223,10 @@ def run_forever(self):
         self.eventloop.run()
         self.remove_handlers()
 
+    def stop(self):
+        """ Stop event loop. """
+        self.eventloop.schedule(lambda: self.eventloop.stop())
+
 
     ## Handler setup and teardown.