Move all imported schema objects outside TLObject

delivrance · delivrance · commit 9ad5e62dea5f · 2019-08-01T18:44:20.000+02:00
diff --git a/pyrogram/api/__init__.py b/pyrogram/api/__init__.py
@@ -19,8 +19,7 @@
 from importlib import import_module
 
 from .all import objects
-from .core.tl_object import TLObject
 
 for k, v in objects.items():
     path, name = v.rsplit(".", 1)
-    TLObject.all[k] = getattr(import_module(path), name)
+    objects[k] = getattr(import_module(path), name)
diff --git a/pyrogram/api/core/__init__.py b/pyrogram/api/core/__init__.py
@@ -22,8 +22,5 @@
 from .list import List
 from .message import Message
 from .msg_container import MsgContainer
-from .primitives import (
-    Bool, BoolTrue, BoolFalse, Bytes, Double,
-    Int, Long, Int128, Int256, Null, String, Vector
-)
+from .primitives import *
 from .tl_object import TLObject
diff --git a/pyrogram/api/core/tl_object.py b/pyrogram/api/core/tl_object.py
@@ -20,17 +20,17 @@
 from io import BytesIO
 from json import dumps
 
+from ..all import objects
 
-class TLObject:
-    all = {}
 
+class TLObject:
     __slots__ = []
 
     QUALNAME = "Base"
 
     @staticmethod
     def read(b: BytesIO, *args):  # TODO: Rename b -> data
-        return TLObject.all[int.from_bytes(b.read(4), "little")].read(b, *args)
+        return objects[int.from_bytes(b.read(4), "little")].read(b, *args)
 
     def write(self, *args) -> bytes:
         pass
