Add draft of Account, Wallet and JSONRPC backend

monero/backends/jsonrpc.py

@@ -0,0 +1,131 @@
+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
+from ..numbers import from_atomic, to_atomic
+
+_log = logging.getLogger(__name__)
+
+
+class JSONRPC(object):
+    def __init__(self, protocol='http', host='127.0.0.1', port=18082, path='/json_rpc', user='', password=''):
+        self.url = '{protocol}://{host}:{port}/json_rpc'.format(
+                protocol=protocol,
+                host=host,
+                port=port)
+        _log.debug("JSONRPC backend URL: {url}".format(url=self.url))
+        self.user = user
+        self.password = password
+        _log.debug("JSONRPC backend auth: '{user}'/'{stars}'".format(
+            user=user, stars=('*' * len(password)) if password else ''))
+
+    def get_accounts(self):
+        accounts = []
+        try:
+            _accounts = self.raw_request('get_accounts')
+        except MethodNotFound:
+            # monero <= 0.11
+            return [Account(self, 0)]
+        idx = 0
+        for _acc in _accounts['subaddress_accounts']:
+            assert idx == _acc['account_index']
+            accounts.append(Account(self, _acc['account_index']))
+            idx += 1
+        return accounts
+
+    def get_addresses(self, account=0):
+        _addresses = self.raw_request('getaddress', {'account_index': account})
+        if 'addresses' not in _addresses:
+            # monero <= 0.11
+            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'])
+        return addresses
+
+    def get_balance(self, account=0):
+        _balance = self.raw_request('getbalance', {'account_index': account})
+        return (from_atomic(_balance['balance']), from_atomic(_balance['unlocked_balance']))
+
+    def get_payments_in(self, account=0):
+        _payments = self.raw_request('get_transfers',
+                {'account_index': account, 'in': True, 'out': False, 'pool': False})
+        return map(self._pythonify_tx, _payments.get('in', []))
+
+    def get_payments_out(self, account=0):
+        _payments = self.raw_request('get_transfers',
+                {'account_index': account, 'in': False, 'out': True, 'pool': False})
+        return map(self._pythonify_tx, _payments.get('out', ''))
+
+    def _pythonify_tx(self, tx):
+        return {
+            'id': tx['txid'],
+            'when': datetime.fromtimestamp(tx['timestamp']),
+            'amount': from_atomic(tx['amount']),
+            'fee': from_atomic(tx['fee']),
+            'height': tx['height'],
+            'payment_id': tx['payment_id'],
+            'note': tx['note']
+        }
+
+    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']
+            # TODO: resolve code, raise exception
+            _log.error(u"JSON RPC error:\n{result}".format(result=_ppresult))
+            if err['code'] in _err2exc:
+                raise _err2exc[err['code']](err['message'], method=method, data=data, result=result)
+            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']
+
+
+class RPCError(exceptions.MoneroException):
+    def __init__(self, message, method=None, data=None, result=None):
+        self.method = method
+        self.data = data
+        self.result = result
+        super().__init__(message)
+
+    def __str__(self):
+        return "'{method}': {error}".format(
+            method=self.method,
+            error=super().__str__())
+
+
+class Unauthorized(RPCError):
+    pass
+
+
+class MethodNotFound(RPCError):
+    pass
+
+
+_err2exc = {
+    -32601: MethodNotFound,
+}