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

Browse Source
This commit is contained in:
NIC619
2019-07-30 23:41:28 +08:00
parent 76de01a17d
commit 5e215901c0
8 changed files with 17 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
libp2p/peer/id.py
View File

@ -20,16 +20,15 @@ MAX_INLINE_KEY_LENGTH = 42
class ID:
_id_str: str
_id_str: bytes
def __init__(self, id_str: str) -> None:
def __init__(self, id_str: bytes) -> None:
self._id_str = id_str
# FIXME: Should return type `bytes`
def to_bytes(self) -> str:
def to_bytes(self) -> bytes:
return self._id_str
def get_raw_id(self) -> str:
def get_raw_id(self) -> bytes:
return self._id_str
def pretty(self) -> str:
@ -86,6 +85,6 @@ def id_from_private_key(key: RsaKey) -> ID:
return id_from_public_key(key.publickey())
def digest(data: Union[str, bytes]) -> bytes:
if not isinstance(data, bytes):
data = str(data).encode('utf8')
if isinstance(data, str):
data = data.encode('utf8')
return hashlib.sha1(data).digest()

2
libp2p/peer/peerdata.py
View File

@ -31,7 +31,7 @@ class PeerData(IPeerData):
self.protocols = list(protocols)
def add_addrs(self, addrs: Sequence[Multiaddr]) -> None:
self.addrs.extend(list(addrs))
self.addrs.extend(addrs)
def get_addrs(self) -> List[Multiaddr]:
return self.addrs