Fix the rest of the typing hints (#232)

* ignore kad

* fix swarm, and minor

* fix init and swarm

* ignore pb

* enable mypy

* fix basic host

* fix tcp

* fix mplex

* add typing for pb

* skip format pyi

* [mypy] no need to ignore pb now

* add typing to chat

libp2p/transport/tcp/tcp.py

@@ -15,9 +15,8 @@ from libp2p.transport.typing import THandler
 class TCPListener(IListener):
     multiaddrs: List[Multiaddr]
     server = None
-    handler = None
 
-    def __init__(self, handler_function: THandler = None) -> None:
+    def __init__(self, handler_function: THandler) -> None:
         self.multiaddrs = []
         self.server = None
         self.handler = handler_function

libp2p/transport/typing.py

@@ -1,4 +1,4 @@
 from asyncio import StreamReader, StreamWriter
-from typing import Callable
+from typing import Awaitable, Callable
 
-THandler = Callable[[StreamReader, StreamWriter], None]
+THandler = Callable[[StreamReader, StreamWriter], Awaitable[None]]

libp2p/transport/upgrader.py

@@ -1,4 +1,4 @@
-from typing import Dict, Sequence
+from typing import Mapping, Sequence
 
 from libp2p.network.connection.raw_connection_interface import IRawConnection
 from libp2p.network.typing import GenericProtocolHandlerFn
@@ -17,7 +17,9 @@ class TransportUpgrader:
     security_multistream: SecurityMultistream
     muxer: Sequence[str]
 
-    def __init__(self, secOpt: Dict[TProtocol, ISecureTransport], muxerOpt: Sequence[str]) -> None:
+    def __init__(
+        self, secOpt: Mapping[TProtocol, ISecureTransport], muxerOpt: Sequence[str]
+    ) -> None:
         # Store security option
         self.security_multistream = SecurityMultistream()
         for key in secOpt:
@@ -45,7 +47,7 @@ class TransportUpgrader:
 
     @staticmethod
     def upgrade_connection(
-        conn: IRawConnection, generic_protocol_handler: GenericProtocolHandlerFn, peer_id: ID
+        conn: ISecureConn, generic_protocol_handler: GenericProtocolHandlerFn, peer_id: ID
     ) -> Mplex:
         """
         Upgrade raw connection to muxed connection