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

Typing and linting fixes

Browse Source
This commit is contained in:
Alex Stokes
2019-08-23 23:43:36 +02:00
parent 0fa3331b8c
commit 1adef05e94
8 changed files with 53 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
libp2p/crypto/keys.py
View File

@ -33,8 +33,10 @@ class Key(ABC):
"""
...
def __eq__(self, other: "Key") -> bool:
return self.impl == other.impl
def __eq__(self, other: object) -> bool:
if not isinstance(other, Key):
return NotImplemented
return self.to_bytes() == other.to_bytes()
class PublicKey(Key):
@ -66,9 +68,7 @@ class PublicKey(Key):
@classmethod
def deserialize_from_protobuf(cls, protobuf_data: bytes) -> protobuf.PublicKey:
protobuf_key = protobuf.PublicKey()
protobuf_key.ParseFromString(protobuf_data)
return protobuf_key
return protobuf.PublicKey.FromString(protobuf_data)
class PrivateKey(Key):
@ -110,9 +110,7 @@ class PrivateKey(Key):
@classmethod
def deserialize_from_protobuf(cls, protobuf_data: bytes) -> protobuf.PrivateKey:
protobuf_key = protobuf.PrivateKey()
protobuf_key.ParseFromString(protobuf_data)
return protobuf_key
return protobuf.PrivateKey.FromString(protobuf_data)
@dataclass(frozen=True)