clean-up: use is_segwit_input in wallet.py

SomberNight · SomberNight · commit 4e4a7746fcb1 · 2017-12-11T01:49:38.000+01:00
diff --git a/lib/wallet.py b/lib/wallet.py
@@ -1096,7 +1096,7 @@ def add_input_info(self, txin):
         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
