LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-15 10:32:28 +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

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

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