mir/pyrogram
2
0
Fork 0
mirror of https://github.com/pyrogram/pyrogram synced 2025-08-28 12:57:52 +00:00
Code Issues Releases Wiki Activity

Strip None fields away

Browse Source
This commit is contained in:
Dan 2018-03-26 15:16:09 +02:00
parent 3441db17b0
commit 676f9feba8
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pyrogram/api/core/object.py
View File

@ -47,6 +47,15 @@ class Object:
pass
def remove_none(obj):
if isinstance(obj, (list, tuple, set)):
return type(obj)(remove_none(x) for x in obj if x is not None)
elif isinstance(obj, dict):
return type(obj)((remove_none(k), remove_none(v)) for k, v in obj.items() if k is not None and v is not None)
else:
return obj
class Encoder(JSONEncoder):
def default(self, o: Object):
try:
@ -57,7 +66,10 @@ class Encoder(JSONEncoder):
else:
return repr(o)
return OrderedDict(
[("_", objects.get(getattr(o, "ID", None), None))]
+ [i for i in content.items()]
)
if "pyrogram" in objects.get(getattr(o, "ID", "")):
return remove_none(OrderedDict([i for i in content.items()]))
else:
return OrderedDict(
[("_", objects.get(getattr(o, "ID", None), None))]
+ [i for i in content.items()]
)