tools: fix signtest.py

pull/25/head
Pavol Rusnak 8 years ago
parent 046ced3f81
commit d1bfc76d91
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D

@ -11,8 +11,8 @@ import trezorlib.ckd_public as bip32
import hashlib import hashlib
from trezorlib.client import TrezorClient from trezorlib.client import TrezorClient
from trezorlib.client import TrezorClientDebug from trezorlib.client import TrezorClientDebug
from trezorlib.tx_api import TXAPITestnet from trezorlib.tx_api import TxApiTestnet
from trezorlib.tx_api import TXAPIBitcoin from trezorlib.tx_api import TxApiBitcoin
from trezorlib.transport_hid import HidTransport from trezorlib.transport_hid import HidTransport
from trezorlib.transport_bridge import BridgeTransport from trezorlib.transport_bridge import BridgeTransport
@ -44,7 +44,7 @@ def string_to_int(s):
result = (result << 8) + c result = (result << 8) + c
return result return result
class MyTXAPIBitcoin(object): class MyTxApiBitcoin(object):
def set_publickey(self, node): def set_publickey(self, node):
self.node = node.node self.node = node.node
@ -113,8 +113,8 @@ class MyTXAPIBitcoin(object):
outi = self.inputs.append( outi = self.inputs.append(
proto_types.TxInputType( proto_types.TxInputType(
address_n=self.client.expand_path("44'/0'/0'/0/"+str(idx)), address_n=self.client.expand_path("44'/0'/0'/0/"+str(idx)),
script_type = (proto_types.SPENDWADDRESS if segwit == 1 else script_type = (proto_types.SPENDWITNESS if segwit == 1 else
proto_types.SPENDP2SHWADDRESS if segwit == 2 else proto_types.SPENDP2SHWITNESS if segwit == 2 else
proto_types.SPENDADDRESS), proto_types.SPENDADDRESS),
prev_hash=txhash, prev_hash=txhash,
prev_index = myout, prev_index = myout,
@ -159,18 +159,18 @@ def main():
# transport = BridgeTransport(devices[0][0]) # transport = BridgeTransport(devices[0][0])
transport = HidTransport(devices[0]) transport = HidTransport(devices[0])
txstore = MyTXAPIBitcoin() txstore = MyTxApiBitcoin()
# Creates object for manipulating TREZOR # Creates object for manipulating TREZOR
client = TrezorClient(transport) client = TrezorClient(transport)
# client.set_tx_api(TXAPITestnet()) # client.set_tx_api(TxApiTestnet)
txstore.set_client(client) txstore.set_client(client)
txstore.set_publickey(client.get_public_node(client.expand_path("44'/0'/0'"))) txstore.set_publickey(client.get_public_node(client.expand_path("44'/0'/0'")))
print("creating input txs") print("creating input txs")
txstore.create_inputs(numinputs, sizeinputtx) txstore.create_inputs(numinputs, sizeinputtx)
print("go") print("go")
client.set_tx_api(txstore) client.set_tx_api(txstore)
# client.set_tx_api(MyTXAPIBitcoin()) # client.set_tx_api(MyTxApiBitcoin())
# Print out TREZOR's features and settings # Print out TREZOR's features and settings
print(client.features) print(client.features)

Loading…
Cancel
Save