LBRYFoundation/lbry-sdk
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/lbry-sdk.git synced 2025-09-02 02:05:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added --account to most commands, also claim_list_mine, transaction_list, address_is_mine and other cleanup

Browse source
This commit is contained in:
Lex Berezhny 2018-09-21 09:47:06 -04:00
parent 0adcaa6a5d
commit 53d4c149bc
4 changed files with 110 additions and 127 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

182
lbrynet/daemon/Daemon.py
View file

@ -1016,7 +1016,7 @@ class Daemon(AuthJSONRPCServer):
def jsonrpc_wallet_list(self): def jsonrpc_wallet_list(self):
pass pass
@AuthJSONRPCServer.deprecated("address_generate") @AuthJSONRPCServer.deprecated("address_unused")
def jsonrpc_wallet_new_address(self): def jsonrpc_wallet_new_address(self):
pass pass
@ -1026,7 +1026,7 @@ class Daemon(AuthJSONRPCServer):
@requires(WALLET_COMPONENT, conditions=[WALLET_IS_UNLOCKED]) @requires(WALLET_COMPONENT, conditions=[WALLET_IS_UNLOCKED])
@defer.inlineCallbacks @defer.inlineCallbacks
def jsonrpc_wallet_send(self, amount, address=None, claim_id=None): def jsonrpc_wallet_send(self, amount, address=None, claim_id=None, account_id=None):
""" """
Send credits. If given an address, send credits to it. If given a claim id, send a tip Send credits. If given an address, send credits to it. If given a claim id, send a tip
to the owner of a claim specified by uri. A tip is a claim support where the recipient to the owner of a claim specified by uri. A tip is a claim support where the recipient
@ -1035,11 +1035,13 @@ class Daemon(AuthJSONRPCServer):
Usage: Usage:
wallet_send (<amount> | --amount=<amount>) wallet_send (<amount> | --amount=<amount>)
((<address> | --address=<address>) | (<claim_id> | --claim_id=<claim_id>)) ((<address> | --address=<address>) | (<claim_id> | --claim_id=<claim_id>))
[--account=<account_id>]
Options: Options:
--amount=<amount> : (Decimal) amount of credit to send --amount=<amount> : (Decimal) amount of credit to send
--address=<address> : (str) address to send credits to --address=<address> : (str) address to send credits to
--claim_id=<claim_id> : (str) claim_id of the claim to send to tip to --claim_id=<claim_id> : (str) claim_id of the claim to send to tip to
--account=<account_id> : (str) account to fund the transaction
Returns: Returns:
If sending to an address: If sending to an address:
@ -1081,11 +1083,12 @@ class Daemon(AuthJSONRPCServer):
reserved_points = self.wallet_manager.reserve_points(address, amount) reserved_points = self.wallet_manager.reserve_points(address, amount)
if reserved_points is None: if reserved_points is None:
raise InsufficientFundsError() raise InsufficientFundsError()
result = yield self.wallet_manager.send_points_to_address(reserved_points, amount) account = self.get_account_or_default(account_id)
result = yield self.wallet_manager.send_points_to_address(reserved_points, amount, account)
self.analytics_manager.send_credits_sent() self.analytics_manager.send_credits_sent()
else: else:
log.info("This command is deprecated for sending tips, please use the newer claim_tip command") log.info("This command is deprecated for sending tips, please use the newer claim_tip command")
result = yield self.jsonrpc_claim_tip(claim_id, amount) result = yield self.jsonrpc_claim_tip(claim_id=claim_id, amount=amount, account_id=account_id)
return result return result
@requires(WALLET_COMPONENT, conditions=[WALLET_IS_UNLOCKED]) @requires(WALLET_COMPONENT, conditions=[WALLET_IS_UNLOCKED])
@ -1110,8 +1113,8 @@ class Daemon(AuthJSONRPCServer):
""" """
broadcast = not no_broadcast broadcast = not no_broadcast
return self.jsonrpc_account_fund( return self.jsonrpc_account_fund(
self.default_account.name, self.default_account.id,
self.default_account.name, self.default_account.id,
amount=amount, amount=amount,
outputs=num_addresses, outputs=num_addresses,
broadcast=broadcast broadcast=broadcast
@ -1140,7 +1143,7 @@ class Daemon(AuthJSONRPCServer):
(map) balance of account(s) (map) balance of account(s)
""" """
if account_id: if account_id:
account = self.get_account_or_error('account', account_id) account = self.get_account_or_error(account_id)
args = { args = {
'confirmations': confirmations, 'confirmations': confirmations,
'include_reserved': include_reserved, 'include_reserved': include_reserved,
@ -1182,10 +1185,7 @@ class Daemon(AuthJSONRPCServer):
""" """
if address is not None: if address is not None:
raise NotImplementedError("Limiting by address needs to be re-implemented in new wallet.") raise NotImplementedError("Limiting by address needs to be re-implemented in new wallet.")
if account_id is not None: account = self.get_account_or_default(account_id)
account = self.get_account_or_error('account', account_id)
else:
account = self.default_account
dewies = yield account.get_balance( dewies = yield account.get_balance(
0 if include_unconfirmed else 6 0 if include_unconfirmed else 6
) )
@ -1281,16 +1281,16 @@ class Daemon(AuthJSONRPCServer):
Remove an existing account. Remove an existing account.
Usage: Usage:
account (<account_id> | --account_id=<account_id>) account (<account_id> | --account=<account_id>)
Options: Options:
--account_id=<account_id> : (str) id of the account to remove --account=<account_id> : (str) id of the account to remove
Returns: Returns:
(map) details of removed account (map) details of removed account
""" """
account = self.get_account_or_error('account_id', account_id) account = self.get_account_or_error(account_id)
self.default_wallet.accounts.remove(account) self.default_wallet.accounts.remove(account)
self.default_wallet.save() self.default_wallet.save()
result = account.to_dict() result = account.to_dict()
@ -1307,13 +1307,13 @@ class Daemon(AuthJSONRPCServer):
Change various settings on an account. Change various settings on an account.
Usage: Usage:
account (<account_id> | --account_id=<account_id>) account (<account_id> | --account=<account_id>)
[--default] [--new_name=<new_name>] [--default] [--new_name=<new_name>]
[--change_gap=<change_gap>] [--change_max_uses=<change_max_uses>] [--change_gap=<change_gap>] [--change_max_uses=<change_max_uses>]
[--receiving_gap=<receiving_gap>] [--receiving_max_uses=<receiving_max_uses>] [--receiving_gap=<receiving_gap>] [--receiving_max_uses=<receiving_max_uses>]
Options: Options:
--account_id=<account_id> : (str) id of the account to change --account=<account_id> : (str) id of the account to change
--default : (bool) make this account the default --default : (bool) make this account the default
--new_name=<new_name> : (str) new name for the account --new_name=<new_name> : (str) new name for the account
--receiving_gap=<receiving_gap> : (int) set the gap for receiving addresses --receiving_gap=<receiving_gap> : (int) set the gap for receiving addresses
@ -1327,7 +1327,7 @@ class Daemon(AuthJSONRPCServer):
(map) updated account details (map) updated account details
""" """
account = self.get_account_or_error('account_id', account_id) account = self.get_account_or_error(account_id)
change_made = False change_made = False
if account.receiving.name == HierarchicalDeterministic.name: if account.receiving.name == HierarchicalDeterministic.name:
@ -1421,7 +1421,7 @@ class Daemon(AuthJSONRPCServer):
return self.wallet_manager.encrypt_wallet(new_password) return self.wallet_manager.encrypt_wallet(new_password)
@requires("wallet") @requires("wallet")
def jsonrpc_account_max_address_gap(self, account_name): def jsonrpc_account_max_address_gap(self, account_id):
""" """
Finds ranges of consecutive addresses that are unused and returns the length Finds ranges of consecutive addresses that are unused and returns the length
of the longest such range: for change and receiving address chains. This is of the longest such range: for change and receiving address chains. This is
@ -1429,15 +1429,15 @@ class Daemon(AuthJSONRPCServer):
account settings. account settings.
Usage: Usage:
account_max_address_gap (<account_name> | --account=<account_name>) account_max_address_gap (<account_id> | --account=<account_id>)
Options: Options:
--account=<account_name> : (str) account for which to get max gaps --account=<account_id> : (str) account for which to get max gaps
Returns: Returns:
(map) maximum gap for change and receiving addresses (map) maximum gap for change and receiving addresses
""" """
return self.get_account_or_error('account', account_name).get_max_gap() return self.get_account_or_error(account_id).get_max_gap()
@requires("wallet") @requires("wallet")
def jsonrpc_account_fund(self, to_account, from_account, amount=0, def jsonrpc_account_fund(self, to_account, from_account, amount=0,
@ -1467,8 +1467,8 @@ class Daemon(AuthJSONRPCServer):
(map) transaction performing requested action (map) transaction performing requested action
""" """
to_account = self.get_account_or_error('to_account', to_account) to_account = self.get_account_or_error(to_account, 'to_account')
from_account = self.get_account_or_error('from_account', from_account) from_account = self.get_account_or_error(from_account, 'from_account')
amount = self.get_dewies_or_error('amount', amount) if amount else None amount = self.get_dewies_or_error('amount', amount) if amount else None
if not isinstance(outputs, int): if not isinstance(outputs, int):
raise ValueError("--outputs must be an integer.") raise ValueError("--outputs must be an integer.")
@ -1480,20 +1480,24 @@ class Daemon(AuthJSONRPCServer):
) )
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_address_is_mine(self, address): def jsonrpc_address_is_mine(self, address, account_id=None):
""" """
Checks if an address is associated with the current wallet. Checks if an address is associated with the current wallet.
Usage: Usage:
wallet_is_address_mine (<address> | --address=<address>) wallet_is_address_mine (<address> | --address=<address>)
[<account_id> | --account=<account_id>]
Options: Options:
--address=<address> : (str) address to check --address=<address> : (str) address to check
--account=<account_id> : (str) id of the account to use
Returns: Returns:
(bool) true, if address is associated with current wallet (bool) true, if address is associated with current wallet
""" """
return self.wallet_manager.address_is_mine(address) return self.wallet_manager.address_is_mine(
address, self.get_account_or_default(account_id)
)
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_address_public_key(self, address): def jsonrpc_address_public_key(self, address):
@ -1513,53 +1517,37 @@ class Daemon(AuthJSONRPCServer):
return self.wallet_manager.get_pub_keys(address) return self.wallet_manager.get_pub_keys(address)
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_address_list(self): def jsonrpc_address_list(self, account_id=None):
""" """
List wallet addresses List account addresses
Usage: Usage:
wallet_list address_list [<account_id> | --account=<account_id>]
Options: Options:
None --account=<account_id> : (str) id of the account to use
Returns: Returns:
List of wallet addresses List of wallet addresses
""" """
return self.wallet_manager.list_addresses() return self.get_account_or_default(account_id).get_addresses()
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_address_generate(self): def jsonrpc_address_unused(self, account_id=None):
"""
Generate a new wallet address
Usage:
wallet_new_address
Options:
None
Returns:
(str) New wallet address in base58
"""
return self.wallet_manager.get_new_address()
@requires(WALLET_COMPONENT)
def jsonrpc_address_unused(self):
""" """
Return an address containing no balance, will create Return an address containing no balance, will create
a new address if there is none. a new address if there is none.
Usage: Usage:
wallet_unused_address address_unused [--account=<account_id>]
Options: Options:
None --account=<account_id> : (str) id of the account to use
Returns: Returns:
(str) Unused wallet address in base58 (str) Unused wallet address in base58
""" """
return self.wallet_manager.get_unused_address() return self.get_account_or_default(account_id).receiving.get_or_create_usable_address()
@requires(FILE_MANAGER_COMPONENT) @requires(FILE_MANAGER_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2352,9 +2340,7 @@ class Daemon(AuthJSONRPCServer):
txid : (str) txid of resulting transaction txid : (str) txid of resulting transaction
} }
""" """
account = self.default_account account = self.get_account_or_default(account_id)
if account_id is not None:
account = self.get_account_or_error("account_id", account_id, lbc_only=True)
if claim_id is None and txid is None and nout is None: if claim_id is None and txid is None and nout is None:
raise Exception('Must specify claim_id, or txid and nout') raise Exception('Must specify claim_id, or txid and nout')
@ -2378,13 +2364,13 @@ class Daemon(AuthJSONRPCServer):
Usage: Usage:
claim_new_support (<name> | --name=<name>) (<claim_id> | --claim_id=<claim_id>) claim_new_support (<name> | --name=<name>) (<claim_id> | --claim_id=<claim_id>)
(<amount> | --amount=<amount>) [--account_id=<account_id>] (<amount> | --amount=<amount>) [--account=<account_id>]
Options: Options:
--name=<name> : (str) name of the claim to support --name=<name> : (str) name of the claim to support
--claim_id=<claim_id> : (str) claim_id of the claim to support --claim_id=<claim_id> : (str) claim_id of the claim to support
--amount=<amount> : (float) amount of support --amount=<amount> : (float) amount of support
--account_id=<account_id> : (str) id of the account to use --account=<account_id> : (str) id of the account to use
Returns: Returns:
(dict) Dictionary containing the transaction information (dict) Dictionary containing the transaction information
@ -2398,11 +2384,7 @@ class Daemon(AuthJSONRPCServer):
"txid": (str) txid of the transaction, "txid": (str) txid of the transaction,
} }
""" """
account = self.get_account_or_default(account_id)
account = self.default_account
if account_id is not None:
account = self.get_account_or_error("account_id", account_id, lbc_only=True)
amount = self.get_dewies_or_error("amount", amount) amount = self.get_dewies_or_error("amount", amount)
result = yield self.wallet_manager.support_claim(name, claim_id, amount, account) result = yield self.wallet_manager.support_claim(name, claim_id, amount, account)
self.analytics_manager.send_claim_action('new_support') self.analytics_manager.send_claim_action('new_support')
@ -2415,12 +2397,12 @@ class Daemon(AuthJSONRPCServer):
Tip the owner of the claim Tip the owner of the claim
Usage: Usage:
claim_tip (<claim_id> | --claim_id=<claim_id>) (<amount> | --amount=<amount>) [--account_id=<account_id>] claim_tip (<claim_id> | --claim_id=<claim_id>) (<amount> | --amount=<amount>) [--account=<account_id>]
Options: Options:
--claim_id=<claim_id> : (str) claim_id of the claim to support --claim_id=<claim_id> : (str) claim_id of the claim to support
--amount=<amount> : (float) amount of support --amount=<amount> : (float) amount of support
--account_id=<account_id> : (str) id of the account to use --account=<account_id> : (str) id of the account to use
Returns: Returns:
(dict) Dictionary containing the transaction information (dict) Dictionary containing the transaction information
@ -2434,11 +2416,7 @@ class Daemon(AuthJSONRPCServer):
"txid": (str) txid of the transaction, "txid": (str) txid of the transaction,
} }
""" """
account = self.get_account_or_default(account_id)
account = self.default_account
if account_id is not None:
account = self.get_account_or_error("account_id", account_id, lbc_only=True)
amount = self.get_dewies_or_error("amount", amount) amount = self.get_dewies_or_error("amount", amount)
validate_claim_id(claim_id) validate_claim_id(claim_id)
result = yield self.wallet_manager.tip_claim(amount, claim_id, account) result = yield self.wallet_manager.tip_claim(amount, claim_id, account)
@ -2486,7 +2464,7 @@ class Daemon(AuthJSONRPCServer):
else: else:
height = int(height) height = int(height)
result = yield self.wallet_manager.claim_renew_all_before_expiration(height) result = yield self.wallet_manager.claim_renew_all_before_expiration(height)
defer.returnValue(result) return result
@requires(WALLET_COMPONENT, conditions=[WALLET_IS_UNLOCKED]) @requires(WALLET_COMPONENT, conditions=[WALLET_IS_UNLOCKED])
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2518,19 +2496,19 @@ class Daemon(AuthJSONRPCServer):
""" """
result = yield self.wallet_manager.send_claim_to_address(claim_id, address, amount) result = yield self.wallet_manager.send_claim_to_address(claim_id, address, amount)
response = yield self._render_response(result) response = yield self._render_response(result)
defer.returnValue(response) return response
# TODO: claim_list_mine should be merged into claim_list, but idk how to authenticate it -Grin # TODO: claim_list_mine should be merged into claim_list, but idk how to authenticate it -Grin
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_claim_list_mine(self): def jsonrpc_claim_list_mine(self, account_id=None):
""" """
List my name claims List my name claims
Usage: Usage:
claim_list_mine claim_list_mine [<account_id> | --account=<account_id>]
Options: Options:
None --account=<account_id> : (str) id of the account to query
Returns: Returns:
(list) List of name claims owned by user (list) List of name claims owned by user
@ -2554,10 +2532,7 @@ class Daemon(AuthJSONRPCServer):
}, },
] ]
""" """
return self.get_account_or_default(account_id).get_claims()
d = self.wallet_manager.get_name_claims()
d.addCallback(lambda claims: self._render_response(claims))
return d
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2593,10 +2568,9 @@ class Daemon(AuthJSONRPCServer):
'last_takeover_height': (int) the height of last takeover for the name 'last_takeover_height': (int) the height of last takeover for the name
} }
""" """
claims = yield self.wallet_manager.get_claims_for_name(name) # type: dict claims = yield self.wallet_manager.get_claims_for_name(name) # type: dict
sort_claim_results(claims['claims']) sort_claim_results(claims['claims'])
defer.returnValue(claims) return claims
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2687,18 +2661,18 @@ class Daemon(AuthJSONRPCServer):
results[u]['claims_in_channel'] = resolved[u].get('claims_in_channel', []) results[u]['claims_in_channel'] = resolved[u].get('claims_in_channel', [])
response = yield self._render_response(results) response = yield self._render_response(results)
defer.returnValue(response) return response
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_transaction_list(self): def jsonrpc_transaction_list(self, account_id=None):
""" """
List transactions belonging to wallet List transactions belonging to wallet
Usage: Usage:
transaction_list transaction_list [<account_id> | --account=<account_id>]
Options: Options:
None --account=<account_id> : (str) id of the account to query
Returns: Returns:
(list) List of transactions (list) List of transactions
@ -2746,10 +2720,7 @@ class Daemon(AuthJSONRPCServer):
} }
""" """
return self.wallet_manager.get_history(self.get_account_or_default(account_id))
d = self.wallet_manager.get_history()
d.addCallback(lambda r: self._render_response(r))
return d
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_transaction_show(self, txid): def jsonrpc_transaction_show(self, txid):
@ -2795,9 +2766,7 @@ class Daemon(AuthJSONRPCServer):
... ...
] ]
""" """
return self.wallet_manager.get_utxos( return self.get_account_or_default(account_id).get_unspent_outputs()
self.get_account_or_default('account', account_id)
)
@requires(WALLET_COMPONENT) @requires(WALLET_COMPONENT)
def jsonrpc_block_show(self, blockhash=None, height=None): def jsonrpc_block_show(self, blockhash=None, height=None):
@ -2856,8 +2825,7 @@ class Daemon(AuthJSONRPCServer):
else: else:
result = "Downloaded blob %s" % blob_hash result = "Downloaded blob %s" % blob_hash
response = yield self._render_response(result) return result
defer.returnValue(response)
@requires(BLOB_COMPONENT, DATABASE_COMPONENT) @requires(BLOB_COMPONENT, DATABASE_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2876,16 +2844,14 @@ class Daemon(AuthJSONRPCServer):
""" """
if blob_hash not in self.blob_manager.blobs: if blob_hash not in self.blob_manager.blobs:
response = yield self._render_response("Don't have that blob") return "Don't have that blob"
defer.returnValue(response)
try: try:
stream_hash = yield self.storage.get_stream_hash_for_sd_hash(blob_hash) stream_hash = yield self.storage.get_stream_hash_for_sd_hash(blob_hash)
yield self.storage.delete_stream(stream_hash) yield self.storage.delete_stream(stream_hash)
except Exception as err: except Exception as err:
pass pass
yield self.blob_manager.delete_blobs([blob_hash]) yield self.blob_manager.delete_blobs([blob_hash])
response = yield self._render_response("Deleted %s" % blob_hash) return "Deleted %s" % blob_hash
defer.returnValue(response)
@requires(DHT_COMPONENT) @requires(DHT_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2924,7 +2890,7 @@ class Daemon(AuthJSONRPCServer):
} }
for node_id, host, port in peers for node_id, host, port in peers
] ]
defer.returnValue(results) return results
@requires(DATABASE_COMPONENT) @requires(DATABASE_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2960,8 +2926,7 @@ class Daemon(AuthJSONRPCServer):
else: else:
raise Exception('single argument must be specified') raise Exception('single argument must be specified')
yield self.storage.should_single_announce_blobs(blob_hashes, immediate=True) yield self.storage.should_single_announce_blobs(blob_hashes, immediate=True)
response = yield self._render_response(True) return True
defer.returnValue(response)
@requires(FILE_MANAGER_COMPONENT) @requires(FILE_MANAGER_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -2997,7 +2962,7 @@ class Daemon(AuthJSONRPCServer):
lbry_file = lbry_files[0] lbry_file = lbry_files[0]
results = yield reupload.reflect_file(lbry_file, reflector_server=reflector_server) results = yield reupload.reflect_file(lbry_file, reflector_server=reflector_server)
defer.returnValue(results) return results
@requires(BLOB_COMPONENT, WALLET_COMPONENT) @requires(BLOB_COMPONENT, WALLET_COMPONENT)
@defer.inlineCallbacks @defer.inlineCallbacks
@ -3059,9 +3024,7 @@ class Daemon(AuthJSONRPCServer):
page = page or 0 page = page or 0
start_index = page * page_size start_index = page * page_size
stop_index = start_index + page_size stop_index = start_index + page_size
blob_hashes_for_return = blob_hashes[start_index:stop_index] return blob_hashes[start_index:stop_index]
response = yield self._render_response(blob_hashes_for_return)
defer.returnValue(response)
@requires(BLOB_COMPONENT) @requires(BLOB_COMPONENT)
def jsonrpc_blob_reflect(self, blob_hashes, reflector_server=None): def jsonrpc_blob_reflect(self, blob_hashes, reflector_server=None):
@ -3132,15 +3095,14 @@ class Daemon(AuthJSONRPCServer):
try: try:
contact = yield self.dht_node.findContact(unhexlify(node_id)) contact = yield self.dht_node.findContact(unhexlify(node_id))
except TimeoutError: except TimeoutError:
result = {'error': 'timeout finding peer'} return {'error': 'timeout finding peer'}
defer.returnValue(result)
if not contact: if not contact:
defer.returnValue({'error': 'peer not found'}) return {'error': 'peer not found'}
try: try:
result = (yield contact.ping()).decode() result = (yield contact.ping()).decode()
except TimeoutError: except TimeoutError:
result = {'error': 'ping timeout'} result = {'error': 'ping timeout'}
defer.returnValue(result) return result
@requires(DHT_COMPONENT) @requires(DHT_COMPONENT)
def jsonrpc_routing_table_get(self): def jsonrpc_routing_table_get(self):
@ -3344,19 +3306,19 @@ class Daemon(AuthJSONRPCServer):
return certificates[0] return certificates[0]
raise ValueError("Couldn't find channel because a channel name or channel_id was not provided.") raise ValueError("Couldn't find channel because a channel name or channel_id was not provided.")
def get_account_or_default(self, argument: str, account_id: str, lbc_only=True): def get_account_or_default(self, account_id: str, argument_name: str="account", lbc_only=True):
if account_id is None: if account_id is None:
return self.default_account return self.default_account
return self.get_account_or_error(argument, account_id, lbc_only) return self.get_account_or_error(account_id, argument_name, lbc_only)
def get_account_or_error(self, argument: str, account_id: str, lbc_only=True): def get_account_or_error(self, account_id: str, argument_name: str="account", lbc_only=True):
for account in self.default_wallet.accounts: for account in self.default_wallet.accounts:
if account.id == account_id: if account.id == account_id:
if lbc_only and not isinstance(account, LBCAccount): if lbc_only and not isinstance(account, LBCAccount):
raise ValueError( raise ValueError(
"Found '{}', but it's an {} ledger account. " "Found '{}', but it's an {} ledger account. "
"'{}' requires specifying an LBC ledger account." "'{}' requires specifying an LBC ledger account."
.format(account_id, account.ledger.symbol, argument) .format(account_id, account.ledger.symbol, argument_name)
) )
return account return account
raise ValueError("Couldn't find account: {}.".format(account_id)) raise ValueError("Couldn't find account: {}.".format(account_id))

5
lbrynet/wallet/account.py
View file

@ -141,7 +141,7 @@ class Account(BaseAccount):
'nout': utxo.position, 'nout': utxo.position,
'have_certificate': utxo.ref.id in self.certificates 'have_certificate': utxo.ref.id in self.certificates
}) })
defer.returnValue(channels) return channels
@classmethod @classmethod
def get_private_key_from_seed(cls, ledger: 'baseledger.BaseLedger', seed: str, password: str): def get_private_key_from_seed(cls, ledger: 'baseledger.BaseLedger', seed: str, password: str):
@ -168,3 +168,6 @@ class Account(BaseAccount):
def get_claim(self, claim_id=None, txid=None, nout=None): def get_claim(self, claim_id=None, txid=None, nout=None):
return self.ledger.db.get_claim(self, claim_id, txid, nout) return self.ledger.db.get_claim(self, claim_id, txid, nout)
def get_claims(self):
return self.ledger.db.get_claims(self)

