wownero-python/monero/backends/jsonrpc.py

275 lines
11 KiB
Python
Raw Normal View History

import binascii
from datetime import datetime
import operator
import json
import logging
import pprint
import requests
from .. import exceptions
from ..account import Account
from ..address import address, Address, SubAddress
2018-01-06 22:12:42 +00:00
from ..numbers import from_atomic, to_atomic, PaymentID
2017-12-27 00:49:59 +00:00
from ..transaction import Transaction, Payment, Transfer
_log = logging.getLogger(__name__)
2018-01-14 22:51:18 +00:00
class JSONRPCDaemon(object):
def __init__(self, protocol='http', host='127.0.0.1', port=18081, path='/json_rpc'):
2018-01-15 03:12:53 +00:00
self.url = '{protocol}://{host}:{port}'.format(
2018-01-14 22:51:18 +00:00
protocol=protocol,
host=host,
port=port)
_log.debug("JSONRPC daemon backend URL: {url}".format(url=self.url))
def get_info(self):
info = self.raw_jsonrpc_request('get_info')
return info
2018-01-15 03:12:53 +00:00
def send_transaction(self, blob):
res = self.raw_request('/sendrawtransaction', {
'tx_as_hex': blob,
'do_not_relay': False})
2018-01-15 03:12:53 +00:00
if res['status'] == 'OK':
return res
raise exceptions.TransactionBroadcastError(
"{status}: {reason}".format(**res),
details=res)
def raw_request(self, path, data):
hdr = {'Content-Type': 'application/json'}
_log.debug(u"Request: {path}\nData: {data}".format(
path=path,
data=pprint.pformat(data)))
rsp = requests.post(self.url + path, headers=hdr, data=json.dumps(data))
if rsp.status_code != 200:
raise RPCError("Invalid HTTP status {code} for path {path}.".format(
code=rsp.status_code,
path=path))
result = rsp.json()
_ppresult = pprint.pformat(result)
_log.debug(u"Result:\n{result}".format(result=_ppresult))
return result
2018-01-14 22:51:18 +00:00
def raw_jsonrpc_request(self, method, params=None):
hdr = {'Content-Type': 'application/json'}
data = {'jsonrpc': '2.0', 'id': 0, 'method': method, 'params': params or {}}
_log.debug(u"Method: {method}\nParams:\n{params}".format(
method=method,
params=pprint.pformat(params)))
2018-01-15 03:12:53 +00:00
rsp = requests.post(self.url + '/json_rpc', headers=hdr, data=json.dumps(data))
2018-01-14 22:51:18 +00:00
if rsp.status_code != 200:
raise RPCError("Invalid HTTP status {code} for method {method}.".format(
code=rsp.status_code,
method=method))
result = rsp.json()
_ppresult = pprint.pformat(result)
_log.debug(u"Result:\n{result}".format(result=_ppresult))
if 'error' in result:
err = result['error']
_log.error(u"JSON RPC error:\n{result}".format(result=_ppresult))
# if err['code'] in _err2exc:
# raise _err2exc[err['code']](err['message'])
# else:
# raise RPCError(
# "Method '{method}' failed with RPC Error of unknown code {code}, "
# "message: {message}".format(method=method, data=data, result=result, **err))
raise RPCError(
"Method '{method}' failed with RPC Error of unknown code {code}, "
"message: {message}".format(method=method, data=data, result=result, **err))
#
#
#
return result['result']
2017-12-26 21:02:17 +00:00
class JSONRPCWallet(object):
2017-12-27 00:49:59 +00:00
_master_address = None
_addresses = None
2018-01-18 01:59:54 +00:00
def __init__(self, protocol='http', host='127.0.0.1', port=18088, path='/json_rpc', user='', password=''):
self.url = '{protocol}://{host}:{port}/json_rpc'.format(
protocol=protocol,
host=host,
port=port)
2018-01-14 22:51:18 +00:00
_log.debug("JSONRPC wallet backend URL: {url}".format(url=self.url))
self.user = user
self.password = password
2018-01-14 22:51:18 +00:00
_log.debug("JSONRPC wallet backend auth: '{user}'/'{stars}'".format(
user=user, stars=('*' * len(password)) if password else ''))
2018-01-14 05:28:28 +00:00
def get_view_key(self):
return self.raw_request('query_key', {'key_type': 'view_key'})['key']
def get_seed(self):
return self.raw_request('query_key', {'key_type': 'mnemonic'})['key']
def get_accounts(self):
accounts = []
try:
_accounts = self.raw_request('get_accounts')
except MethodNotFound:
2017-12-27 00:49:59 +00:00
# monero <= 0.11 : there's only one account and one address
2018-01-17 14:03:18 +00:00
_log.debug('Monero <= 0.11 found, no accounts')
2017-12-27 00:49:59 +00:00
self._master_address = self.get_addresses()[0]
return [Account(self, 0)]
idx = 0
2017-12-27 00:49:59 +00:00
self._master_address = Address(_accounts['subaddress_accounts'][0]['base_address'])
for _acc in _accounts['subaddress_accounts']:
assert idx == _acc['account_index']
accounts.append(Account(self, _acc['account_index']))
idx += 1
return accounts
def new_account(self, label=None):
_account = self.raw_request('create_account', {'label': label})
return Account(self, _account['account_index']), SubAddress(_account['address'])
def get_addresses(self, account=0):
_addresses = self.raw_request('getaddress', {'account_index': account})
if 'addresses' not in _addresses:
# monero <= 0.11
2018-01-17 14:03:18 +00:00
_log.debug('Monero <= 0.11 found, assuming single address')
return [Address(_addresses['address'])]
addresses = [None] * (max(map(operator.itemgetter('address_index'), _addresses['addresses'])) + 1)
for _addr in _addresses['addresses']:
addresses[_addr['address_index']] = address(
_addr['address'],
label=_addr.get('label', None))
return addresses
def new_address(self, account=0, label=None):
_address = self.raw_request(
'create_address', {'account_index': account, 'label': label})
return SubAddress(_address['address'])
def get_balances(self, account=0):
_balance = self.raw_request('getbalance', {'account_index': account})
return (from_atomic(_balance['balance']), from_atomic(_balance['unlocked_balance']))
2017-12-27 00:49:59 +00:00
def get_payments(self, account=0, payment_id=0):
2018-01-06 22:12:42 +00:00
payment_id = PaymentID(payment_id)
2017-12-27 00:49:59 +00:00
_log.debug("Getting payments for account {acc}, payment_id {pid}".format(
acc=account, pid=payment_id))
_payments = self.raw_request('get_payments', {
'account_index': account,
2018-01-06 22:12:42 +00:00
'payment_id': str(payment_id)})
2017-12-27 00:49:59 +00:00
pmts = []
for tx in _payments['payments']:
data = self._tx2dict(tx)
# Monero <= 0.11 : no address is passed because there's only one
2018-01-06 15:44:38 +00:00
data['local_address'] = data['local_address'] or self._master_address
2017-12-27 00:49:59 +00:00
pmts.append(Payment(**data))
return pmts
def get_transactions_in(self, account=0):
_transfers = self.raw_request('get_transfers',
{'account_index': account, 'in': True, 'out': False, 'pool': False})
return [Payment(**self._tx2dict(tx)) for tx in
2018-01-06 22:25:41 +00:00
sorted(_transfers.get('in', []), key=operator.itemgetter('timestamp'))]
2017-12-27 00:49:59 +00:00
def get_transactions_out(self, account=0):
_transfers = self.raw_request('get_transfers',
{'account_index': account, 'in': False, 'out': True, 'pool': False})
return [Transfer(**self._tx2dict(tx)) for tx in
2018-01-06 22:25:41 +00:00
sorted(_transfers.get('out', []), key=operator.itemgetter('timestamp'))]
2017-11-29 03:38:29 +00:00
2017-12-27 00:49:59 +00:00
def _tx2dict(self, tx):
pid = tx.get('payment_id', None)
2017-11-29 03:38:29 +00:00
return {
2017-12-27 00:49:59 +00:00
'hash': tx.get('txid', tx.get('tx_hash')),
'timestamp': datetime.fromtimestamp(tx['timestamp']) if 'timestamp' in tx else None,
'amount': from_atomic(tx['amount']),
'fee': from_atomic(tx['fee']) if 'fee' in tx else None,
'height': tx.get('height', tx.get('block_height')),
'payment_id': None if pid is None else PaymentID(pid),
2017-12-27 00:49:59 +00:00
'note': tx.get('note'),
# NOTE: address will be resolved only after PR#3010 has been merged to Monero
2018-01-06 15:44:38 +00:00
'local_address': address(tx['address']) if 'address' in tx else None,
2017-12-27 00:49:59 +00:00
'key': tx.get('key'),
'blob': tx.get('blob', None),
2017-11-29 03:38:29 +00:00
}
def transfer(self, destinations, priority, ringsize,
payment_id=None, unlock_time=0, account=0,
relay=True):
2017-11-29 03:38:29 +00:00
data = {
'account_index': account,
'destinations': list(map(
2017-11-30 02:43:34 +00:00
lambda dst: {'address': str(address(dst[0])), 'amount': to_atomic(dst[1])},
2017-11-29 03:38:29 +00:00
destinations)),
'mixin': ringsize - 1,
2017-11-29 03:38:29 +00:00
'priority': priority,
'unlock_time': 0,
'get_tx_keys': True,
'get_tx_hex': True,
'new_algorithm': True,
'do_not_relay': not relay,
2017-11-29 03:38:29 +00:00
}
if payment_id is not None:
data['payment_id'] = str(PaymentID(payment_id))
2017-11-29 03:38:29 +00:00
_transfers = self.raw_request('transfer_split', data)
2018-01-07 00:26:50 +00:00
_pertx = [dict(_tx) for _tx in map(
lambda vs: zip(('txid', 'amount', 'fee', 'key', 'blob', 'payment_id'), vs),
zip(*[_transfers[k] for k in (
'tx_hash_list', 'amount_list', 'fee_list', 'tx_key_list', 'tx_blob_list')]))]
for d in _pertx:
d['payment_id'] = payment_id
return [Transfer(**self._tx2dict(tx)) for tx in _pertx]
def raw_request(self, method, params=None):
hdr = {'Content-Type': 'application/json'}
data = {'jsonrpc': '2.0', 'id': 0, 'method': method, 'params': params or {}}
_log.debug(u"Method: {method}\nParams:\n{params}".format(
method=method,
params=pprint.pformat(params)))
auth = requests.auth.HTTPDigestAuth(self.user, self.password)
rsp = requests.post(self.url, headers=hdr, data=json.dumps(data), auth=auth)
if rsp.status_code == 401:
raise Unauthorized("401 Unauthorized. Invalid RPC user name or password.")
elif rsp.status_code != 200:
raise RPCError("Invalid HTTP status {code} for method {method}.".format(
code=rsp.status_code,
method=method))
result = rsp.json()
_ppresult = pprint.pformat(result)
_log.debug(u"Result:\n{result}".format(result=_ppresult))
if 'error' in result:
err = result['error']
_log.error(u"JSON RPC error:\n{result}".format(result=_ppresult))
if err['code'] in _err2exc:
2017-11-29 03:38:29 +00:00
raise _err2exc[err['code']](err['message'])
else:
raise RPCError(
"Method '{method}' failed with RPC Error of unknown code {code}, "
"message: {message}".format(method=method, data=data, result=result, **err))
return result['result']
2017-11-29 03:38:29 +00:00
class RPCError(exceptions.BackendException):
pass
class Unauthorized(RPCError):
pass
class MethodNotFound(RPCError):
pass
_err2exc = {
2017-11-29 03:38:29 +00:00
-2: exceptions.WrongAddress,
-4: exceptions.NotEnoughUnlockedMoney,
2018-01-11 02:18:22 +00:00
-5: exceptions.WrongPaymentId,
2018-01-11 05:14:34 +00:00
-16: exceptions.TransactionNotPossible,
2017-11-29 03:38:29 +00:00
-20: exceptions.AmountIsZero,
-32601: MethodNotFound,
}