Supply local priv and pub key when upgrading to a secure transport

This commit is contained in:
Alex Stokes
2019-08-02 21:55:37 -07:00
parent 7942b7eaa7
commit 9e18d7561d
3 changed files with 10 additions and 4 deletions

View File

@ -8,6 +8,6 @@ class BaseSecureTransport(ISecureTransport):
is only meant to be used in clases that derive from it.
"""
def __init__(self, local_private_key: bytes) -> None:
def __init__(self, local_private_key: bytes, local_public_key: bytes) -> None:
self.local_private_key = local_private_key
self.local_peer = ID.from_privkey(local_private_key)
self.local_peer = ID.from_pubkey(local_public_key)