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

Automatically call .get_input_peer on the requests that need it

Browse Source
This commit is contained in:
Lonami Exo 2017-06-15 16:35:40 +02:00
parent c02fbae5aa
commit b8e46446ba
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
telethon_generator/tl_generator.py
View File

@ -186,6 +186,12 @@ class TLGenerator:
builder.writeln('from {}.tl.mtproto_request import MTProtoRequest'
.format('.' * depth))
if any(a for a in tlobject.args if a.type == 'InputPeer'):
# We can automatically convert a normal peer to an InputPeer,
# it will make invoking a lot of requests a lot simpler.
builder.writeln('from {}.utils import get_input_peer'
.format('.' * depth))
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')
@ -306,6 +312,10 @@ class TLGenerator:
)
else:
raise ValueError('Cannot infer a value for ', arg)
elif arg.type == 'InputPeer':
# Well-known case, auto-cast it to the right type
builder.writeln(
'self.{0} = get_input_peer({0})'.format(arg.name))
else:
builder.writeln('self.{0} = {0}'.format(arg.name))