From 4cd422493b3a50df2ab8e923e357def4fd8e8f38 Mon Sep 17 00:00:00 2001 From: Joscha Date: Wed, 10 Apr 2019 17:07:30 +0000 Subject: [PATCH] Update own session on who command --- yaboli/room.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/yaboli/room.py b/yaboli/room.py index 2fb8a92..8526736 100644 --- a/yaboli/room.py +++ b/yaboli/room.py @@ -460,11 +460,13 @@ class Room: reply = await self._connection.send("who", {}) data = self._extract_data(reply) - own_id = self._session.session_id if self._session is not None else None - self._users = LiveSessionListing.from_data( - self, - data["listing"], - exclude_id = own_id - ) + users = LiveSessionListing.from_data(self, data["listing"]) + # Assumes that self._session is set (we're connected) + session = users.get(self.session.session_id) + if session is not None: + self._session = session + self._users = users.with_part(self._session) + else: + self._users = users return self._users