Add type hints to peer folder

libp2p/peer/peerdata.py

@@ -1,35 +1,48 @@
+from typing import (
+    Any,
+    Dict,
+    List,
+    Sequence,
+)
+
+from multiaddr import Multiaddr
+
 from .peerdata_interface import IPeerData
 
 
 class PeerData(IPeerData):
 
-    def __init__(self):
+    metadata: Dict[Any, Any]
+    protocols: List[str]
+    addrs: List[Multiaddr]
+
+    def __init__(self) -> None:
         self.metadata = {}
         self.protocols = []
         self.addrs = []
 
-    def get_protocols(self):
+    def get_protocols(self) -> List[str]:
         return self.protocols
 
-    def add_protocols(self, protocols):
-        self.protocols.extend(protocols)
+    def add_protocols(self, protocols: Sequence[str]) -> None:
+        self.protocols.extend(list(protocols))
 
-    def set_protocols(self, protocols):
-        self.protocols = protocols
+    def set_protocols(self, protocols: Sequence[str]) -> None:
+        self.protocols = list(protocols)
 
-    def add_addrs(self, addrs):
-        self.addrs.extend(addrs)
+    def add_addrs(self, addrs: Sequence[Multiaddr]) -> None:
+        self.addrs.extend(list(addrs))
 
-    def get_addrs(self):
+    def get_addrs(self) -> List[Multiaddr]:
         return self.addrs
 
-    def clear_addrs(self):
+    def clear_addrs(self) -> None:
         self.addrs = []
 
-    def put_metadata(self, key, val):
+    def put_metadata(self, key: Any, val: Any) -> None:
         self.metadata[key] = val
 
-    def get_metadata(self, key):
+    def get_metadata(self, key: Any) -> Any:
         if key in self.metadata:
             return self.metadata[key]
         raise PeerDataError("key not found")