Faster Object deserialization (~3x)

delivrance · delivrance · commit 7374c10afd1f · 2017-12-11T11:35:38.000+01:00
diff --git a/pyrogram/api/__init__.py b/pyrogram/api/__init__.py
@@ -15,3 +15,12 @@
 #
 # You should have received a copy of the GNU Lesser General Public License
 # along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
+
+from importlib import import_module
+
+from .all import objects
+from .core.object import Object
+
+for k, v in objects.items():
+    path, name = v.rsplit(".", 1)
+    Object.all[k] = getattr(import_module("pyrogram.api." + path), name)
diff --git a/pyrogram/api/core/object.py b/pyrogram/api/core/object.py
@@ -18,24 +18,18 @@
 
 from collections import OrderedDict
 from datetime import datetime
-from importlib import import_module
 from io import BytesIO
 from json import JSONEncoder, dumps
 
 from ..all import objects
 
 
 class Object:
+    all = {}
+
     @staticmethod
     def read(b: BytesIO, *args):
-        id = int.from_bytes(b.read(4), "little")
-        name = objects.get(id)
-        path, name = name.rsplit(".", 1)
-
-        return getattr(
-            import_module("pyrogram.api." + path),
-            name
-        ).read(b, *args)
+        return Object.all[int.from_bytes(b.read(4), "little")].read(b, *args)
 
     def write(self, *args) -> bytes:
         pass
