funga/crypto_dev_signer/keystore/dict.py
2021-01-09 22:05:24 +01:00

34 lines
885 B
Python

# standard imports
import logging
# local imports
from . import keyapi
from .interface import Keystore
from crypto_dev_signer.error import UnknownAccountError
from crypto_dev_signer.common import strip_hex_prefix
logg = logging.getLogger()
class DictKeystore(Keystore):
def __init__(self):
self.keys = {}
def get(self, address, password=None):
if password != None:
logg.debug('password ignored as dictkeystore doesnt do encryption')
try:
return self.keys[address]
except KeyError:
raise UnknownAccountError(address)
def import_key(self, pk, password=None):
pubk = keyapi.private_key_to_public_key(pk)
address_hex = pubk.to_checksum_address()
address_hex_clean = strip_hex_prefix(address_hex)
self.keys[address_hex_clean] = pk.to_bytes()
return address_hex