Allow generate_key to have a pubkey param

pull/55/head
tuxxy 2017-09-26 13:06:52 -07:00
parent 82864f643d
commit b3f0335679
1 changed files with 3 additions and 2 deletions

View File

@ -16,14 +16,15 @@ class EncryptingKeypair(object):
self.priv_key = privkey_bytes
self.pub_key = self.pre.priv2pub(self.priv_key)
def generate_key(self):
def generate_key(self, pubkey=None):
"""
Generate a raw symmetric key and its encrypted counterpart.
:rtype: Tuple(bytes, bytes)
:return: Tuple of the raw encrypted key and the encrypted key
"""
symm_key, enc_symm_key = self.pre.encapsulate(self.pub_key)
pubkey = pubkey or self.pub_key
symm_key, enc_symm_key = self.pre.encapsulate(pubkey)
return (symm_key, enc_symm_key)
def decrypt_key(self, enc_key):