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

Pull request feedback

Browse Source
This commit is contained in:
Aratz M. Lasa
2019-10-15 20:32:25 +02:00
parent fa1637850e
commit ac9feef26c
5 changed files with 31 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
libp2p/routing/kademlia/kademlia_peer_router.py
View File

@ -1,3 +1,7 @@
import json
import multiaddr
from libp2p.kademlia.network import KademliaServer
from libp2p.peer.id import ID
from libp2p.peer.peerinfo import PeerInfo
@ -21,5 +25,18 @@ class KadmeliaPeerRouter(IPeerRouting):
# ignore type for kad
value = await self.server.get(xor_id) # type: ignore
return (
PeerInfo.info_from_string(value) if value else None
peer_info_from_str(value) if value else None
) # TODO: should raise error if None?
def peer_info_to_str(peer_info: PeerInfo) -> str:
return json.dumps(
[peer_info.peer_id.to_string(), list(map(lambda a: str(a), peer_info.addrs))]
)
def peer_info_from_str(string: str) -> PeerInfo:
peer_id, raw_addrs = json.loads(string)
return PeerInfo(
ID.from_base58(peer_id), list(map(lambda a: multiaddr.Multiaddr(a), raw_addrs))
)