varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix lint

Browse Source
This commit is contained in:
NIC619
2019-11-25 17:32:53 +08:00
parent e28a974425
commit 566e4c080d
5 changed files with 17 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
libp2p/peer/peerstore.py
View File

@ -1,4 +1,4 @@
from typing import Any, Dict, List, Optional, Sequence, Tuple
from typing import Any, Dict, List, Optional, Sequence, Set
from multiaddr import Multiaddr
@ -24,7 +24,8 @@ class PeerStore(IPeerStore):
def __create_or_get_peer(self, peer_id: ID) -> PeerData:
"""
Returns the peer data for peer_id or creates a new peer data (and
stores it in peer_data_map) if peer data for peer_id does not yet exist.
stores it in peer_data_map) if peer data for peer_id does not yet
exist.
:param peer_id: peer ID
:return: peer data
@ -158,7 +159,9 @@ class PeerStore(IPeerStore):
:raise PeerStoreError: if peer ID already has pubkey set
"""
if peer_id in self.peer_pubkey_map:
raise PeerStoreError(f"peer ID already has pubkey: {self.peer_pubkey_map[peer_id]}")
raise PeerStoreError(
f"peer ID already has pubkey: {self.peer_pubkey_map[peer_id]}"
)
self.peer_pubkey_map[peer_id] = pubkey
def pubkey(self, peer_id: ID) -> PublicKey:
@ -178,7 +181,9 @@ class PeerStore(IPeerStore):
:raise PeerStoreError: if peer ID already has privkey set
"""
if peer_id in self.peer_privkey_map:
raise PeerStoreError(f"peer ID already has privkey: {self.peer_privkey_map[peer_id]}")
raise PeerStoreError(
f"peer ID already has privkey: {self.peer_privkey_map[peer_id]}"
)
self.peer_privkey_map[peer_id] = privkey
def privkey(self, peer_id: ID) -> PrivateKey:
@ -191,7 +196,7 @@ class PeerStore(IPeerStore):
return self.peer_privkey_map[peer_id]
raise PeerStoreError("peer ID not found")
def peers_with_keys(self) -> Tuple[ID]:
def peers_with_keys(self) -> Set[ID]:
"""
:return: all of the peer IDs which has pubkey/privkey stored in peer store
"""

4
libp2p/peer/peerstore_interface.py
View File

@ -1,5 +1,5 @@
from abc import abstractmethod
from typing import Any, List, Sequence, Tuple
from typing import Any, List, Sequence, Set
from multiaddr import Multiaddr
@ -132,7 +132,7 @@ class IPeerStore(IAddrBook, IPeerMetadata):
"""
@abstractmethod
def peers_with_keys(self) -> Tuple[ID]:
def peers_with_keys(self) -> Set[ID]:
"""
:return: all of the peer IDs which has pubkey/privkey stored in peer store
"""