return python objects from commands, and display them as json

This commit is contained in:
thomasv 2013-02-26 17:57:48 +01:00
parent 8a14679530
commit 2e9c2c1f96
5 changed files with 50 additions and 29 deletions

View File

@ -488,7 +488,9 @@ if __name__ == '__main__':
func = eval('cmd_runner.' + cmd)
if password:
cmd_runner.password = password
func(*args[1:])
result = func(*args[1:])
util.print_json(result)
if cmd not in offline_commands and not options.offline:

View File

@ -727,6 +727,15 @@ class Transaction:
return is_send, v, fee
def as_dict(self):
import json
out = {
"hex":self.raw,
"complete":self.is_complete
}
if not self.is_complete:
out['input_info'] = repr(self.input_info).replace(' ','')
return out
def test_bip32():

View File

@ -100,26 +100,30 @@ protected_commands = ['payto', 'password', 'mktx', 'get_seed', 'importprivkey','
class Commands:
def __init__(self, wallet, interface):
def __init__(self, wallet, interface, callback = None):
self.wallet = wallet
self.interface = interface
self.callback = callback
def _run(self, method, args, password_getter):
if method in protected_commands:
self.password = apply(password_getter,())
f = eval('self.'+method)
apply(f,args)
result = apply(f,args)
self.password = None
if self.callback:
apply(self.callback, ())
return result
def get_history(self, addr):
h = self.wallet.get_history(addr)
if h is None: h = self.wallet.interface.synchronous_get([ ('blockchain.address.get_history',[addr]) ])[0]
print_json(h)
return h
def listunspent(self):
l = self.wallet.get_unspent_coins()
for i in l: i["value"] = i["value"]*1e-8
print_json(l)
return l
def createrawtransaction(self, inputs, outputs):
# convert to own format
@ -128,7 +132,7 @@ class Commands:
i['index'] = i['vout']
outputs = map(lambda x: (x[0],int(1e8*x[1])), outputs.items())
tx = Transaction.from_io(inputs, outputs)
print_msg( tx )
return tx.as_dict()
def signrawtransaction(self, raw_tx, input_info, private_keys):
tx = Transaction(raw_tx)
@ -181,39 +185,38 @@ class Commands:
txin['address'] = addr
tx.sign( private_keys )
print_json({ "hex":str(tx),"complete":tx.is_complete})
return tx.as_dict()
def decoderawtransaction(self, raw):
tx = Transaction(raw)
print_json( tx.deserialize() )
return tx.deserialize()
def sendrawtransaction(self, raw):
tx = Transaction(raw)
r, h = wallet.sendtx( tx )
print_msg(h)
return h
def createmultisig(self, num, pubkeys):
assert isinstance(pubkeys, list)
print_json( Transaction.multisig_script(pubkeys, num) )
return Transaction.multisig_script(pubkeys, num)
def freeze(self,addr):
print_msg(self.wallet.freeze(addr))
return self.wallet.freeze(addr)
def unfreeze(self,addr):
print_msg(self.wallet.unfreeze(addr))
return self.wallet.unfreeze(addr)
def prioritize(self, addr):
print_msg(self.wallet.prioritize(addr))
return self.wallet.prioritize(addr)
def unprioritize(self, addr):
print_msg(self.wallet.unprioritize(addr))
return self.wallet.unprioritize(addr)
def dumpprivkey(self, addr):
print_msg( self.wallet.get_private_key(addr, self.password) )
return self.wallet.get_private_key(addr, self.password)
def dumpprivkeys(self, addresses):
print_json( self.wallet.get_private_keys(addresses, self.password) )
return self.wallet.get_private_keys(addresses, self.password)
def validateaddress(self,addr):
is_valid = self.wallet.is_valid(addr)
@ -225,7 +228,7 @@ class Commands:
if is_mine:
out['pubkey'] = self.wallet.get_public_key(addr)
print_json(out)
return out
def balance(self, addresses = []):
@ -259,16 +262,16 @@ class Commands:
def sign_message(self, address, message):
print_msg(self.wallet.sign_message(address, message, self.password))
return self.wallet.sign_message(address, message, self.password)
def verify_message(self, address, signature, message):
try:
EC_KEY.verify_message(address, signature, message)
print_msg(True)
return True
except BaseException as e:
print_error("Verification error: {0}".format(e))
print_msg(False)
return False
def _mktx(self, to_address, amount, fee = None, change_addr = None, from_addr = None):
@ -287,16 +290,13 @@ class Commands:
def mktx(self, to_address, amount, fee = None, change_addr = None, from_addr = None):
tx = self._mktx(to_address, amount, fee, change_addr, from_addr)
out = {"hex":str(tx), "complete":tx.is_complete}
if not tx.is_complete:
out['input_info'] = repr(tx.input_info).replace(' ','')
print_json(out)
return tx.as_dict()
def payto(self, to_address, amount, fee = None, change_addr = None, from_addr = None):
tx = self._mktx(to_address, amount, fee, change_addr, from_addr)
r, h = wallet.sendtx( tx )
print_msg(h)
return h
def history(self):
@ -326,7 +326,7 @@ class Commands:
c = {}
for addr in self.wallet.addressbook:
c[addr] = self.wallet.labels.get(addr)
print_json(c)
return c
def addresses(self, show_all):

View File

@ -1270,7 +1270,7 @@ class ElectrumWindow(QMainWindow):
console.updateNamespace({'wallet' : self.wallet, 'interface' : self.wallet.interface, 'gui':self})
console.updateNamespace({'util' : util, 'bitcoin':bitcoin})
c = commands.Commands(self.wallet, self.wallet.interface)
c = commands.Commands(self.wallet, self.wallet.interface, lambda: self.console.set_json(True))
methods = {}
def mkfunc(f, method):
return lambda *args: apply( f, (method, args, self.password_dialog ))

View File

@ -4,6 +4,7 @@ import sys, os, re
import traceback
from PyQt4 import QtCore
from PyQt4 import QtGui
import util
@ -23,6 +24,10 @@ class Console(QtGui.QPlainTextEdit):
self.showMessage(startup_message)
self.updateNamespace({'run':self.run_script})
self.set_json(False)
def set_json(self, b):
self.is_json = b
def run_script(self, filename):
with open(filename) as f:
@ -187,7 +192,10 @@ class Console(QtGui.QPlainTextEdit):
try:
result = eval(command, self.namespace, self.namespace)
if result != None:
self.appendPlainText(repr(result))
if self.is_json:
util.print_json(result)
else:
self.appendPlainText(repr(result))
except SyntaxError:
exec command in self.namespace
except SystemExit:
@ -200,6 +208,8 @@ class Console(QtGui.QPlainTextEdit):
self.appendPlainText('\n'.join(traceback_lines))
sys.stdout = tmp_stdout
self.newPrompt()
self.set_json(False)
def keyPressEvent(self, event):
if event.key() == QtCore.Qt.Key_Tab: