Add on_open() callback

05bit · Oct 9, 2016 · 560eada · 560eada
1 parent 1e618fc
commit 560eada
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/bachata/tornado.py b/bachata/tornado.py
@@ -40,6 +40,7 @@ def open_async(self):
         if channel:
             mc = self.get_messages_center()
             mc.add_socket(channel, self)
+            yield from self.on_open()
         else:
             io_loop = tornado.ioloop.IOLoop.current()
             io_loop.add_callback(self.on_auth_error)
@@ -54,8 +55,15 @@ def authenticate(self):
         """
         return
 
+    @asyncio.coroutine
+    def on_open(self):
+        """Is called on connection success.
+        """
+        pass
+
     def on_close(self):
-        """Remove handler from messages center."""
+        """Is called on connection failure.
+        """
         self.get_messages_center().del_socket(self.get_channel(), self)
 
     def on_message(self, raw_message):