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

8
libp2p/crypto/ecc.py
View File

@ -1,3 +1,5 @@
from typing import cast
from Crypto.PublicKey import ECC
from Crypto.PublicKey.ECC import EccKey
@ -9,7 +11,7 @@ class ECCPublicKey(PublicKey):
self.impl = impl
def to_bytes(self) -> bytes:
return self.impl.export_key("DER")
return cast(bytes, self.impl.export_key(format="DER"))
@classmethod
def from_bytes(cls, data: bytes) -> "ECCPublicKey":
@ -33,7 +35,7 @@ class ECCPrivateKey(PrivateKey):
return cls(private_key_impl)
def to_bytes(self) -> bytes:
return self.impl.export_key("DER")
return cast(bytes, self.impl.export_key(format="DER"))
def get_type(self) -> KeyType:
return KeyType.ECC_P256
@ -42,7 +44,7 @@ class ECCPrivateKey(PrivateKey):
raise NotImplementedError
def get_public_key(self) -> PublicKey:
return ECCPublicKey(self.impl.publickey())
return ECCPublicKey(self.impl.public_key())
def create_new_key_pair(curve: str) -> KeyPair: