arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-23 06:48:16 +00:00
Code Issues Releases Wiki Activity

core/sign_tx: improve signer selection readability

Browse Source
This commit is contained in:
matejcik 2020-04-30 15:27:30 +02:00 committed by Andrew Kozlik
parent 8b89a30955
commit ff41e5c304
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
core/src/apps/wallet/sign_tx/__init__.py
View File

@ -20,25 +20,31 @@ if not utils.BITCOIN_ONLY:
from apps.wallet.sign_tx import bitcoinlike, decred, zcash
if False:
from typing import Union
from typing import Type, Union
BITCOIN_NAMES = ("Bitcoin", "Regtest", "Testnet")
async def sign_tx(ctx: wire.Context, msg: SignTx, keychain: seed.Keychain) -> TxRequest:
coin_name = msg.coin_name if msg.coin_name is not None else "Bitcoin"
coin = coins.by_name(coin_name)
try:
if not utils.BITCOIN_ONLY and coin.decred:
signer = decred.Decred(msg, keychain, coin).signer()
elif not utils.BITCOIN_ONLY and coin.overwintered:
signer = zcash.Overwintered(msg, keychain, coin).signer()
elif not utils.BITCOIN_ONLY and coin_name not in (
"Bitcoin",
"Regtest",
"Testnet",
):
signer = bitcoinlike.Bitcoinlike(msg, keychain, coin).signer()
if not utils.BITCOIN_ONLY:
if coin.decred:
signer_class = decred.Decred # type: Type[bitcoin.Bitcoin]
elif coin.overwintered:
signer_class = zcash.Overwintered
elif coin_name not in BITCOIN_NAMES:
signer_class = bitcoinlike.Bitcoinlike
else:
signer = bitcoin.Bitcoin(msg, keychain, coin).signer()
signer_class = bitcoin.Bitcoin
else:
signer_class = bitcoin.Bitcoin
try:
signer = signer_class(msg, keychain, coin).signer()
except common.SigningError as e:
raise wire.Error(*e.args)