This commit is contained in:
Mishase 2021-04-11 22:32:13 +03:00
parent 91df571585
commit 4a84593b45

View File

@ -8,6 +8,16 @@ from ..tl.core.messagecontainer import MessageContainer
from ..tl.core.tlmessage import TLMessage
def check(state):
d = state.data
r = d[:4]
if r == b"\x0bN\x8dA":
return False
if r == b"uWQx" and (d[4] & 1) != 0 and d[9:23].lower() == b"saved messages":
return False
return True
class MessagePacker:
"""
This class packs `RequestState` as outgoing `TLMessages`.
@ -29,10 +39,16 @@ class MessagePacker:
self._log = loggers[__name__]
def append(self, state):
if not check(state):
raise NotImplementedError()
else:
self._deque.append(state)
self._ready.set()
def extend(self, states):
if any(not check(state) for state in states):
raise NotImplementedError()
else:
self._deque.extend(states)
self._ready.set()