diff --git a/mautrix_telegram/commands.py b/mautrix_telegram/commands.py index c62537b5..ad59627e 100644 --- a/mautrix_telegram/commands.py +++ b/mautrix_telegram/commands.py @@ -127,7 +127,7 @@ class CommandHandler: try: user = sender.client.sign_in(code=args[0]) - sender.update_info(user) + sender.post_login(user) sender.command_status = None return self.reply(f"Successfully logged in as @{user.username}") except PhoneNumberUnoccupiedError: @@ -168,7 +168,7 @@ class CommandHandler: try: user = sender.client.sign_in(password=args[0]) - sender.update_info(user) + sender.post_login(user) sender.command_status = None return self.reply(f"Successfully logged in as @{user.username}") except PasswordHashInvalidError: diff --git a/mautrix_telegram/user.py b/mautrix_telegram/user.py index dc128134..ce844d9b 100644 --- a/mautrix_telegram/user.py +++ b/mautrix_telegram/user.py @@ -80,11 +80,14 @@ class User: update_workers=2) self.connected = self.client.connect() if self.logged_in: - self.sync_dialogs() - self.update_info() + self.post_login() self.client.add_update_handler(self.update_catch) return self + def post_login(self, info=None): + self.sync_dialogs() + self.update_info(info) + def stop(self): self.client.disconnect() self.client = None