Implement non-blocking TCP connection (

RknDeveloper · web-flow · commit 7b4101c999ac · 2024-06-26T10:25:33.000+05:45
diff --git a/pyrogram/client.py b/pyrogram/client.py
@@ -31,7 +31,7 @@
 from io import StringIO, BytesIO
 from mimetypes import MimeTypes
 from pathlib import Path
-from typing import Union, List, Optional, Callable, AsyncGenerator
+from typing import Union, List, Optional, Callable, AsyncGenerator, Type
 
 import pyrogram
 from pyrogram import __version__, __license__
@@ -51,6 +51,8 @@
 from pyrogram.storage import FileStorage, MemoryStorage
 from pyrogram.types import User, TermsOfService
 from pyrogram.utils import ainput
+from .connection import Connection
+from .connection.transport import TCP, TCPAbridged
 from .dispatcher import Dispatcher
 from .file_id import FileId, FileType, ThumbnailSource
 from .mime_types import mime_types
@@ -251,6 +253,9 @@ def __init__(
         else:
             self.storage = FileStorage(self.name, self.workdir)
 
+        self.connection_factory = Connection
+        self.protocol_factory = TCPAbridged
+        
         self.dispatcher = Dispatcher(self)
 
         self.rnd_id = MsgId
