LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-04-21 01:21:59 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix issues

Browse Source
This commit is contained in:
Jahongir Qurbonov 2024-10-24 10:43:51 +05:00
parent 6b694b241f
commit 13c870d114
1 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
client/src/telethon/_impl/mtsender/sender.py
View File

@ -258,8 +258,9 @@ class Sender:
async def _do_read(self) -> None:
self._step_done.clear()
timeout = self._next_ping - asyncio.get_running_loop().time()
try:
async with asyncio.timeout(PING_DELAY):
async with asyncio.timeout(timeout):
recv_data = await self._reader.read(MAXIMUM_DATA)
except TimeoutError:
pass
@ -271,15 +272,9 @@ class Sender:
async def _do_write(self) -> None:
self._step_done.clear()
try:
async with asyncio.timeout(PING_DELAY):
await self._try_fill_write()
except TimeoutError:
pass
finally:
self._try_timeout_ping()
self._step_done.set()
await self._try_fill_write()
self._try_timeout_ping()
self._step_done.set()
async def _try_fill_write(self) -> None:
if not self._requests:
@ -297,12 +292,12 @@ class Sender:
container_msg_id, mtp_buffer = result
self._transport.pack(mtp_buffer, self._writer.write)
await self._writer.drain()
for request in self._requests:
if isinstance(request.state, Serialized):
request.state = Sent(request.state.msg_id, container_msg_id)
await self._writer.drain()
def _try_timeout_ping(self) -> None:
current_time = asyncio.get_running_loop().time()