TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-18 13:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make TCPAbridged async

Browse Source
This commit is contained in:
Dan 2018-10-08 20:16:04 +02:00
parent a28ab0e8a8
commit ee06907bda
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
pyrogram/connection/transport/tcp/tcp_abridged.py
View File

@ -27,30 +27,30 @@ class TCPAbridged(TCP):
def __init__(self, ipv6: bool, proxy: dict):
super().__init__(ipv6, proxy)
def connect(self, address: tuple):
super().connect(address)
super().sendall(b"\xef")
async def connect(self, address: tuple):
await super().connect(address)
await super().send(b"\xef")
def sendall(self, data: bytes, *args):
async def send(self, data: bytes, *args):
length = len(data) // 4
super().sendall(
await super().send(
(bytes([length])
if length <= 126
else b"\x7f" + length.to_bytes(3, "little"))
+ data
)
def recvall(self, length: int = 0) -> bytes or None:
length = super().recvall(1)
async def recv(self, length: int = 0) -> bytes or None:
length = await super().recv(1)
if length is None:
return None
if length == b"\x7f":
length = super().recvall(3)
length = await super().recv(3)
if length is None:
return None
return super().recvall(int.from_bytes(length, "little") * 4)
return await super().recv(int.from_bytes(length, "little") * 4)