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

Implement automatic cast to InputUser too (closes #159)

Browse Source
This commit is contained in:
Lonami Exo
2017-07-10 16:04:10 +02:00
parent 84bb3bb325
commit bdee94eaf3
2 changed files with 40 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
telethon_generator/tl_generator.py
View File

@@ -184,16 +184,20 @@ class TLGenerator:
.format('.' * depth))
if tlobject.is_function:
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))
util_imports = set()
for a in tlobject.args:
# We can automatically convert some "full" types to
# "input only" (like User -> InputPeerUser, etc.)
if a.type == 'InputPeer':
util_imports.add('get_input_peer')
elif a.type == 'InputChannel':
util_imports.add('get_input_channel')
elif a.type == 'InputUser':
util_imports.add('get_input_user')
if any(a for a in tlobject.args if a.type == 'InputChannel'):
# Same applies to channels
builder.writeln('from {}.utils import get_input_channel'
.format('.' * depth))
if util_imports:
builder.writeln('from {}.utils import {}'.format(
'.' * 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
@@ -327,6 +331,10 @@ class TLGenerator:
builder.writeln(
'self.{0} = get_input_channel({0})'.format(arg.name)
)
elif arg.type == 'InputUser' and tlobject.is_function:
builder.writeln(
'self.{0} = get_input_user({0})'.format(arg.name)
)
else:
builder.writeln('self.{0} = {0}'.format(arg.name))