handle badserversalt .,.,.,,.

smart_tv_telegram/mtproto.py

@@ -2,6 +2,7 @@
 import functools
 import os
 import pickle
+import typing
 
 import pyrogram
 
@@ -11,9 +12,10 @@
 from pyrogram.raw.functions.help import GetConfig
 from pyrogram.raw.functions.messages import GetMessages
 from pyrogram.raw.functions.upload import GetFile
-from pyrogram.raw.types import InputMessageID, Message, InputDocumentFileLocation
+from pyrogram.raw.types import InputMessageID, Message, InputDocumentFileLocation, BadServerSalt
 from pyrogram.errors import FloodWait
 import pyrogram.session
+from pyrogram.raw.types.upload import File
 
 from . import Config
 
@@ -81,13 +83,13 @@ async def get_block(self, message: Message, offset: int, block_size: int) -> byt
             )
         )
 
-        while True:
+        result: typing.Optional[File] = None
+
+        while not isinstance(result, File):
             try:
                 result = await session.send(request, sleep_threshold=0)
             except FloodWait:  # file floodwait is fake
                 await asyncio.sleep(self._config.file_fake_fw_wait)
-            else:
-                break
 
         return result.bytes