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

Added ability to encode/decode TLObjects, updated README

Browse Source
This commit is contained in:
Lonami 2016-09-24 21:17:41 +02:00
parent 87969745f5
commit fddb3e9aac
2 changed files with 44 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -166,7 +166,6 @@ please stick with the suggested name or give one which is still descriptive enou
### Updating the `scheme.tl`
Have you found a more updated version of the `scheme.tl` file? Those are great news! Updating is as simple as grabbing the
[latest version](https://github.com/telegramdesktop/tdesktop/blob/master/Telegram/SourceFiles/mtproto/scheme.tl) and
replacing the one you can find in this same directory by the updated one. Don't forget to run `python3 tl_generator.py`
afterwards and specifying the new layer number to be used when creating the `TelegramClient`.
replacing the one you can find in this same directory by the updated one. Don't forget to run `python3 tl_generator.py`.
If the changes weren't too big, everything should still work the same way as it did before; but with extra features.

43
telethon_generator/tl_generator.py
View File

@ -62,6 +62,7 @@ class TLGenerator:
with SourceBuilder(file) as builder:
# Both types and functions inherit from MTProtoRequest so they all can be sent
builder.writeln('from telethon.tl.mtproto_request import MTProtoRequest')
builder.writeln('import json')
builder.writeln()
builder.writeln()
builder.writeln('class {}(MTProtoRequest):'.format(TLGenerator.get_class_name(tlobject)))
@ -153,6 +154,42 @@ class TLGenerator:
builder.writeln('def __str__(self):')
builder.writeln("return {}".format(str(tlobject)))
builder.end_block()
# Write JSON encoding
builder.writeln('def json_encode(self):')
builder.writeln('return json.dumps({')
builder.current_indent += 1
for arg in args:
if TLGenerator.is_tlobject(arg.type):
builder.writeln("'{0}': self.{0}.json_encode(),".format(arg.name))
else:
builder.writeln("'{0}': self.{0},".format(arg.name))
builder.current_indent -= 1
builder.writeln('})')
builder.end_block()
# Write JSON decoding
builder.writeln('@staticmethod')
builder.writeln('def json_decode(json_string):')
builder.writeln('# Create an empty instance which will be filled with the JSON values')
builder.writeln('instance = {}({})'.format(
TLGenerator.get_class_name(tlobject),
', '.join('None' for _ in range(len(args)))
))
builder.writeln('dictionary = json.loads(json_string)')
builder.writeln()
for arg in args:
if TLGenerator.is_tlobject(arg.type):
builder.writeln("instance.{0} = dictionary['{0}'].json_decode() if '{0}' in dictionary "
"and dictionary['{0}'] is not None else None".format(arg.name))
else:
builder.writeln("instance.{0} = dictionary.get('{0}', None)".format(arg.name))
builder.writeln()
builder.writeln('return instance')
# builder.end_block() # There is no need to end the last block
# Once all the objects have been generated, we can now group them in a single file
@ -223,6 +260,12 @@ class TLGenerator:
else:
return result
@staticmethod
def is_tlobject(type):
"""Determines whether the type is a "basic" type or another TLObject"""
return type not in ['int', 'long', 'int128', 'int256', 'double',
'string', 'Bool', 'true', 'bytes', 'date']
@staticmethod
def write_onsend_code(builder, arg, args, name=None):
"""