LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-07-22 05:49:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

More accurate clear pendings

Browse Source
This commit is contained in:
Andrey Egorov 2017-12-12 21:22:42 +03:00
parent 7c0af2c080
commit 2f1d5e277e
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon/network/mtproto_sender.py
View File

@ -54,10 +54,11 @@ class MtProtoSender:
def is_connected(self): def is_connected(self):
return self.connection.is_connected() return self.connection.is_connected()
def disconnect(self): def disconnect(self, clear_pendings=True):
"""Disconnects from the server""" """Disconnects from the server"""
self.connection.close() self.connection.close()
self._clear_all_pending() if clear_pendings:
self._clear_all_pending()
# region Send and receive # region Send and receive
@ -75,6 +76,7 @@ class MtProtoSender:
# Finally send our packed request(s) # Finally send our packed request(s)
messages = [TLMessage(self.session, r) for r in requests] messages = [TLMessage(self.session, r) for r in requests]
self._pending_receive.update({m.msg_id: m for m in messages}) self._pending_receive.update({m.msg_id: m for m in messages})
msg_ids = [m.msg_id for m in messages]
if len(messages) == 1: if len(messages) == 1:
message = messages[0] message = messages[0]
@ -87,6 +89,12 @@ class MtProtoSender:
m.container_msg_id = message.msg_id m.container_msg_id = message.msg_id
await self._send_message(message) await self._send_message(message)
return msg_ids
def forget_pendings(self, msg_ids):
for msg_id in msg_ids:
if msg_id in self._pending_receive:
del self._pending_receive[msg_id]
async def _send_acknowledge(self, msg_id): async def _send_acknowledge(self, msg_id):
"""Sends a message acknowledge for the given msg_id""" """Sends a message acknowledge for the given msg_id"""