Fix some type annotations

Author: romis2012

python_socks/async_/anyio/_connect.py

@@ -1,11 +1,12 @@
+from typing import Optional
 import anyio
 import anyio.abc
 
 
 async def connect_tcp(
     host: str,
     port: int,
-    local_host: str = None,
+    local_host: Optional[str] = None,
 ) -> anyio.abc.SocketStream:
 
     return await anyio.connect_tcp(

python_socks/async_/anyio/_proxy.py

@@ -130,6 +130,6 @@ def create(cls, *args, **kwargs):  # for backward compatibility
         return cls(*args, **kwargs)
 
     @classmethod
-    def from_url(cls, url: str, **kwargs):
+    def from_url(cls, url: str, **kwargs) -> 'AnyioProxy':
         url_args = parse_proxy_url(url)
         return cls(*url_args, **kwargs)

python_socks/async_/anyio/v2/_connect.py

@@ -1,3 +1,4 @@
+from typing import Optional
 import anyio
 import anyio.abc
 from ._stream import AnyioSocketStream
@@ -6,7 +7,7 @@
 async def connect_tcp(
     host: str,
     port: int,
-    local_host: str = None,
+    local_host: Optional[str] = None,
 ) -> AnyioSocketStream:
     s = await anyio.connect_tcp(
         remote_host=host,

python_socks/async_/anyio/v2/_proxy.py

@@ -59,9 +59,7 @@ async def connect(
                     dest_ssl=dest_ssl,
                 )
         except TimeoutError as e:
-            raise ProxyTimeoutError(
-                'Proxy connection timed out: {}'.format(timeout)
-            ) from e
+            raise ProxyTimeoutError('Proxy connection timed out: {}'.format(timeout)) from e
 
     async def _connect(
         self,
@@ -127,6 +125,6 @@ def create(cls, *args, **kwargs):  # for backward compatibility
         return cls(*args, **kwargs)
 
     @classmethod
-    def from_url(cls, url: str, **kwargs):
+    def from_url(cls, url: str, **kwargs) -> 'AnyioProxy':
         url_args = parse_proxy_url(url)
         return cls(*url_args, **kwargs)

python_socks/async_/asyncio/_connect.py

@@ -23,7 +23,7 @@ async def connect_tcp(
     if is_ipv6_address(host):
         address = (host, port, 0, 0)  # to fix OSError: [WinError 10022]
     else:
-        address = (host, port)
+        address = (host, port)  # type: ignore[assignment]
 
     await loop.sock_connect(sock=sock, address=address)
     return sock

python_socks/async_/asyncio/_proxy.py

@@ -129,6 +129,6 @@ def create(cls, *args, **kwargs):  # for backward compatibility
         return cls(*args, **kwargs)
 
     @classmethod
-    def from_url(cls, url: str, **kwargs):
+    def from_url(cls, url: str, **kwargs) -> 'AsyncioProxy':
         url_args = parse_proxy_url(url)
         return cls(*url_args, **kwargs)

python_socks/async_/asyncio/v2/_connect.py

@@ -16,7 +16,7 @@ async def connect_tcp(
     reader, writer = await asyncio.open_connection(
         host=host,
         port=port,
-        **kwargs,
+        **kwargs,  # type: ignore
     )
 
     return AsyncioSocketStream(

python_socks/async_/asyncio/v2/_proxy.py

@@ -80,9 +80,7 @@ async def connect(
                     dest_ssl=dest_ssl,
                 )
         except asyncio.TimeoutError as e:
-            raise ProxyTimeoutError(
-                'Proxy connection timed out: {}'.format(timeout)
-            ) from e
+            raise ProxyTimeoutError('Proxy connection timed out: {}'.format(timeout)) from e
 
     async def _connect(
         self,
@@ -149,6 +147,6 @@ def create(cls, *args, **kwargs):  # for backward compatibility
         return cls(*args, **kwargs)
 
     @classmethod
-    def from_url(cls, url: str, **kwargs):
+    def from_url(cls, url: str, **kwargs) -> 'AsyncioProxy':
         url_args = parse_proxy_url(url)
         return cls(*url_args, **kwargs)

python_socks/async_/curio/_proxy.py

@@ -123,6 +123,6 @@ def create(cls, *args, **kwargs):  # for backward compatibility
         return cls(*args, **kwargs)
 
     @classmethod
-    def from_url(cls, url: str, **kwargs):
+    def from_url(cls, url: str, **kwargs) -> 'CurioProxy':
         url_args = parse_proxy_url(url)
         return cls(*url_args, **kwargs)

python_socks/async_/trio/_proxy.py

@@ -122,6 +122,6 @@ def create(cls, *args, **kwargs):  # for backward compatibility
         return cls(*args, **kwargs)
 
     @classmethod
-    def from_url(cls, url: str, **kwargs):
+    def from_url(cls, url: str, **kwargs) -> 'TrioProxy':
         url_args = parse_proxy_url(url)
         return cls(*url_args, **kwargs)