LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-02-18 04:20:57 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add support for non-recursive TLObject.to_dict()

Browse Source
This commit is contained in:
Lonami Exo 2017-09-23 10:41:36 +02:00
parent 73fbfde7ef
commit 0b48b1ec8a
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
telethon_generator/tl_generator.py
View File

@ -265,7 +265,7 @@ class TLGenerator:
# Write the to_dict(self) method # Write the to_dict(self) method
if args: if args:
builder.writeln('def to_dict(self):') builder.writeln('def to_dict(self, recursive=True):')
builder.writeln('return {') builder.writeln('return {')
builder.current_indent += 1 builder.current_indent += 1
@ -285,13 +285,15 @@ class TLGenerator:
else: else:
if arg.is_vector: if arg.is_vector:
builder.write( builder.write(
'[] if self.{0} is None else [None ' '([] if self.{0} is None else [None '
'if x is None else x.to_dict() for x in self.{0}]' 'if x is None else x.to_dict() for x in self.{0}]'
') if recursive else self.{0}'
.format(arg.name) .format(arg.name)
) )
else: else:
builder.write( builder.write(
'None if self.{0} is None else self.{0}.to_dict()' '(None if self.{0} is None else '
'self.{0}.to_dict()) if recursive else self.{0}'
.format(arg.name) .format(arg.name)
) )
builder.writeln(',') builder.writeln(',')
@ -299,8 +301,7 @@ class TLGenerator:
builder.current_indent -= 1 builder.current_indent -= 1
builder.writeln("}") builder.writeln("}")
else: else:
builder.writeln('@staticmethod') builder.writeln('def to_dict(self, recursive=True):')
builder.writeln('def to_dict():')
builder.writeln('return {}') builder.writeln('return {}')
builder.end_block() builder.end_block()