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

Internalize the protobuf serialization to the concept of a Key

Browse Source 
Given its use across various components of `libp2p` (not just peer IDs),
it makes the abstraction cleaner to pull the serialization into the
key class and expose the canonical serialization to bytes.
This commit is contained in:
Alex Stokes
2019-08-20 19:01:36 +02:00
parent 5d611801c7
commit 87d943aa39
3 changed files with 21 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/peer/id.py
View File

@ -7,13 +7,6 @@ import multihash
from libp2p.crypto.keys import PublicKey
def _serialize_public_key(key: PublicKey) -> bytes:
"""
Serializes ``key`` in the way expected to form valid peer ids.
"""
return key.serialize_to_protobuf().SerializeToString()
class ID:
_bytes: bytes
_xor_id: int = None
@ -62,7 +55,7 @@ class ID:
@classmethod
def from_pubkey(cls, key: PublicKey) -> "ID":
serialized_key = _serialize_public_key(key)
serialized_key = key.serialize()
algo = multihash.Func.sha2_256
mh_digest = multihash.digest(serialized_key, algo)
return cls(mh_digest.encode())