clean-up: use is_segwit_input in wallet.py

2017-12-11 01:49:38 +01:00 · 2017-12-11 01:49:38 +01:00 · 4e4a7746fc
parent d403c59ad7
commit 4e4a7746fc
1 changed files with 1 additions and 1 deletions
--- a/lib/wallet.py
+++ b/lib/wallet.py
@ -1096,7 +1096,7 @@ class Abstract_Wallet(PrintError):
        if self.is_mine(address):
            txin['type'] = self.get_txin_type(address)
            # segwit needs value to sign
-            if txin.get('value') is None and txin['type'] in ['p2wpkh', 'p2wsh', 'p2wpkh-p2sh', 'p2wsh-p2sh']:
+            if txin.get('value') is None and Transaction.is_segwit_input(txin):
                received, spent = self.get_addr_io(address)
                item = received.get(txin['prevout_hash']+':%d'%txin['prevout_n'])
                tx_height, value, is_cb = item