diff --git a/telethon_generator/tl_generator.py b/telethon_generator/tl_generator.py index ecdbd5fe..6faedc8d 100644 --- a/telethon_generator/tl_generator.py +++ b/telethon_generator/tl_generator.py @@ -5,6 +5,8 @@ from zlib import crc32 from collections import defaultdict from .parser import SourceBuilder, TLParser +AUTO_GEN_NOTICE = \ + '"""File generated by TLObjects\' generator. All changes will be ERASED"""' class TLGenerator: @@ -75,8 +77,7 @@ class TLGenerator: filename = os.path.join(self._get_file('all_tlobjects.py')) with open(filename, 'w', encoding='utf-8') as file: with SourceBuilder(file) as builder: - builder.writeln( - '"""File generated by TLObjects\' generator. All changes will be ERASED"""') + builder.writeln(AUTO_GEN_NOTICE) builder.writeln() builder.writeln('from . import types, functions') @@ -119,6 +120,7 @@ class TLGenerator: file = os.path.join(out_dir, ns + '.py' if ns else '__init__.py') with open(file, 'w', encoding='utf-8') as f, \ SourceBuilder(f) as builder: + builder.writeln(AUTO_GEN_NOTICE) # Add the relative imports to the namespaces, # unless we already are in a namespace. if not ns: @@ -179,15 +181,6 @@ class TLGenerator: builder.writeln('class {}(TLObject):'.format( TLGenerator.get_class_name(tlobject))) - # Write the original .tl definition, - # along with a "generated automatically" message - builder.writeln( - '"""Class generated by TLObjects\' generator. ' - 'All changes will be ERASED. TL definition below.' - ) - builder.writeln('{}"""'.format(repr(tlobject))) - builder.writeln() - # Class-level variable to store its constructor ID builder.writeln("# Telegram's constructor (U)ID for this class") builder.writeln('constructor_id = {}'.format(hex(tlobject.id)))