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

Replace errors with exceptions

Browse Source
This commit is contained in:
Stuckinaboot
2018-10-29 13:13:39 +01:00
parent 2544f8dff5
commit e93ced5f6a
4 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
peer/peerstore.py
View File

@ -32,8 +32,9 @@ class PeerStore(IPeerStore):
def get_protocols(self, peer_id):
if peer_id in self.peer_map:
return self.peer_map[peer_id].get_protocols(), None
return None, peer_id + " not found"
return self.peer_map[peer_id].get_protocols()
else:
raise Exception("peer ID not found")
def add_protocols(self, peer_id, protocols):
peer = self.__create_or_get_peer(peer_id)
@ -44,9 +45,10 @@ class PeerStore(IPeerStore):
def get(self, peer_id, key):
if peer_id in self.peer_map:
val, error = self.peer_map[peer_id].get_metadata(key)
return val, error
return None, peer_id + " not found"
val = self.peer_map[peer_id].get_metadata(key)
return val
else:
raise Exception("peer ID not found")
def put(self, peer_id, key, val):
# <<?>>
@ -64,8 +66,9 @@ class PeerStore(IPeerStore):
def addrs(self, peer_id):
if peer_id in self.peer_map:
return self.peer_map[peer_id].get_addrs(), None
return None, peer_id + " not found"
return self.peer_map[peer_id].get_addrs()
else:
raise Exception("peer ID not found")
def clear_addrs(self, peer_id):
# Only clear addresses if the peer is in peer map