Implement after_msg_id back

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

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