yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-02 05:24:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Small fixed to PyrogramType

Browse source
This commit is contained in:
Dan 2018-12-16 17:10:08 +01:00
parent d62bc1d8ba
commit cffafa8c61
1 changed files with 4 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pyrogram/client/types/pyrogram_type.py
View file

@ -24,6 +24,9 @@ class PyrogramType:
def __str__(self):
return dumps(self, cls=Encoder, indent=4)
def __getitem__(self, item):
return getattr(self, item)
def remove_none(obj):
if isinstance(obj, (list, tuple, set)):
@ -37,8 +40,4 @@ def remove_none(obj):
class Encoder(JSONEncoder):
def default(self, o: PyrogramType):
content = {i: getattr(o, i) for i in filter(lambda x: not x.startswith("_"), o.__dict__)}
return OrderedDict(
[("_", "pyrogram:{}".format(o.__class__.__name__))]
+ [i for i in remove_none(content).items()]
)
return remove_none(OrderedDict([("_", "pyrogram:" + o.__class__.__name__)] + [i for i in content.items()]))