hoelee/Telethon
1
0
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2025-08-10 10:49:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Implemented init and write code on TLObjects Generator

Browse Source 
The code generated by the generator now classifies the output files
in their corresponding categories, also writing their __init__(...)
with documented arguments, and the on_send(...) method
This commit is contained in:
Lonami
2016-08-27 11:59:23 +02:00
parent 1974569927
commit 06832f8108
8 changed files with 310 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
tl/tlobject.py
View File

@@ -96,7 +96,8 @@ class TLArg:
"""
self.name = name
# Default flag values
# Default values
self.is_vector = False
self.is_flag = False
self.flag_index = -1
@@ -119,15 +120,27 @@ class TLArg:
self.flag_index = int(flag_match.group(1))
self.type = flag_match.group(2) # Update the type to match the exact type, not the "flagged" one
# Then check if the type is a Vector<REAL_TYPE>
vector_match = re.match(r'vector<(\w+)>', self.type, re.IGNORECASE)
if vector_match:
self.is_vector = True
self.type = vector_match.group(1) # Update the type to match the one inside the vector
self.generic_definition = generic_definition
def __str__(self):
type = ('!{}'.format(self.type) if self.is_generic
else
('flags.{}?{}'.format(self.flag_index, self.type) if self.is_flag
else self.type))
# Find the real type representation by updating it as required
real_type = self.type
if self.is_vector:
real_type = 'Vector<{}>'.format(real_type)
if self.is_generic:
real_type = '!{}'.format(real_type)
if self.is_flag:
real_type = 'flags.{}?{}'.format(self.flag_index, real_type)
if self.generic_definition:
return '{{{}:{}}}'.format(self.name, type)
return '{{{}:{}}}'.format(self.name, real_type)
else:
return '{}:{}'.format(self.name, type)
return '{}:{}'.format(self.name, real_type)