Refactor tcp_abridged

delivrance · delivrance · commit 64fe9163d25f · 2018-05-15T19:17:22.000+02:00
diff --git a/pyrogram/connection/transport/tcp/tcp_abridged.py b/pyrogram/connection/transport/tcp/tcp_abridged.py
@@ -26,28 +26,23 @@
 class TCPAbridged(TCP):
     def __init__(self, proxy: dict):
         super().__init__(proxy)
-        self.is_first_packet = None
 
     def connect(self, address: tuple):
         super().connect(address)
-        self.is_first_packet = True
+        super().sendall(b"\xef")
+
         log.info("Connected{}!".format(" with proxy" if self.proxy_enabled else ""))
 
     def sendall(self, data: bytes, *args):
         length = len(data) // 4
 
-        data = (
-            bytes([length]) + data
-            if length <= 126
-            else b"\x7f" + int.to_bytes(length, 3, "little") + data
+        super().sendall(
+            (bytes([length])
+             if length <= 126
+             else b"\x7f" + length.to_bytes(3, "little"))
+            + data
         )
 
-        if self.is_first_packet:
-            data = b"\xef" + data
-            self.is_first_packet = False
-
-        super().sendall(data)
-
     def recvall(self, length: int = 0) -> bytes or None:
         length = super().recvall(1)
 