24
lbrynet/wallet/database.py
View file

@ -100,11 +100,31 @@ class WalletDatabase(BaseDatabase):
""".format(filter_sql), filter_value """.format(filter_sql), filter_value
) )
output_class = account.ledger.transaction_class.output_class output_class = account.ledger.transaction_class.output_class
defer.returnValue([ return [
output_class( output_class(
values[0], values[0],
output_class.script_class(values[1]), output_class.script_class(values[1]),
TXRefImmutable.from_id(values[2]), TXRefImmutable.from_id(values[2]),
position=values[3] position=values[3]
) for values in utxos ) for values in utxos
]) ]
@defer.inlineCallbacks
def get_claims(self, account):
utxos = yield self.db.runQuery(
"""
SELECT amount, script, txo.txid, position
FROM txo JOIN tx ON tx.txid=txo.txid
WHERE (is_claim OR is_update) AND txoid NOT IN (SELECT txoid FROM txi)
ORDER BY tx.height DESC;
"""
)
output_class = account.ledger.transaction_class.output_class
return [
output_class(
values[0],
output_class.script_class(values[1]),
TXRefImmutable.from_id(values[2]),
position=values[3]
) for values in utxos
]

26
lbrynet/wallet/manager.py
View file

@ -160,23 +160,20 @@ class LbryWalletManager(BaseWalletManager):
def get_new_address(self): def get_new_address(self):
return self.get_unused_address() return self.get_unused_address()
def list_addresses(self):
return self.default_account.get_addresses()
def reserve_points(self, address, amount: int): def reserve_points(self, address, amount: int):
# TODO: check if we have enough to cover amount # TODO: check if we have enough to cover amount
return ReservedPoints(address, amount) return ReservedPoints(address, amount)
@defer.inlineCallbacks @defer.inlineCallbacks
def send_amount_to_address(self, amount: int, destination_address: bytes): def send_amount_to_address(self, amount: int, destination_address: bytes, account=None):
account = self.default_account account = account or self.default_account
tx = yield Transaction.pay(amount, destination_address, [account], account) tx = yield Transaction.pay(amount, destination_address, [account], account)
yield account.ledger.broadcast(tx) yield account.ledger.broadcast(tx)
return tx return tx
def send_points_to_address(self, reserved: ReservedPoints, amount: int): def send_points_to_address(self, reserved: ReservedPoints, amount: int, account=None):
destination_address: bytes = reserved.identifier.encode('latin1') destination_address: bytes = reserved.identifier.encode('latin1')
return self.send_amount_to_address(amount, destination_address) return self.send_amount_to_address(amount, destination_address, account)
def get_wallet_info_query_handler_factory(self): def get_wallet_info_query_handler_factory(self):
return LBRYcrdAddressQueryHandlerFactory(self) return LBRYcrdAddressQueryHandlerFactory(self)
@ -198,17 +195,18 @@ class LbryWalletManager(BaseWalletManager):
def get_claims_for_name(self, name: str): def get_claims_for_name(self, name: str):
return self.ledger.network.get_claims_for_name(name) return self.ledger.network.get_claims_for_name(name)
def get_name_claims(self): @defer.inlineCallbacks
return defer.succeed([]) def address_is_mine(self, unknown_address, account):
for my_address in (yield account.get_addresses()):
def address_is_mine(self, address): if unknown_address == my_address:
return defer.succeed(True) return True
return False
def get_transaction(self, txid): def get_transaction(self, txid):
return self.default_account.ledger.get_transaction(txid) return self.default_account.ledger.get_transaction(txid)
def get_history(self): def get_history(self, account):
return defer.succeed([]) return account.get_transactions()
def get_utxos(self, account): def get_utxos(self, account):
return account.get_unspent_outputs() return account.get_unspent_outputs()