feat: improve typehint of proxy param in Client class

Author: KurimuzonAkuma

pyrogram/client.py

@@ -37,20 +37,19 @@
 
 import pyrogram
 from pyrogram import __license__, __version__, enums, raw, utils
+from pyrogram.connection.transport.tcp import ProxyDict
 from pyrogram.crypto import aes
 from pyrogram.errors import (
     AuthBytesInvalid,
+    AuthTokenExpired,
     BadRequest,
     CDNFileHashMismatch,
     ChannelPrivate,
-    FloodPremiumWait,
-    FloodWait,
     PersistentTimestampInvalid,
     PersistentTimestampOutdated,
     SessionPasswordNeeded,
     Unauthorized,
     VolumeLocNotFound,
-    AuthTokenExpired
 )
 from pyrogram.handlers.handler import Handler
 from pyrogram.methods import Methods
@@ -117,8 +116,8 @@ class Client(Methods):
             after which the server address will be updated (works both ways).
             Defaults to False (IPv4).
 
-        proxy (``dict`` | ``str``, *optional*):
-            The Proxy settings as dict.
+        proxy (``str`` | ``dict``, *optional*):
+            The Proxy settings as url or dict.
             E.g.: *dict(scheme="socks5", hostname="11.22.33.44", port=1234, username="user", password="pass")*
             or *"http://11.22.33.44:1234"* or *"socks5://user:pass@11.22.33.44:1234"* or *"tg://user:pass@11.22.33.44:1234"*.
             The *username* and *password* can be omitted if the proxy doesn't require authorization.
@@ -288,7 +287,7 @@ def __init__(
         lang_code: str = LANG_CODE,
         system_lang_code: str = SYSTEM_LANG_CODE,
         ipv6: Optional[bool] = False,
-        proxy: Optional[Union[dict, str]] = None,
+        proxy: Optional[Union[str, ProxyDict]] = None,
         test_mode: Optional[bool] = False,
         bot_token: Optional[str] = None,
         session_string: Optional[str] = None,

pyrogram/connection/transport/tcp/__init__.py

@@ -16,7 +16,7 @@
 #  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 .tcp import TCP, Proxy
+from .tcp import TCP, Proxy, ProxyDict
 from .tcp_abridged import TCPAbridged
 from .tcp_abridged_o import TCPAbridgedO
 from .tcp_full import TCPFull