Change peerinfo to a class rather than a dictionary

This commit is contained in:
Stuckinaboot
2018-10-29 14:09:47 +01:00
parent e93ced5f6a
commit 95d3847d5a
3 changed files with 8 additions and 4 deletions

View File

@ -1,5 +1,6 @@
from .peerstore_interface import IPeerStore
from .peerdata import PeerData
from .peerinfo import PeerInfo
class PeerStore(IPeerStore):
@ -24,10 +25,7 @@ class PeerStore(IPeerStore):
def peer_info(self, peer_id):
if peer_id in self.peer_map:
peer = self.peer_map[peer_id]
return {
"peer_id": peer_id,
"addrs": peer.get_addrs()
}
return PeerInfo(peer_id, peer)
return None
def get_protocols(self, peer_id):