Sync dialogs after login

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:

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