#!/usr/bin/python
import binascii

import trezorlib.types_pb2 as proto_types

from trezorlib.client import TrezorClient, TrezorClientDebug
from trezorlib.tx_api import TXAPITestnet
from trezorlib.tx_api import TXAPIBitcoin
from trezorlib.transport_hid import HidTransport

FROM_BIP="44'/0'/0'/0/0"
PREVHASH='67ca16b97014a56189f50f1a30b4db68e8ea314b33473bd0553916078f6a8250'
N=2
TO='1H6w1CUk8cmSeaoGJ363LY9LWe57FPrCeA'
OPRETURN_DATA=binascii.unhexlify("6f6bf889b4e439735d37c3611b0942d26df866aaf773c825208fbb6a")
TOTAL=22853000
FEE=10000
DUST=5430

def main():
    # List all connected TREZORs on USB
    devices = HidTransport.enumerate()

    # Check whether we found any
    if len(devices) == 0:
        print 'No TREZOR found'
        return

    # Use first connected device
    transport = HidTransport(devices[0])

    # Creates object for manipulating TREZOR
    client = TrezorClient(transport)
    client.set_tx_api(TXAPIBitcoin())

    # The list of inputs.
    inputs = [
        proto_types.TxInputType(
            address_n=client.expand_path(FROM_BIP),
            prev_hash=binascii.unhexlify(PREVHASH),
            prev_index=N,
        ),
    ]

    outputs = [
        proto_types.TxOutputType(
            amount=DUST,
            script_type=proto_types.PAYTOADDRESS,
            address=TO,
        ),
        proto_types.TxOutputType(
            amount=0,
            script_type=proto_types.PAYTOOPRETURN,
            op_return_data=OPRETURN_DATA,
        ),

        proto_types.TxOutputType(
            amount=TOTAL-FEE-DUST,
            script_type=proto_types.PAYTOADDRESS,
            address_n=client.expand_path(FROM_BIP),
        ),
    ]
 
    (signatures, serialized_tx) = client.sign_tx('Bitcoin', inputs, outputs)
    print 'Transaction:', binascii.hexlify(serialized_tx)

    client.close()

if __name__ == '__main__':
    main()