From a4fa652d35a3c8f5990137ee35dbd387900d0ac0 Mon Sep 17 00:00:00 2001
From: Lonami Exo <totufals@hotmail.com>
Date: Fri, 29 Sep 2017 13:03:35 +0200
Subject: [PATCH] Precompute the flag index on the generated code

---
 telethon_generator/tl_generator.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/telethon_generator/tl_generator.py b/telethon_generator/tl_generator.py
index 84598999..76de12e5 100644
--- a/telethon_generator/tl_generator.py
+++ b/telethon_generator/tl_generator.py
@@ -472,8 +472,8 @@ class TLGenerator:
         elif arg.flag_indicator:
             # Calculate the flags with those items which are not None
             builder.write("struct.pack('<I', {})".format(
-                ' | '.join('(1 << {} if {} else 0)'.format(
-                    flag.flag_index, 'self.{}'.format(flag.name)
+                ' | '.join('({} if {} else 0)'.format(
+                    1 << flag.flag_index, 'self.{}'.format(flag.name)
                 ) for flag in args if flag.is_flag)
             ))
 
@@ -550,8 +550,8 @@ class TLGenerator:
         was_flag = False
         if arg.is_flag:
             was_flag = True
-            builder.writeln('if (flags & (1 << {})) != 0:'.format(
-                arg.flag_index
+            builder.writeln('if flags & {}:'.format(
+                1 << arg.flag_index
             ))
             # Temporary disable .is_flag not to enter this if
             # again when calling the method recursively