hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-06-17 18:46:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Don't treat False field as flag omission

Browse Source 
8724949b54 was only half the story.
This commit is contained in:
Lonami Exo 2021-03-16 08:21:04 +01:00
parent 58013f4f44
commit b06f496a27
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon_generator/generators/tlobject.py
View File

@ -459,12 +459,16 @@ def _write_arg_to_bytes(builder, arg, tlobject, name=None):
# There's a flag indicator, but no flag arguments so it's 0 # There's a flag indicator, but no flag arguments so it's 0
builder.write(r"b'\0\0\0\0'") builder.write(r"b'\0\0\0\0'")
else: else:
def fmt_flag(flag):
if flag.type == 'Bool':
fmt = '(0 if {0} is None else {1})'
else:
fmt = '(0 if {0} is None or {0} is False else {1})'
return fmt.format('self.{}'.format(flag.name), 1 << flag.flag_index)
builder.write("struct.pack('<I', ") builder.write("struct.pack('<I', ")
builder.write( builder.write(
' | '.join('(0 if {0} is None or {0} is False else {1})' ' | '.join(fmt_flag(flag) for flag in tlobject.args if flag.is_flag)
.format('self.{}'.format(flag.name),
1 << flag.flag_index)
for flag in tlobject.args if flag.is_flag)
) )
builder.write(')') builder.write(')')