Use __name__ instead of .split(".")[-1]

pyrogram · May 1, 2018 · a94cf96 · a94cf96
1 parent f29b8bb
commit a94cf96
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyrogram/api/core/object.py b/pyrogram/api/core/object.py
@@ -72,11 +72,12 @@ def default(self, o: Object):
             else:
                 return repr(o)
 
+        name = o.__class__.__name__
         o = objects.get(getattr(o, "ID", None), None)
 
         if o is not None:
             if o.startswith("pyrogram.client"):
-                r = remove_none(OrderedDict([("_", o.split(".")[-1])] + [i for i in content.items()]))
+                r = remove_none(OrderedDict([("_", name)] + [i for i in content.items()]))
                 r.pop("client", None)
 
                 return r