Clean up key gen

This commit is contained in:
Alex Stokes
2019-08-15 18:36:50 -07:00
parent d17e6f3392
commit 7535a02da7
7 changed files with 44 additions and 98 deletions

View File

@ -1,4 +1,4 @@
from libp2p.crypto.keys import PrivateKey, PublicKey
from libp2p.crypto.keys import KeyPair
from libp2p.peer.id import ID
from libp2p.security.secure_transport_interface import ISecureTransport
@ -9,8 +9,6 @@ class BaseSecureTransport(ISecureTransport):
is only meant to be used in clases that derive from it.
"""
def __init__(
self, local_private_key: PrivateKey, local_public_key: PublicKey
) -> None:
self.local_private_key = local_private_key
self.local_peer = ID.from_pubkey(local_public_key)
def __init__(self, local_key_pair: KeyPair) -> None:
self.local_private_key = local_key_pair.private_key
self.local_peer = ID.from_pubkey(local_key_pair.public_key)