LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-22 17:36:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support chat_id autocast

Browse Source
This commit is contained in:
Lonami Exo 2018-07-07 12:53:46 +02:00
parent cb3846cb7f
commit 066004acd1
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
telethon_generator/generators/tlobject.py
View File

@ -29,6 +29,10 @@ AUTO_CASTS = {
'InputMessage': 'utils.get_input_message({})' 'InputMessage': 'utils.get_input_message({})'
} }
NAMED_AUTO_CASTS = {
('chat_id', 'int'): 'await client.get_peer_id({})'
}
BASE_TYPES = ('string', 'bytes', 'int', 'long', 'int128', BASE_TYPES = ('string', 'bytes', 'int', 'long', 'int128',
'int256', 'double', 'Bool', 'true', 'date') 'int256', 'double', 'Bool', 'true', 'date')
@ -233,13 +237,18 @@ def _write_class_init(tlobject, kind, type_constructors, builder):
def _write_resolve(tlobject, builder): def _write_resolve(tlobject, builder):
if tlobject.is_function and\ if tlobject.is_function and any(
any(arg.type in AUTO_CASTS for arg in tlobject.real_args): (arg.type in AUTO_CASTS
or (arg.name, arg.type in NAMED_AUTO_CASTS))
for arg in tlobject.real_args
):
builder.writeln('async def resolve(self, client, utils):') builder.writeln('async def resolve(self, client, utils):')
for arg in tlobject.real_args: for arg in tlobject.real_args:
ac = AUTO_CASTS.get(arg.type, None) ac = AUTO_CASTS.get(arg.type)
if not ac: if not ac:
continue ac = NAMED_AUTO_CASTS.get((arg.name, arg.type))
if not ac:
continue
if arg.is_flag: if arg.is_flag:
builder.writeln('if self.{}:', arg.name) builder.writeln('if self.{}:', arg.name)