varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-10 15:10:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply PR feedback: add_key_pair

Browse Source
This commit is contained in:
NIC619
2019-11-26 11:52:31 +08:00
parent 076dae50f3
commit 828ae69c66
4 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
libp2p/peer/peerstore.py
View File

@ -2,7 +2,7 @@ from typing import Any, Dict, List, Optional, Sequence, Set
from multiaddr import Multiaddr
from libp2p.crypto.keys import PrivateKey, PublicKey
from libp2p.crypto.keys import KeyPair, PrivateKey, PublicKey
from .id import ID
from .peerdata import PeerData, PeerDataError
@ -34,7 +34,7 @@ class PeerStore(IPeerStore):
return self.peer_data_map[peer_id]
data = PeerData()
self.peer_data_map[peer_id] = data
return self.peer_data_map[peer_id]
return data
def peer_info(self, peer_id: ID) -> Optional[PeerInfo]:
"""
@ -196,6 +196,15 @@ class PeerStore(IPeerStore):
return self.peer_privkey_map[peer_id]
raise PeerStoreError("peer ID not found")
def add_key_pair(self, peer_id: ID, key_pair: KeyPair) -> None:
"""
:param peer_id: peer ID to add private key for
:param key_pair:
:raise PeerStoreError: if peer ID already has pubkey or privkey set
"""
self.add_pubkey(peer_id, key_pair.public_keypubkey)
self.add_privkey(peer_id, key_pair.private_key)
def peers_with_keys(self) -> Set[ID]:
"""
:return: all of the peer IDs which has pubkey/privkey stored in peer store