import requests import six import json import operator from tipbot import config from decimal import Decimal PICOWOW = Decimal('0.00000000001') class Wallet(object): def __init__(self): self.endpoint = f'http://127.0.0.1:9999/json_rpc' self.auth = requests.auth.HTTPDigestAuth( config.WALLET_RPC_USER, config.WALLET_RPC_PASS ) try: r = self.height() height = r['height'] self.connected = True except: self.connected = False def make_wallet_rpc(self, method, params={}): r = requests.get( self.endpoint, data=json.dumps({'method': method, 'params': params}), auth=self.auth ) if 'error' in r.json(): return r.json()['error'] else: return r.json()['result'] def height(self): return self.make_wallet_rpc('get_height', {}) def spend_key(self): return self.make_wallet_rpc('query_key', {'key_type': 'spend_key'})['key'] def view_key(self): return self.make_wallet_rpc('query_key', {'key_type': 'view_key'})['key'] def seed(self): return self.make_wallet_rpc('query_key', {'key_type': 'mnemonic'})['key'] def accounts(self): _accounts = self.make_wallet_rpc('get_accounts') return [(i['account_index'], i['base_address']) for i in _accounts['subaddress_accounts']] def new_account(self, label=None): _account = self.make_wallet_rpc('create_account', {'label': label}) return (_account['account_index'], _account['address']) def addresses(self, account, addr_indices=None): qdata = {'account_index': account} if addr_indices: qdata['address_index'] = addr_indices _addresses = self.make_wallet_rpc('get_address', qdata) addresses = [None] * (max(map(operator.itemgetter('address_index'), _addresses['addresses'])) + 1) for _addr in _addresses['addresses']: addresses[_addr['address_index']] = _addr['address'] return addresses def new_address(self, account, label=None): data = {'account_index': account, 'label': label} _address = self.make_wallet_rpc('create_address', data) return (_address['address_index'], _address['address']) def balances(self, account): data = {'account_index': account} _balance = self.make_wallet_rpc('get_balance', data) return (from_atomic(_balance['balance']), from_atomic(_balance['unlocked_balance'])) def transfer(self, dest_address, amount, priority, account): data = { 'account_index': account, 'destinations': [{'address': dest_address, 'amount': to_atomic(amount)}], 'priority': priority, 'unlock_time': 0, 'get_tx_key': True, 'get_tx_hex': True, 'new_algorithm': True, 'do_not_relay': False, 'ring_size': 22 } transfer = self.make_wallet_rpc('transfer', data) return transfer def to_atomic(amount): if not isinstance(amount, (Decimal, float) + six.integer_types): raise ValueError("Amount '{}' doesn't have numeric type. Only Decimal, int, long and " "float (not recommended) are accepted as amounts.") return int(amount * 10**11) def from_atomic(amount): return (Decimal(amount) * PICOWOW).quantize(PICOWOW) def as_wownero(amount): return Decimal(amount).quantize(PICOWOW)