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

Assert flag params with same flag index are all set/unset

Browse Source
This commit is contained in:
Lonami Exo 2017-09-30 19:47:19 +02:00
parent a5ab49c707
commit ce48c9752e
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
telethon_generator/tl_generator.py
View File

@ -301,6 +301,26 @@ class TLGenerator:
# Write the .to_bytes() function
builder.writeln('def to_bytes(self):')
# Some objects require more than one flag parameter to be set
# at the same time. In this case, add an assertion.
repeated_args = defaultdict(list)
for arg in tlobject.args:
if arg.is_flag:
repeated_args[arg.flag_index].append(arg)
for args in repeated_args.values():
if len(args) > 1:
cnd1 = ('self.{} is None'.format(a.name) for a in args)
cnd2 = ('self.{} is not None'.format(a.name) for a in args)
builder.writeln(
"assert ({}) or ({}), '{} parameters must all "
"be None or neither be None'".format(
' and '.join(cnd1), ' and '.join(cnd2),
', '.join(a.name for a in args)
)
)
builder.writeln("return b''.join((")
builder.current_indent += 1