152 lines
5.2 KiB
Python
152 lines
5.2 KiB
Python
import os
|
|
import sys
|
|
from decimal import Decimal
|
|
from bitcoinrpc.authproxy import AuthServiceProxy
|
|
|
|
from lbryum.wallet import Wallet, WalletStorage
|
|
from lbryum.commands import known_commands, Commands
|
|
from lbryum.simple_config import SimpleConfig
|
|
from lbryum.blockchain import get_blockchain
|
|
from lbryum.network import Network
|
|
|
|
|
|
def get_lbrycrdd_connection_string(wallet_conf):
|
|
settings = {"username": "rpcuser",
|
|
"password": "rpcpassword",
|
|
"rpc_port": 9245}
|
|
if wallet_conf and os.path.exists(wallet_conf):
|
|
with open(wallet_conf, "r") as conf:
|
|
conf_lines = conf.readlines()
|
|
for l in conf_lines:
|
|
if l.startswith("rpcuser="):
|
|
settings["username"] = l[8:].rstrip('\n')
|
|
if l.startswith("rpcpassword="):
|
|
settings["password"] = l[12:].rstrip('\n')
|
|
if l.startswith("rpcport="):
|
|
settings["rpc_port"] = int(l[8:].rstrip('\n'))
|
|
|
|
rpc_user = settings["username"]
|
|
rpc_pass = settings["password"]
|
|
rpc_port = settings["rpc_port"]
|
|
rpc_url = "127.0.0.1"
|
|
return "http://%s:%s@%s:%i" % (rpc_user, rpc_pass, rpc_url, rpc_port)
|
|
|
|
|
|
class LBRYumWallet(object):
|
|
def __init__(self, lbryum_path):
|
|
self.config = SimpleConfig()
|
|
self.config.set_key('chain', 'lbrycrd_main')
|
|
self.storage = WalletStorage(lbryum_path)
|
|
self.wallet = Wallet(self.storage)
|
|
self.cmd_runner = Commands(self.config, self.wallet, None)
|
|
if not self.wallet.has_seed():
|
|
seed = self.wallet.make_seed()
|
|
self.wallet.add_seed(seed, "derp")
|
|
self.wallet.create_master_keys("derp")
|
|
self.wallet.create_main_account()
|
|
self.wallet.update_password("derp", "")
|
|
self.network = Network(self.config)
|
|
self.blockchain = get_blockchain(self.config, self.network)
|
|
print self.config.get('chain'), self.blockchain
|
|
self.wallet.storage.write()
|
|
|
|
def command(self, command_name, *args, **kwargs):
|
|
cmd_runner = Commands(self.config, self.wallet, None)
|
|
cmd = known_commands[command_name]
|
|
func = getattr(cmd_runner, cmd.name)
|
|
return func(*args, **kwargs)
|
|
|
|
def generate_address(self):
|
|
address = self.wallet.create_new_address()
|
|
self.wallet.storage.write()
|
|
return address
|
|
|
|
|
|
class LBRYcrd(object):
|
|
def __init__(self, lbrycrdd_path):
|
|
self.lbrycrdd_conn_str = get_lbrycrdd_connection_string(lbrycrdd_path)
|
|
|
|
def __call__(self, method, *args, **kwargs):
|
|
return self.rpc(method)(*args, **kwargs)
|
|
|
|
def rpc(self, method):
|
|
return AuthServiceProxy(self.lbrycrdd_conn_str, service_name=method)
|
|
|
|
|
|
def get_wallet_path():
|
|
cwd = os.getcwd()
|
|
wallet_path = os.path.join(cwd, "wallet.json")
|
|
if not os.path.exists(wallet_path):
|
|
return wallet_path
|
|
i = 1
|
|
while True:
|
|
wallet_path = os.path.join(cwd, "wallet_%i.json" % i)
|
|
if not os.path.exists(wallet_path):
|
|
return wallet_path
|
|
i += 1
|
|
|
|
|
|
def coin_chooser(lbrycrdd, amount, fee=0.001):
|
|
def iter_txis():
|
|
unspent = lbrycrdd("listunspent")
|
|
unspent = sorted(unspent, key=lambda x: x['amount'], reverse=True)
|
|
spendable = Decimal(0.0)
|
|
for txi in unspent:
|
|
if spendable >= amount:
|
|
break
|
|
else:
|
|
spendable += txi['amount']
|
|
yield txi
|
|
if spendable < amount:
|
|
print spendable, amount
|
|
raise Exception("Not enough funds")
|
|
|
|
coins = list(iter(iter_txis()))
|
|
total = sum(c['amount'] for c in coins)
|
|
change = Decimal(total) - Decimal(amount) - Decimal(fee)
|
|
|
|
if change < 0:
|
|
raise Exception("Not enough funds")
|
|
if change:
|
|
change_address = lbrycrdd("getnewaddress")
|
|
else:
|
|
change_address = None
|
|
|
|
print "Total: %f, amount: %f, change: %f" % (total, amount, change)
|
|
|
|
return coins, change, change_address
|
|
|
|
|
|
def get_raw_tx(lbrycrdd, addresses, coins, amount, change, change_address):
|
|
txi = [{'txid': c['txid'], 'vout': c['vout']} for c in coins]
|
|
txo = {address: float(amount) for address in addresses}
|
|
if change_address:
|
|
txo[change_address] = float(change)
|
|
return lbrycrdd("createrawtransaction", txi, txo)
|
|
|
|
|
|
def main(count, value=None, lbryum_path=None, lbrycrdd_path=None):
|
|
count = int(count)
|
|
lbryum_path = lbryum_path or get_wallet_path()
|
|
if sys.platform == "darwin":
|
|
default_lbrycrdd = os.path.join(os.path.expanduser("~"),
|
|
"Library/Application Support/lbrycrd/lbrycrd.conf")
|
|
else:
|
|
default_lbrycrdd = os.path.join(os.path.expanduser("~"), ".lbrycrd/lbrycrd.conf")
|
|
lbrycrdd_path = lbrycrdd_path or default_lbrycrdd
|
|
l = LBRYcrd(lbrycrdd_path=lbrycrdd_path)
|
|
s = LBRYumWallet(lbryum_path)
|
|
value = value or 1.0
|
|
value = Decimal(value)
|
|
|
|
coins, change, change_address = coin_chooser(l, count * value)
|
|
addresses = [s.generate_address() for i in range(count)]
|
|
raw_tx = get_raw_tx(l, addresses, coins, value, change, change_address)
|
|
signed = l("signrawtransaction", raw_tx)['hex']
|
|
txid = l("sendrawtransaction", signed)
|
|
print txid
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = sys.argv[1:]
|
|
main(*args)
|