LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-23 01:46:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Import os only once on the generated code

Browse Source
This commit is contained in:
Lonami Exo 2017-09-18 20:23:03 +02:00
parent 2c5c674531
commit 2595d45bd7
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
telethon_generator/tl_generator.py
View File

@ -137,13 +137,17 @@ class TLGenerator:
x for x in namespace_tlobjects.keys() if x x for x in namespace_tlobjects.keys() if x
))) )))
# Import 'os' for those needing access to 'os.urandom()'
# Currently only 'random_id' needs 'os' to be imported,
# for all those TLObjects with arg.can_be_inferred.
builder.writeln('import os')
# Generate the class for every TLObject # Generate the class for every TLObject
for t in sorted(tlobjects, key=lambda x: x.name): for t in sorted(tlobjects, key=lambda x: x.name):
TLGenerator._write_source_code( TLGenerator._write_source_code(
t, builder, depth, type_constructors t, builder, depth, type_constructors
) )
while builder.current_indent != 0: builder.current_indent = 0
builder.end_block()
@staticmethod @staticmethod
def _write_source_code(tlobject, builder, depth, type_constructors): def _write_source_code(tlobject, builder, depth, type_constructors):
@ -170,10 +174,6 @@ class TLGenerator:
builder.writeln('from {}.utils import {}'.format( builder.writeln('from {}.utils import {}'.format(
'.' * depth, ', '.join(util_imports))) '.' * depth, ', '.join(util_imports)))
if any(a for a in tlobject.args if a.can_be_inferred):
# Currently only 'random_id' needs 'os' to be imported
builder.writeln('import os')
builder.writeln() builder.writeln()
builder.writeln() builder.writeln()
builder.writeln('class {}(TLObject):'.format( builder.writeln('class {}(TLObject):'.format(