diff --git a/yaboli/connection.py b/yaboli/connection.py index 862ff68..d2aa03b 100644 --- a/yaboli/connection.py +++ b/yaboli/connection.py @@ -200,27 +200,27 @@ class Connection(): return str(self._send_id) - def add_callback(self, ptype, callback, *args, **kwargs): + def subscribe(self, ptype, callback, *args, **kwargs): """ - add_callback(ptype, callback, *args, **kwargs) -> None + subscribe(ptype, callback, *args, **kwargs) -> None Add a function to be called when a packet of type ptype is received. """ self._callbacks.add(ptype, callback, *args, **kwargs) - def add_id_callback(self, pid, callback, *args, **kwargs): + def subscribe_to_id(self, pid, callback, *args, **kwargs): """ - add_id_callback(pid, callback, *args, **kwargs) -> None + subscribe_to_id(pid, callback, *args, **kwargs) -> None Add a function to be called when a packet with id pid is received. """ self._id_callbacks.add(pid, callback, *args, **kwargs) - def add_next_callback(self, callback, *args, **kwargs): + def subscribe_to_next(self, callback, *args, **kwargs): """ - add_next_callback(callback, *args, **kwargs) -> None + subscribe_to_next(callback, *args, **kwargs) -> None Add a function to be called when the answer to the next message sent is received. """ diff --git a/yaboli/session.py b/yaboli/session.py index 83501b7..212c348 100644 --- a/yaboli/session.py +++ b/yaboli/session.py @@ -31,9 +31,9 @@ class Session(): def __init__(self, room, password=None): self._connection = Connection(room) - self._connection.add_callback("disconnect", self._on_disconnect) - self._connection.add_callback("bounce-event", self.handle_bounce_event) - self._connection.add_callback("ping-event", self.handle_ping_event) + self._connection.subscribe("disconnect", self._on_disconnect) + self._connection.subscribe("bounce-event", self.handle_bounce_event) + self._connection.subscribe("ping-event", self.handle_ping_event) self._callbacks = Callbacks() self._hello_event_completed = False @@ -60,7 +60,7 @@ class Session(): def name(self, new_name): with self._connection as conn: logger.debug("setting name to {!r}".format(new_name)) - conn.add_next_callback(self.handle_nick_reply) + conn.subscribe_to_next(self.handle_nick_reply) conn.send_packet("nick", name=new_name) def _on_disconnect(self): @@ -74,7 +74,7 @@ class Session(): if data.get("reason") == "authentication required": if self.password: with self._connection as conn: - conn.add_next_callback(self.handle_auth_reply) + conn.subscribe_to_next(self.handle_auth_reply) conn.send_packet("auth", type="passcode", passcode=self.password) else: logger.warn("Could not access &{}: No password.".format(self._connection.room))