From c26f5e6d5b53b57a8d6aa84dcb607071387ea228 Mon Sep 17 00:00:00 2001 From: jMyles Date: Sat, 24 Feb 2018 01:13:40 -0800 Subject: [PATCH] Better way to get public keys. --- nkms/characters.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/nkms/characters.py b/nkms/characters.py index 49e32e6e3..053bf9fe3 100644 --- a/nkms/characters.py +++ b/nkms/characters.py @@ -266,9 +266,9 @@ class Character(object): def id(self): return hexlify(bytes(self.stamp)) - def public_key(self, key_class): - # TODO: Does it make sense to have a specialized exception here? Probably. - return self._crypto_power.public_keys[key_class] + def public_key(self, power_up_class): + power_up = self._crypto_power.power_ups(power_up_class) + return power_up.public_key() class Alice(Character): @@ -729,7 +729,7 @@ class SignatureStamp(object): return self.character._crypto_power.sign(*args, **kwargs) def __bytes__(self): - return self.character._crypto_power.pubkey_sig_bytes() + return bytes(self.character.public_key(SigningPower)) def __eq__(self, other): return other == bytes(self)