From 07bdbe9999ad0ade3ec01d86608cbde982891fe3 Mon Sep 17 00:00:00 2001 From: nolash Date: Fri, 25 Dec 2020 12:34:05 +0100 Subject: [PATCH] Handle string and binary in signer backend --- crypto_dev_signer/eth/signer/defaultsigner.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/crypto_dev_signer/eth/signer/defaultsigner.py b/crypto_dev_signer/eth/signer/defaultsigner.py index 46d185d..1490550 100644 --- a/crypto_dev_signer/eth/signer/defaultsigner.py +++ b/crypto_dev_signer/eth/signer/defaultsigner.py @@ -44,6 +44,12 @@ class ReferenceSigner(Signer): #msg = b'\x19Ethereum Signed Message:\n{}{}'.format(len(message), message) k = keys.PrivateKey(self.keyGetter.get(address, password)) #z = keys.ecdsa_sign(message_hash=g, private_key=k) - z = k.sign_msg(message.encode('utf-8')) + z = None + if type(message).__name__ == 'str': + z = k.sign_msg(message.encode('utf-8')) + elif type(message).__name__ == 'bytes': + z = k.sign_msg(message) + else: + raise ValueError('message must be type str or bytes, received {}'.format(type(message).__name__)) return z