LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-23 01:46:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix-up 5a9a00e to handle exporting senders to the same DC

Browse Source
This commit is contained in:
Lonami Exo 2018-07-22 00:40:39 +02:00
parent 46b2d910d7
commit a332d29c4c
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
telethon/client/downloads.py
View File

@ -203,7 +203,20 @@ class DownloadMethods(UserMethods):
dc_id, input_location = utils.get_input_location(input_location)
exported = dc_id and self.session.dc_id != dc_id
if exported:
sender = await self._borrow_exported_sender(dc_id)
try:
sender = await self._borrow_exported_sender(dc_id)
except errors.DcIdInvalidError:
# Can't export a sender for the ID we are currently in
config = await self(functions.help.GetConfigRequest())
for option in config.dc_options:
if option.ip_address == self.session.server_address:
self.session.set_dc(
option.id, option.ip_address, option.port)
self.session.save()
break
# TODO Figure out why the session may have the wrong DC ID
sender = self._sender
else:
# The used sender will also change if ``FileMigrateError`` occurs
sender = self._sender