LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-29 12:53:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Reference argument types to objects when parsing .tl

Browse Source
This commit is contained in:
Lonami Exo 2018-10-15 19:23:09 +02:00
parent b219da6b87
commit e3c4bd46fb
1 changed files with 24 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
telethon_generator/parsers/tlobject.py
View File

@ -1,3 +1,4 @@
import collections
import re import re
import struct import struct
from zlib import crc32 from zlib import crc32
@ -127,6 +128,7 @@ class TLArg:
self.is_flag = False self.is_flag = False
self.skip_constructor_id = False self.skip_constructor_id = False
self.flag_index = -1 self.flag_index = -1
self.cls = None
# Special case: some types can be inferred, which makes it # Special case: some types can be inferred, which makes it
# less annoying to type. Currently the only type that can # less annoying to type. Currently the only type that can
@ -274,10 +276,18 @@ def _from_line(line, is_function, layer):
def parse_tl(file_path, layer, invalid_bot_methods=None): def parse_tl(file_path, layer, invalid_bot_methods=None):
"""This method yields TLObjects from a given .tl file.""" """
This method yields TLObjects from a given .tl file.
Note that the file is parsed completely before the function yields
because references to other objects may appear later in the file.
"""
if invalid_bot_methods is None: if invalid_bot_methods is None:
invalid_bot_methods = set() invalid_bot_methods = set()
obj_all = []
obj_by_name = {}
obj_by_type = collections.defaultdict(list)
with open(file_path, 'r', encoding='utf-8') as file: with open(file_path, 'r', encoding='utf-8') as file:
is_function = False is_function = False
for line in file: for line in file:
@ -298,11 +308,23 @@ def parse_tl(file_path, layer, invalid_bot_methods=None):
try: try:
result = _from_line(line, is_function, layer=layer) result = _from_line(line, is_function, layer=layer)
result.bot_usable = result.fullname not in invalid_bot_methods result.bot_usable = result.fullname not in invalid_bot_methods
yield result obj_all.append(result)
obj_by_name[result.fullname] = result
obj_by_type[result.result].append(result)
except ValueError as e: except ValueError as e:
if 'vector#1cb5c415' not in str(e): if 'vector#1cb5c415' not in str(e):
raise raise
# Once all objects have been parsed, replace the
# string type from the arguments with references
for obj in obj_all:
for arg in obj.args:
arg.cls = obj_by_type.get(arg.type) or (
[obj_by_name[arg.type]] if arg.type in obj_by_name else []
)
yield from obj_all
def find_layer(file_path): def find_layer(file_path):
"""Finds the layer used on the specified scheme.tl file.""" """Finds the layer used on the specified scheme.tl file."""