lib/commands.py

   1 #!/usr/bin/env python
   2 #
   3 # Electrum - lightweight Bitcoin client
   4 # Copyright (C) 2011 thomasv@gitorious
   5 #
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18
  19
  20 from util import *
  21 from bitcoin import *
  22 from decimal import Decimal
  23 import bitcoin
  24
  25 known_commands = {
  26     'help':'Prints this help',
  27     'validateaddress':'Check that the address is valid',
  28     'balance': "Display the balance of your wallet or of an address.\nSyntax: balance [<address>]",
  29     'contacts': "Show your list of contacts",
  30     'create':'Create a wallet',
  31     'restore':'Restore a wallet',
  32     'payto':"""Create and broadcast a transaction.
  33 Syntax: payto <recipient> <amount> [label]
  34 <recipient> can be a bitcoin address or a label
  35 options:\n  --fee, -f: set transaction fee\n  --fromaddr, -s: send from address -\n  --changeaddr, -c: send change to address
  36             """,
  37     'sendrawtransaction':
  38             'Broadcasts a transaction to the network. \nSyntax: sendrawtransaction <tx in hexadecimal>',
  39     'password':
  40             "Changes your password",
  41     'addresses':
  42             """Shows your list of addresses.
  43 options:
  44   -a: show all addresses, including change addresses""",
  45
  46     'history':"Shows the transaction history",
  47     'setlabel':'Assign a label to an item\nSyntax: label <tx_hash> <label>',
  48     'mktx':
  49         """Create a signed transaction, password protected.
  50 Syntax: mktx <recipient> <amount> [label]
  51 options:\n  --fee, -f: set transaction fee\n  --fromaddr, -s: send from address -\n  --changeaddr, -c: send change to address
  52         """,
  53     'get_seed':
  54             "Print the generation seed of your wallet.",
  55     'importprivkey':
  56             'Import a private key\nSyntax: importprivkey <privatekey>',
  57     'signmessage':
  58             'Signs a message with a key\nSyntax: signmessage <address> <message>\nIf you want to lead or end a message with spaces, or want double spaces inside the message make sure you quote the string. I.e. " Hello  This is a weird String "',
  59     'verifymessage':
  60              'Verifies a signature\nSyntax: verifymessage <address> <signature> <message>\nIf you want to lead or end a message with spaces, or want double spaces inside the message make sure you quote the string. I.e. " Hello  This is a weird String "',
  61     'eval':
  62              "Run python eval() on an object\nSyntax: eval <expression>\nExample: eval \"wallet.aliases\"",
  63     'get':
  64              "Get config parameter.",
  65     'set':
  66              "Set config parameter.",
  67     'deseed':
  68             "Create a seedless, watching-only wallet.",
  69     'freeze':'',
  70     'unfreeze':'',
  71     'prioritize':'',
  72     'unprioritize':'',
  73     'dumpprivkey':'similar to bitcoind\'s command',
  74     'dumpprivkeys':'dump all private keys',
  75     'listunspent':'similar to bitcoind\'s command',
  76     'createmultisig':'similar to bitcoind\'s command',
  77     'createrawtransaction':'similar to bitcoind\'s command',
  78     'decoderawtransaction':'similar to bitcoind\'s command',
  79     'signrawtransaction':'similar to bitcoind\'s command',
  80     'get_history': 'get history for an address'
  81
  82     }
  83
  84
  85
  86 offline_commands = [ 'password', 'mktx',
  87                      'setlabel', 'contacts',
  88                      'help', 'validateaddress',
  89                      'signmessage', 'verifymessage',
  90                      'eval', 'set', 'get', 'create', 'addresses',
  91                      'importprivkey', 'get_seed',
  92                      'deseed',
  93                      'freeze','unfreeze',
  94                      'prioritize','unprioritize',
  95                      'dumpprivkey','dumpprivkeys','listunspent',
  96                      'createmultisig', 'createrawtransaction', 'decoderawtransaction', 'signrawtransaction'
  97                      ]
  98
  99 protected_commands = ['payto', 'password', 'mktx', 'get_seed', 'importprivkey','signmessage', 'signrawtransaction', 'dumpprivkey', 'dumpprivkeys' ]
 100
 101 class Commands:
 102
 103     def __init__(self, wallet, interface, callback = None):
 104         self.wallet = wallet
 105         self.interface = interface
 106         self.callback = callback
 107
 108     def _run(self, method, args, password_getter):
 109         if method in protected_commands:
 110             self.password = apply(password_getter,())
 111         f = eval('self.'+method)
 112         result = apply(f,args)
 113         self.password = None
 114         if self.callback:
 115             apply(self.callback, ())
 116         return result
 117
 118     def get_history(self, addr):
 119         h = self.wallet.get_history(addr)
 120         if h is None: h = self.wallet.interface.synchronous_get([ ('blockchain.address.get_history',[addr]) ])[0]
 121         return h
 122
 123     def listunspent(self):
 124         l = self.wallet.get_unspent_coins()
 125         for i in l: i["value"] = i["value"]*1e-8
 126         return l
 127
 128     def createrawtransaction(self, inputs, outputs):
 129         # convert to own format
 130         for i in inputs:
 131             i['tx_hash'] = i['txid']
 132             i['index'] = i['vout']
 133         outputs = map(lambda x: (x[0],int(1e8*x[1])), outputs.items())
 134         tx = Transaction.from_io(inputs, outputs)
 135         return tx.as_dict()
 136
 137     def signrawtransaction(self, raw_tx, input_info, private_keys):
 138         tx = Transaction(raw_tx)
 139         unspent_coins = self.wallet.get_unspent_coins()
 140
 141         # convert private_keys to dict
 142         pk = {}
 143         for sec in private_keys:
 144             address = bitcoin.address_from_private_key(sec)
 145             pk[address] = sec
 146         private_keys = pk
 147
 148         for txin in tx.inputs:
 149             # convert to own format
 150             txin['tx_hash'] = txin['prevout_hash']
 151             txin['index'] = txin['prevout_n']
 152
 153             for item in input_info:
 154                 if item.get('txid') == txin['tx_hash'] and item.get('vout') == txin['index']:
 155                     txin['raw_output_script'] = item['scriptPubKey']
 156                     txin['redeemScript'] = item.get('redeemScript')
 157                     txin['electrumKeyID'] = item.get('electrumKeyID')
 158                     break
 159             else:
 160                 for item in unspent_coins:
 161                     if txin['tx_hash'] == item['tx_hash'] and txin['index'] == item['index']:
 162                         txin['raw_output_script'] = item['raw_output_script']
 163                         break
 164                 else:
 165                     # if neither, we might want to get it from the server..
 166                     raise
 167
 168             # find the address:
 169             import deserialize
 170             if txin.get('electrumKeyID'):
 171                 n, for_change = txin.get('electrumKeyID')
 172                 sec = wallet.sequence.get_private_key(n, for_change, seed)
 173                 address = bitcoin.address_from_private_key(sec)
 174                 txin['address'] = address
 175                 private_keys[address] = sec
 176
 177             elif txin.get("redeemScript"):
 178                 txin['address'] = bitcoin.hash_160_to_bc_address(bitcoin.hash_160(txin.get("redeemScript").decode('hex')), 5)
 179
 180             elif txin.get("raw_output_script"):
 181                 addr = deserialize.get_address_from_output_script(txin.get("raw_output_script").decode('hex'))
 182                 sec = wallet.get_private_key(addr, self.password)
 183                 if sec:
 184                     private_keys[addr] = sec
 185                     txin['address'] = addr
 186
 187         tx.sign( private_keys )
 188         return tx.as_dict()
 189
 190     def decoderawtransaction(self, raw):
 191         tx = Transaction(raw)
 192         return tx.deserialize()
 193
 194     def sendrawtransaction(self, raw):
 195         tx = Transaction(raw)
 196         r, h = wallet.sendtx( tx )
 197         return h
 198
 199     def createmultisig(self, num, pubkeys):
 200         assert isinstance(pubkeys, list)
 201         return Transaction.multisig_script(pubkeys, num)
 202
 203     def freeze(self,addr):
 204         return self.wallet.freeze(addr)
 205
 206     def unfreeze(self,addr):
 207         return self.wallet.unfreeze(addr)
 208
 209     def prioritize(self, addr):
 210         return self.wallet.prioritize(addr)
 211
 212     def unprioritize(self, addr):
 213         return self.wallet.unprioritize(addr)
 214
 215     def dumpprivkey(self, addr):
 216         return self.wallet.get_private_key(addr, self.password)
 217
 218     def dumpprivkeys(self, addresses):
 219         return self.wallet.get_private_keys(addresses, self.password)
 220
 221     def validateaddress(self,addr):
 222         is_valid = self.wallet.is_valid(addr)
 223         out = { 'isvalid':is_valid }
 224         if is_valid:
 225             is_mine = self.wallet.is_mine(addr)
 226             out['address'] = addr
 227             out['ismine'] = is_mine
 228             if is_mine:
 229                 out['pubkey'] = self.wallet.get_public_key(addr)
 230
 231         return out
 232
 233
 234     def balance(self, addresses = []):
 235         if addresses == []:
 236             c, u = self.wallet.get_balance()
 237             if u:
 238                 print_msg(Decimal( c ) / 100000000 , Decimal( u ) / 100000000)
 239             else:
 240                 print_msg(Decimal( c ) / 100000000)
 241         else:
 242             for addr in addresses:
 243                 c, u = wallet.get_addr_balance(addr)
 244                 if u:
 245                     print_msg("%s %s, %s" % (addr, str(Decimal(c)/100000000), str(Decimal(u)/100000000)))
 246                 else:
 247                     print_msg("%s %s" % (addr, str(Decimal(c)/100000000)))
 248
 249
 250     def get_seed(self):
 251         import mnemonic
 252         seed = self.wallet.decode_seed(self.password)
 253         print_msg(seed + ' "' + ' '.join(mnemonic.mn_encode(seed)) + '"')
 254
 255     def importprivkey(self, sec):
 256         try:
 257             addr = wallet.import_key(sec,self.password)
 258             wallet.save()
 259             print_msg("Keypair imported: ", addr)
 260         except BaseException as e:
 261             print_msg("Error: Keypair import failed: " + str(e))
 262
 263
 264     def sign_message(self, address, message):
 265         return self.wallet.sign_message(address, message, self.password)
 266
 267
 268     def verify_message(self, address, signature, message):
 269         try:
 270             EC_KEY.verify_message(address, signature, message)
 271             return True
 272         except BaseException as e:
 273             print_error("Verification error: {0}".format(e))
 274             return False
 275
 276
 277     def _mktx(self, to_address, amount, fee = None, change_addr = None, from_addr = None):
 278         for k, v in self.wallet.labels.items():
 279             if v == to_address:
 280                 to_address = k
 281                 print_msg("alias", to_address)
 282                 break
 283             if change_addr and v == change_addr:
 284                 change_addr = k
 285
 286         amount = int(10000000*amount)
 287         if fee: fee = int(10000000*fee)
 288         return self.wallet.mktx( [(to_address, amount)], self.password, fee , change_addr, from_addr)
 289
 290
 291     def mktx(self, to_address, amount, fee = None, change_addr = None, from_addr = None):
 292         tx = self._mktx(to_address, amount, fee, change_addr, from_addr)
 293         return tx.as_dict()
 294
 295
 296     def payto(self, to_address, amount, fee = None, change_addr = None, from_addr = None):
 297         tx = self._mktx(to_address, amount, fee, change_addr, from_addr)
 298         r, h = wallet.sendtx( tx )
 299         return h
 300
 301
 302     def history(self):
 303         import datetime
 304         balance = 0
 305         for item in self.wallet.get_tx_history():
 306             tx_hash, conf, is_mine, value, fee, balance, timestamp = item
 307             try:
 308                 time_str = datetime.datetime.fromtimestamp( timestamp).isoformat(' ')[:-3]
 309             except:
 310                 time_str = "----"
 311
 312             label, is_default_label = self.wallet.get_label(tx_hash)
 313             if not label: label = tx_hash
 314             else: label = label + ' '*(64 - len(label) )
 315
 316             print_msg("%17s"%time_str, "  " + label + "  " + format_satoshis(value)+ "  "+ format_satoshis(balance))
 317         print_msg("# balance: ", format_satoshis(balance))
 318
 319
 320     def setlabel(self, tx, label):
 321         self.wallet.labels[tx] = label
 322         self.wallet.save()
 323
 324
 325     def contacts(self):
 326         c = {}
 327         for addr in self.wallet.addressbook:
 328             c[addr] = self.wallet.labels.get(addr)
 329         return c
 330
 331
 332     def addresses(self, show_all):
 333         for addr in self.wallet.all_addresses():
 334             if show_all or not self.wallet.is_change(addr):
 335
 336                 flags = self.wallet.get_address_flags(addr)
 337                 label = self.wallet.labels.get(addr,'')
 338                 if label: label = "\"%s\""%label
 339                 b = format_satoshis(self.wallet.get_addr_balance(addr)[0])
 340                 m_addr = "%34s"%addr
 341                 print_msg(flags, m_addr, b, label)
 342