mirror of
https://github.com/LonamiWebs/Telethon.git
synced 2024-11-27 03:43:45 +03:00
be279ce3f5
This simplifies the flow instead of having separate request/body attributes, and also means that BinaryReader.tgread_object() can be used without so many special cases.
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import logging
|
|
import struct
|
|
|
|
from .tlmessage import TLMessage
|
|
from ..tlobject import TLObject
|
|
|
|
__log__ = logging.getLogger(__name__)
|
|
|
|
|
|
class MessageContainer(TLObject):
|
|
CONSTRUCTOR_ID = 0x73f1f8dc
|
|
|
|
def __init__(self, messages):
|
|
super().__init__()
|
|
self.content_related = False
|
|
self.messages = messages
|
|
|
|
def to_dict(self, recursive=True):
|
|
return {
|
|
'content_related': self.content_related,
|
|
'messages':
|
|
([] if self.messages is None else [
|
|
None if x is None else x.to_dict() for x in self.messages
|
|
]) if recursive else self.messages,
|
|
}
|
|
|
|
def __bytes__(self):
|
|
return struct.pack(
|
|
'<Ii', MessageContainer.CONSTRUCTOR_ID, len(self.messages)
|
|
) + b''.join(bytes(m) for m in self.messages)
|
|
|
|
def __str__(self):
|
|
return TLObject.pretty_format(self)
|
|
|
|
def stringify(self):
|
|
return TLObject.pretty_format(self, indent=0)
|
|
|
|
@classmethod
|
|
def from_reader(cls, reader):
|
|
# This assumes that .read_* calls are done in the order they appear
|
|
messages = []
|
|
for _ in range(reader.read_int()):
|
|
msg_id = reader.read_long()
|
|
seq_no = reader.read_int()
|
|
length = reader.read_int()
|
|
before = reader.tell_position()
|
|
obj = reader.tgread_object()
|
|
messages.append(TLMessage(msg_id, seq_no, obj))
|
|
if reader.tell_position() != before + length:
|
|
reader.set_position(before)
|
|
__log__.warning('Data left after TLObject {}: {!r}'
|
|
.format(obj, reader.read(length)))
|
|
return MessageContainer(messages)
|