LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-25 19:03:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make .to_dict() work on invalid TLObjects

Browse Source 
This will improve logging from the previous commit as well.
This commit is contained in:
Lonami Exo 2018-08-08 12:12:00 +02:00
parent d3efc8f57d
commit f25541ea6f
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
telethon_generator/generators/tlobject.py
View File

@ -54,7 +54,11 @@ def _write_modules(
SourceBuilder(f) as builder: SourceBuilder(f) as builder:
builder.writeln(AUTO_GEN_NOTICE) builder.writeln(AUTO_GEN_NOTICE)
builder.writeln('from {}.tl.tlobject import {}', '.' * depth, kind) builder.writeln('from {}.tl.tlobject import TLObject', '.' * depth)
if kind != 'TLObject':
builder.writeln(
'from {}.tl.tlobject import {}', '.' * depth, kind)
builder.writeln('from typing import Optional, List, ' builder.writeln('from typing import Optional, List, '
'Union, TYPE_CHECKING') 'Union, TYPE_CHECKING')
@ -296,13 +300,14 @@ def _write_to_dict(tlobject, builder):
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 [x.to_dict() '
'if x is None else x.to_dict() for x in self.{0}]', 'if isinstance(x, TLObject) else x for x in self.{0}]',
arg.name arg.name
) )
else: else:
builder.write( builder.write(
'None if self.{0} is None else self.{0}.to_dict()', 'self.{0}.to_dict() '
'if isinstance(self.{0}, TLObject) else self.{0}',
arg.name arg.name
) )