Detect current MTProtoLayer automatically

telethon_generator/parser/tl_parser.py

@@ -29,3 +29,13 @@ class TLParser:
 
                     else:
                         yield TLObject.from_tl(line, is_function)
+
+    @staticmethod
+    def find_layer(file_path):
+        """Finds the layer used on the specified scheme.tl file"""
+        layer_regex = re.compile(r'^//\s*LAYER\s*(\d+)$')
+        with open(file_path, encoding='utf-8') as file:
+            for line in file:
+                match = layer_regex.match(line)
+                if match:
+                    return int(match.group(1))

telethon_generator/tl_generator.py

@@ -167,6 +167,10 @@ class TLGenerator:
                     builder.writeln('import {}'.format(TLGenerator.get_full_file_name(tlobject)))
                 builder.writeln()
 
+                # Create a variable to indicate which layer this is
+                builder.writeln('layer = {}  # Current generated layer'.format(TLParser.find_layer(scheme_file)))
+                builder.writeln()
+
                 # Then create the dictionary containing constructor_id: class
                 builder.writeln('tlobjects = {')
                 builder.current_indent += 1