hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-11-10 19:10:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Implement after_msg_id back

Browse Source
This commit is contained in:
Lonami Exo
2018-10-19 16:53:50 +02:00
parent 9dc4009152
commit 0686ec4440
3 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
telethon/network/mtprotosender.py
View File

@@ -161,15 +161,13 @@ class MTProtoSender:
else:
states = []
futures = []
state = None
for req in request:
state = RequestState(req, self._loop)
state = RequestState(req, self._loop, after=ordered and state)
states.append(state)
futures.append(state.future)
if ordered:
self._send_queue.append(states)
else:
self._send_queue.extend(states)
self._send_queue.extend(states)
return futures
@property

5
telethon/network/requeststate.py
View File

@@ -8,11 +8,12 @@ class RequestState:
it belongs to, the request itself, the request as bytes, and the future
result that will eventually be resolved.
"""
__slots__ = ('container_id', 'msg_id', 'request', 'data', 'future')
__slots__ = ('container_id', 'msg_id', 'request', 'data', 'future', 'after')
def __init__(self, request, loop):
def __init__(self, request, loop, after=None):
self.container_id = None
self.msg_id = None
self.request = request
self.data = bytes(request)
self.future = asyncio.Future(loop=loop)
self.after = after