Implement missing .to_dict() and .stringify() on message/container

This commit is contained in:
Lonami Exo
2017-10-29 20:13:00 +01:00
parent 6567f4b567
commit 05626c8274
2 changed files with 27 additions and 6 deletions

View File

@@ -11,15 +11,20 @@ class MessageContainer(TLObject):
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 'MessageContainer(messages=[{}])'\
.format(', '.join(str(m) for m in self.messages))
@staticmethod
def iter_read(reader):
reader.read_int(signed=False) # code
@@ -29,3 +34,9 @@ class MessageContainer(TLObject):
inner_sequence = reader.read_int()
inner_length = reader.read_int()
yield inner_msg_id, inner_sequence, inner_length
def __str__(self):
return TLObject.pretty_format(self)
def stringify(self):
return TLObject.pretty_format(self, indent=0)