Sync dialogs after login
This commit is contained in:
@@ -127,7 +127,7 @@ class CommandHandler:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
user = sender.client.sign_in(code=args[0])
|
user = sender.client.sign_in(code=args[0])
|
||||||
sender.update_info(user)
|
sender.post_login(user)
|
||||||
sender.command_status = None
|
sender.command_status = None
|
||||||
return self.reply(f"Successfully logged in as @{user.username}")
|
return self.reply(f"Successfully logged in as @{user.username}")
|
||||||
except PhoneNumberUnoccupiedError:
|
except PhoneNumberUnoccupiedError:
|
||||||
@@ -168,7 +168,7 @@ class CommandHandler:
|
|||||||
|
|
||||||
try:
|
try:
|
||||||
user = sender.client.sign_in(password=args[0])
|
user = sender.client.sign_in(password=args[0])
|
||||||
sender.update_info(user)
|
sender.post_login(user)
|
||||||
sender.command_status = None
|
sender.command_status = None
|
||||||
return self.reply(f"Successfully logged in as @{user.username}")
|
return self.reply(f"Successfully logged in as @{user.username}")
|
||||||
except PasswordHashInvalidError:
|
except PasswordHashInvalidError:
|
||||||
|
|||||||
@@ -80,11 +80,14 @@ class User:
|
|||||||
update_workers=2)
|
update_workers=2)
|
||||||
self.connected = self.client.connect()
|
self.connected = self.client.connect()
|
||||||
if self.logged_in:
|
if self.logged_in:
|
||||||
self.sync_dialogs()
|
self.post_login()
|
||||||
self.update_info()
|
|
||||||
self.client.add_update_handler(self.update_catch)
|
self.client.add_update_handler(self.update_catch)
|
||||||
return self
|
return self
|
||||||
|
|
||||||
|
def post_login(self, info=None):
|
||||||
|
self.sync_dialogs()
|
||||||
|
self.update_info(info)
|
||||||
|
|
||||||
def stop(self):
|
def stop(self):
|
||||||
self.client.disconnect()
|
self.client.disconnect()
|
||||||
self.client = None
|
self.client = None
|
||||||
|
|||||||
Reference in New Issue
Block a user