def get_core_auth(): c = __get_or_load_config() configured_auth_token = c.get("core_auth_token", "False") if configured_auth_token.lower() != "false": return c.get("core_auth_token") DEFAULT_CONFIG_FILE = os.environ.get("BLOCKSTACK_CLIENT_CONFIG", "~/.blockstack/client.ini") config_file = os.path.expanduser(c.get("core_config", DEFAULT_CONFIG_FILE)) auth = blockstack_client_config.get_config(config_file)['api_password'] assert auth return auth
def walletUnlocked(): local_proxy = get_local_proxy() conf = config.get_config() if local_proxy is not False: wallet_data = local_proxy.get_wallet(conf['rpc_token']) wallet_data = json.loads(wallet_data) if 'error' in wallet_data: return False elif wallet_data['payment_address'] is None: return False else: return True else: return False
from virtualchain.lib.hashing import hex_hash160 # hack around absolute paths current_dir = os.path.abspath(os.path.dirname(__file__)) sys.path.insert(0, current_dir) current_dir = os.path.abspath(os.path.join( os.path.dirname(__file__), "..") ) sys.path.insert(0, current_dir) from blockstack_client.storage import parse_mutable_data, serialize_mutable_data from blockstack_client.config import log, get_config CONFIG_PATH = os.environ.get('BLOCKSTACK_CONFIG_PATH', None) assert CONFIG_PATH, "Missing BLOCKSTACK_CONFIG_PATH from environment" conf = get_config(CONFIG_PATH) print json.dumps(conf, indent=4, sort_keys=True) pk = keylib.ECPrivateKey() data_privkey = pk.to_hex() data_pubkey = pk.public_key().to_hex() test_data = [ ["my_first_datum", "hello world", 1, "unused", None], ["/my/second/datum", "hello world 2", 2, "unused", None], ["user\"_profile", '{"name":{"formatted":"judecn"},"v":"2"}', 3, "unused", None], ["empty_string", "", 4, "unused", None], ] def hash_data( d ): return hex_hash160( d )
import binascii import traceback import jsontokens from test import test_support from binascii import hexlify from blockstack_client import schemas import blockstack_client.storage import blockstack_client.config as blockstack_config import blockstack_client.config as blockstack_constants import blockstack_client.keys BASE_URL = 'http://localhost:5000' API_PASSWORD = blockstack_config.get_config( blockstack_constants.CONFIG_PATH).get('api_password', None) DEFAULT_WALLET_ADDRESS = "1QJQxDas5JhdiXhEbNS14iNjr8auFT96GP" class FakeResponseObj: def __init__(self): self.status_code = 600 self.data = "" class ForwardingClient: def __init__(self, base_url): self.base_url = base_url def get(self, endpoint, headers={}):
def run_cli(argv=None, config_path=CONFIG_PATH): """ Run a CLI command from arguments (defaults to sys.argv) Return the result of the command on success. The result will be a dict, and will have 'error' defined on error condition. """ if argv is None: argv = sys.argv # only version? if '-v' in argv or '--version' in argv: print config.VERSION sys.exit(0) # alternative config path? if '-c' in argv or '--config' in argv: i = 1 while i < len(argv): if argv[i] == '-c' or argv[i] == '--config': if i + 1 >= len(argv): print >> sys.stderr, "%s: missing path" % argv[i] sys.exit(1) config_path = argv[i+1] argv.pop(i) argv.pop(i) else: i+=1 conf = config.get_config(path=config_path) if conf is None: return {'error': 'Failed to load config'} conf_version = conf.get('client_version', '') if not semver_match( conf_version, VERSION ): exit_with_error("Invalid configuration file: %s != %s" % (conf_version, VERSION), \ "Your configuration file (%s) is out of date. Please move it and try again in order to automatically generate a new config file." % config_path) advanced_mode = conf.get('advanced_mode', False) parser = argparse.ArgumentParser( description='Blockstack cli version {}'.format(config.VERSION)) all_methods = [] subparsers = parser.add_subparsers(dest='action') # add basic methods basic_methods = get_cli_basic_methods() basic_method_info = parse_methods( basic_methods ) build_method_subparsers( subparsers, basic_method_info ) all_methods = basic_method_info if advanced_mode: # add advanced methods log.debug("Enabling advanced methods") advanced_methods = get_cli_advanced_methods() advanced_method_info = parse_methods( advanced_methods ) build_method_subparsers( subparsers, advanced_method_info ) all_methods += advanced_method_info # Print default help message, if no argument is given if len(argv) == 1 or '-h' in argv: parser.print_help() return {} interactive = False args = None directive = None try: args, unknown_args = parser.parse_known_args(args=argv[1:]) directive = args.action except SystemExit: # bad arguments # special case: if the method is specified, but no method arguments are given, # then switch to prompting the user for individual arguments. try: directive_parser = argparse.ArgumentParser(description='Blockstack cli version {}'.format(config.VERSION)) directive_subparsers = directive_parser.add_subparsers(dest='action') # only parse the directive build_method_subparsers( directive_subparsers, all_methods, include_args=False, include_opts=False ) directive_args, directive_unknown_args = directive_parser.parse_known_args( args=argv[1:] ) # want interactive prompting interactive = True directive = directive_args.action except SystemExit: # still invalid parser.print_help() return {'error': 'Invalid arguments. Try passing "-h".'} result = {} blockstack_server = conf['server'] blockstack_port = conf['port'] # initialize blockstack connection session(conf=conf, server_host=blockstack_server, server_port=blockstack_port, set_global=True) # dispatch to the apporpriate method for method_info in all_methods: if directive != method_info['command']: continue method = method_info['method'] # interactive? if interactive: print "" print "Interactive prompt engaged. Press Ctrl+C to quit" print "Help for '%s': %s" % (method_info['command'], method_info['help']) print "" required_args = prompt_args( method_info['args'], lambda arghelp, argname: raw_input("%s ('%s'): " % (arghelp, argname)) ) if required_args is None: return {'error': 'Failed to prompt for arguments'} optional_args = prompt_args( method_info['opts'], lambda arghelp, argname: raw_input("optional: %s ('%s'): " % (arghelp, argname) )) if optional_args is None: return {'error': 'Failed to prompt for arguments'} full_args = [method_info['command']] + required_args + optional_args try: args, unknown_args = parser.parse_known_args( args=full_args ) except SystemExit: # invalid arguments return {'error': 'Invalid arguments. Please try again.'} result = method( args, config_path=config_path ) return result # not found return {'error': "No such command '%s'" % args.action}
def run_cli(argv=None, config_path=CONFIG_PATH): """ Run a CLI command from arguments (defaults to sys.argv) Return the result of the command on success. The result will be a dict, and will have 'error' defined on error condition. """ if argv is None: argv = sys.argv # alternative config path? if '-c' in argv or '--config' in argv: i = 1 while i < len(argv): if argv[i] == '-c' or argv[i] == '--config': if i + 1 >= len(argv): print >> sys.stderr, "%s: missing path" % argv[i] sys.exit(1) config_path = argv[i+1] argv.pop(i) argv.pop(i) else: i+=1 conf = config.get_config(path=config_path) if conf is None: return {'error': 'Failed to load config'} conf_version = conf.get('client_version', '') if not semver_match( conf_version, VERSION ): exit_with_error("Invalid configuration file: %s != %s" % (conf_version, VERSION), \ "Your configuration file (%s) is out of date. Please move it and try again in order to automatically generate a new config file." % config_path) advanced_mode = conf.get('advanced_mode', False) parser = argparse.ArgumentParser( description='Blockstack cli version {}'.format(config.VERSION)) all_methods = [] subparsers = parser.add_subparsers(dest='action') # add basic methods basic_methods = get_cli_basic_methods() basic_method_info = parse_methods( basic_methods ) build_method_subparsers( subparsers, basic_method_info ) all_methods = basic_method_info if advanced_mode: # add advanced methods log.debug("Enabling advanced methods") advanced_methods = get_cli_advanced_methods() advanced_method_info = parse_methods( advanced_methods ) build_method_subparsers( subparsers, advanced_method_info ) all_methods += advanced_method_info # Print default help message, if no argument is given if len(argv) == 1: parser.print_help() return {} interactive = False args = None directive = None try: args, unknown_args = parser.parse_known_args(args=argv[1:]) directive = args.action except SystemExit: # bad arguments # special case: if the method is specified, but no method arguments are given, # then switch to prompting the user for individual arguments. try: directive_parser = argparse.ArgumentParser(description='Blockstack cli version {}'.format(config.VERSION)) directive_subparsers = directive_parser.add_subparsers(dest='action') # only parse the directive build_method_subparsers( directive_subparsers, all_methods, include_args=False, include_opts=False ) directive_args, directive_unknown_args = directive_parser.parse_known_args( args=argv[1:] ) # want interactive prompting interactive = True directive = directive_args.action except SystemExit: # still invalid parser.print_help() return {'error': 'Invalid arguments. Try passing "-h".'} result = {} blockstack_server = conf['server'] blockstack_port = conf['port'] # initialize blockstack connection session(conf=conf, server_host=blockstack_server, server_port=blockstack_port, set_global=True) # dispatch to the apporpriate method for method_info in all_methods: if directive != method_info['command']: continue method = method_info['method'] # interactive? if interactive: print "" print "Interactive prompt engaged. Press Ctrl+C to quit" print "Help for '%s': %s" % (method_info['command'], method_info['help']) print "" required_args = prompt_args( method_info['args'], lambda arghelp, argname: raw_input("%s ('%s'): " % (arghelp, argname)) ) if required_args is None: return {'error': 'Failed to prompt for arguments'} optional_args = prompt_args( method_info['opts'], lambda arghelp, argname: raw_input("optional: %s ('%s'): " % (arghelp, argname) )) if optional_args is None: return {'error': 'Failed to prompt for arguments'} full_args = [method_info['command']] + required_args + optional_args try: args, unknown_args = parser.parse_known_args( args=full_args ) except SystemExit: # invalid arguments return {'error': 'Invalid arguments. Please try again.'} result = method( args, config_path=config_path ) return result # not found return {'error': "No such command '%s'" % args.action}
import binascii import traceback import jsontokens from test import test_support from binascii import hexlify from blockstack_client import schemas import blockstack_client.storage import blockstack_client.config as blockstack_config import blockstack_client.config as blockstack_constants import blockstack_client.keys BASE_URL = 'http://localhost:5000' API_PASSWORD = blockstack_config.get_config( blockstack_constants.CONFIG_PATH).get('api_password', None) DEFAULT_WALLET_ADDRESS = "1QJQxDas5JhdiXhEbNS14iNjr8auFT96GP" class FakeResponseObj: def __init__(self): self.status_code = 600 self.data = "" class ForwardingClient: def __init__(self, base_url): self.base_url = base_url def get(self, endpoint, headers = {}): resp = requests.get(self.base_url + endpoint, headers = headers) ret_obj = FakeResponseObj() ret_obj.status_code = resp.status_code
def run_cli(): """ run cli """ conf = config.get_config() if conf is None: log.error("Failed to load config") sys.exit(1) advanced_mode = conf['advanced_mode'] parser = argparse.ArgumentParser( description='Blockstack cli version {}'.format(config.VERSION)) parser.register('action', 'parsers', AliasedSubParsersAction) subparsers = parser.add_subparsers(dest='action') add_subparsers(subparsers) if advanced_mode == "on": add_advanced_subparsers(subparsers) # Print default help message, if no argument is given if len(sys.argv) == 1: parser.print_help() sys.exit(1) args, unknown_args = parser.parse_known_args() result = {} conf = config.get_config() blockstack_server = conf['server'] blockstack_port = conf['port'] proxy = client.session(conf=conf, server_host=blockstack_server, server_port=blockstack_port, set_global=True) # start the two background processes (rpc daemon and monitor queue) start_background_daemons() if args.action == 'balance': if not os.path.exists(WALLET_PATH): initialize_wallet() total_balance, addresses = get_total_balance() result['total_balance'] = total_balance if args.details: result['addresses'] = addresses elif args.action == 'price': fqu = str(args.name) check_valid_name(fqu) try: resp = client.get_name_cost(fqu) except socket_error: exit_with_error("Error connecting to server") if 'error' in resp: exit_with_error(resp['error']) data = get_total_fees(resp) result = data elif args.action == 'config': data = {} settings_updated = False data["message"] = "Updated settings for" if args.host is not None: config.update_config('blockstack-client', 'server', args.host) data["message"] += " host" settings_updated = True if args.port is not None: config.update_config('blockstack-client', 'port', args.port) data["message"] += " port" settings_updated = True if args.advanced is not None: if args.advanced != "on" and args.advanced != "off": exit_with_error("Use --advanced=on or --advanced=off") else: config.update_config('blockstack-client', 'advanced_mode', args.advanced) data["message"] += " advanced" settings_updated = True # reload conf conf = config.get_config() if settings_updated: result['message'] = data['message'] else: result['message'] = "No config settings were updated." elif args.action == 'deposit': if not os.path.exists(WALLET_PATH): initialize_wallet() result['message'] = 'Send bitcoins to the address specified.' result['address'], owner_address = get_addresses_from_file() elif args.action == 'import': if not os.path.exists(WALLET_PATH): initialize_wallet() result['message'] = 'Send the name you want to receive to the' result['message'] += ' address specified.' payment_address, result['address'] = get_addresses_from_file() elif args.action == 'names': if not os.path.exists(WALLET_PATH): initialize_wallet() result['names_owned'] = get_all_names_owned() if args.details: result['addresses'] = get_owner_addresses() elif args.action in ('info', 'status', 'ping', 'details'): resp = client.getinfo() result = {} result['server_host'] = conf['server'] result['server_port'] = str(conf['port']) result['cli_version'] = config.VERSION result['advanced_mode'] = conf['advanced_mode'] if 'error' in resp: result['server_alive'] = False result['server_error'] = resp['error'] else: result['server_alive'] = True result['server_version'] = resp['blockstore_version'] try: result['last_block_processed'] = resp['last_block'] except: result['last_block_processed'] = resp['blocks'] result['last_block_seen'] = resp['bitcoind_blocks'] result['consensus_hash'] = resp['consensus'] if advanced_mode == 'on': result['testset'] = resp['testset'] proxy = get_local_proxy() if proxy is not False: current_state = json.loads(proxy.state()) queue = {} pending_queue = [] preorder_queue = [] register_queue = [] update_queue = [] transfer_queue = [] def format_new_entry(entry): new_entry = {} new_entry['name'] = entry['fqu'] confirmations = get_tx_confirmations(entry['tx_hash']) if confirmations is None: confirmations = 0 new_entry['confirmations'] = confirmations return new_entry def format_queue_display(preorder_queue, register_queue): for entry in register_queue: name = entry['name'] for check_entry in preorder_queue: if check_entry['name'] == name: preorder_queue.remove(check_entry) for entry in current_state: if 'type' in entry: if entry['type'] == 'preorder': preorder_queue.append(format_new_entry(entry)) elif entry['type'] == 'register': register_queue.append(format_new_entry(entry)) elif entry['type'] == 'update': update_queue.append(format_new_entry(entry)) elif entry['type'] == 'transfer': transfer_queue.append(format_new_entry(entry)) format_queue_display(preorder_queue, register_queue) if len(preorder_queue) != 0: queue['preorder'] = preorder_queue if len(register_queue) != 0: queue['register'] = register_queue if len(update_queue) != 0: queue['update'] = update_queue if len(transfer_queue) != 0: queue['transfer'] = transfer_queue if queue != {}: result['queue'] = queue elif args.action == 'lookup': data = {} blockchain_record = None fqu = str(args.name) check_valid_name(fqu) try: blockchain_record = client.get_name_blockchain_record(fqu) except socket_error: exit_with_error("Error connecting to server.") if 'value_hash' not in blockchain_record: exit_with_error("%s is not registered" % fqu) data_id = blockchain_record['value_hash'] owner_address = blockchain_record['address'] profile = client.get_immutable(str(args.name), data_id)['data'] zone_file = profile profile = resolve_zone_file_to_profile(profile, owner_address) if not is_profile_in_legacy_format(profile): data['data_record'] = profile data['zone_file'] = zone_file else: data['data_record'] = json.loads(profile) #except Exception as e: # print e # data['data_record'] = None result = data elif args.action == 'whois': data = {} record = None fqu = str(args.name) check_valid_name(fqu) try: record = client.get_name_blockchain_record(fqu) except socket_error: exit_with_error("Error connecting to server.") if 'value_hash' not in record: result['registered'] = False else: result['registered'] = True result['block_preordered_at'] = record['preorder_block_number'] result['block_renewed_at'] = record['last_renewed'] result['owner_address'] = record['address'] result['owner_public_key'] = record['sender_pubkey'] result['owner_script'] = record['sender'] result['preorder_transaction_id'] = record['txid'] elif args.action == 'register': if not os.path.exists(WALLET_PATH): initialize_wallet() result = {} fqu = str(args.name) check_valid_name(fqu) cost = client.get_name_cost(fqu) if 'error' in cost: exit_with_error(cost['error']) if nameRegistered(fqu): exit_with_error("%s is already registered." % fqu) if not walletUnlocked(): unlock_wallet() fees = get_total_fees(cost) try: cost = fees['total_estimated_cost'] input_prompt = "Registering %s will cost %s BTC." % (fqu, cost) input_prompt += " Continue? (y/n): " user_input = raw_input(input_prompt) user_input = user_input.lower() if user_input != 'y': print "Not registering." exit(0) except KeyboardInterrupt: print "\nExiting." exit(0) payment_address, owner_address = get_addresses_from_file() if not hasEnoughBalance(payment_address, fees['total_estimated_cost']): msg = "Address %s doesn't have enough balance." % payment_address exit_with_error(msg) if recipientNotReady(owner_address): msg = "Address %s owns too many names already." % owner_address exit_with_error(msg) if dontuseAddress(payment_address): msg = "Address %s has pending transactions." % payment_address msg += " Wait and try later." exit_with_error(msg) proxy = get_local_proxy() try: resp = proxy.preorder(fqu) except: exit_with_error("Error talking to server, try again.") if 'success' in resp and resp['success']: result = resp else: if 'error' in resp: exit_with_error(resp['error']) if 'message' in resp: exit_with_error(resp['message']) elif args.action == 'update': if not os.path.exists(WALLET_PATH): initialize_wallet() fqu = str(args.name) check_valid_name(fqu) user_data = str(args.data) try: user_data = json.loads(user_data) except: exit_with_error("Data is not in JSON format.") tests_for_update_and_transfer(fqu) if profileonBlockchain(fqu, user_data): msg = "Data is same as current data record, update not needed." exit_with_error(msg) if not walletUnlocked(): unlock_wallet() proxy = get_local_proxy() try: resp = proxy.update(fqu, user_data) except: exit_with_error("Error talking to server, try again.") if 'success' in resp and resp['success']: result = resp else: if 'error' in resp: exit_with_error(resp['error']) if 'message' in resp: exit_with_error(resp['message']) elif args.action == 'transfer': if not os.path.exists(WALLET_PATH): initialize_wallet() fqu = str(args.name) check_valid_name(fqu) transfer_address = str(args.address) tests_for_update_and_transfer(fqu, transfer_address=transfer_address) if not walletUnlocked(): unlock_wallet() proxy = get_local_proxy() try: resp = proxy.transfer(fqu, transfer_address) except: exit_with_error("Error talking to server, try again.") if 'success' in resp and resp['success']: result = resp else: if 'error' in resp: exit_with_error(resp['error']) if 'message' in resp: exit_with_error(resp['message']) # ---------------------- Advanced options --------------------------------- elif args.action == 'wallet': if not os.path.exists(WALLET_PATH): result = initialize_wallet() else: unlock_wallet(display_enabled=True) elif args.action == 'consensus': if args.block_height is None: # by default get last indexed block resp = client.getinfo() if 'error' in resp: exit_with_error("Error connecting to server.") elif 'last_block' in resp or 'blocks' in resp: if 'last_block' in resp: args.block_height = client.getinfo()['last_block'] elif 'blocks' in resp: args.block_height = client.getinfo()['blocks'] else: result['error'] = "Server is indexing. Try again" exit(0) resp = client.get_consensus_at(int(args.block_height)) data = {} data['consensus'] = resp data['block_height'] = args.block_height result = data elif args.action == 'register_tx': result = client.register(str(args.name), str(args.privatekey), str(args.addr), tx_only=True) elif args.action == 'register_subsidized': result = client.register_subsidized(str(args.name), str(args.privatekey), str(args.addr), str(args.subsidy_key)) elif args.action == 'update_tx': txid = None if args.txid is not None: txid = str(args.txid) result = client.update(str(args.name), str(args.record_json), str(args.privatekey), txid=txid, tx_only=True) elif args.action == 'update_subsidized': txid = None if args.txid is not None: txid = str(args.txid) result = client.update_subsidized(str(args.name), str(args.record_json), str(args.public_key), str(args.subsidy_key), txid=txid) elif args.action == 'transfer_tx': keepdata = False if args.keepdata.lower() not in ["on", "false"]: print >> sys.stderr, "Pass 'true' or 'false' for keepdata" sys.exit(1) if args.keepdata.lower() == "on": keepdata = True result = client.transfer(str(args.name), str(args.address), keepdata, str(args.privatekey), tx_only=True) elif args.action == 'preorder': register_addr = None if args.address is not None: register_addr = str(args.address) result = client.preorder(str(args.name), str(args.privatekey), register_addr=register_addr) elif args.action == 'preorder_tx': register_addr = None if args.address is not None: register_addr = str(args.address) result = client.preorder(str(args.name), str(args.privatekey), register_addr=register_addr, tx_only=True) elif args.action == 'preorder_subsidized': result = client.preorder_subsidized(str(args.name), str(args.public_key), str(args.address), str(args.subsidy_key)) elif args.action == 'transfer_subsidized': keepdata = False if args.keepdata.lower() not in ["on", "false"]: print >> sys.stderr, "Pass 'true' or 'false' for keepdata" sys.exit(1) if args.keepdata.lower() == "on": keepdata = True result = client.transfer_subsidized(str(args.name), str(args.address), keepdata, str(args.public_key), str(args.subsidy_key)) elif args.action == 'renew': result = client.renew(str(args.name), str(args.privatekey)) elif args.action == 'renew_tx': result = client.renew(str(args.name), str(args.privatekey), tx_only=True) elif args.action == 'renew_subsidized': result = client.renew_subsidized(str(args.name), str(args.public_key), str(args.subsidy_key)) elif args.action == 'revoke': result = client.revoke(str(args.name), str(args.privatekey)) elif args.action == 'revoke_tx': result = client.revoke(str(args.name), str(args.privatekey), tx_only=True) elif args.action == 'revoke_subsidized': result = client.revoke_subsidized(str(args.name), str(args.public_key), str(args.subsidy_key)) elif args.action == 'name_import': result = client.name_import(str(args.name), str(args.address), str(args.hash), str(args.privatekey)) elif args.action == 'namespace_preorder': reveal_addr = None if args.address is not None: reveal_addr = str(args.address) result = client.namespace_preorder(str(args.namespace_id), str(args.privatekey), reveal_addr=reveal_addr) elif args.action == 'namespace_reveal': bucket_exponents = args.bucket_exponents.split(',') if len(bucket_exponents) != 16: raise Exception("bucket_exponents must be a 16-value CSV \ of integers") for i in xrange(0, len(bucket_exponents)): try: bucket_exponents[i] = int(bucket_exponents[i]) except: raise Exception("bucket_exponents must contain integers in \ range [0, 16)") lifetime = int(args.lifetime) if lifetime < 0: lifetime = 0xffffffff # means "infinite" to blockstack-server result = client.namespace_reveal(str(args.namespace_id), str(args.addr), lifetime, int(args.coeff), int(args.base), bucket_exponents, int(args.nonalpha_discount), int(args.no_vowel_discount), str(args.privatekey)) elif args.action == 'namespace_ready': result = client.namespace_ready(str(args.namespace_id), str(args.privatekey)) elif args.action == 'put_mutable': result = client.put_mutable(str(args.name), str(args.data_id), str(args.data), str(args.privatekey)) elif args.action == 'put_immutable': result = client.put_immutable(str(args.name), str(args.data), str(args.privatekey), conf=conf) elif args.action == 'get_mutable': result = client.get_mutable(str(args.name), str(args.data_id), conf=conf) elif args.action == 'get_immutable': result = client.get_immutable(str(args.name), str(args.hash)) elif args.action == 'delete_immutable': result = client.delete_immutable(str(args.name), str(args.hash), str(args.privatekey)) elif args.action == 'delete_mutable': result = client.delete_mutable(str(args.name), str(args.data_id), str(args.privatekey)) elif args.action == 'get_name_blockchain_record': result = client.get_name_blockchain_record(str(args.name)) elif args.action == 'get_namespace_blockchain_record': result = client.get_namespace_blockchain_record(str(args.namespace_id)) elif args.action == 'lookup_snv': result = client.lookup_snv(str(args.name), int(args.block_id), str(args.consensus_hash)) elif args.action == 'get_name_record': result = client.get_name_record(str(args.name)) elif args.action == 'get_names_owned_by_address': result = client.get_names_owned_by_address(str(args.address)) elif args.action == 'get_namespace_cost': result = client.get_namespace_cost(str(args.namespace_id)) elif args.action == 'get_all_names': offset = None count = None if args.offset is not None: offset = int(args.offset) if args.count is not None: count = int(args.count) result = client.get_all_names(offset, count) elif args.action == 'get_names_in_namespace': offset = None count = None if args.offset is not None: offset = int(args.offset) if args.count is not None: count = int(args.count) result = client.get_names_in_namespace(str(args.namespace_id), offset, count) elif args.action == 'get_nameops_at': result = client.get_nameops_at(int(args.block_id)) print_result(result)
def run_cli(argv=None, config_path=CONFIG_PATH): """ Run a CLI command from arguments (defaults to sys.argv) Return the result of the command on success. The result will be a dict, and will have 'error' defined on error condition. """ if argv is None: argv = sys.argv cli_debug = False cli_config_argv = False cli_default_yes = False cli_api_pass = None cli_dry_run = False if '-v' in argv or '--version' in argv: print(VERSION) sys.exit(0) # debug? new_argv, cli_debug = find_arg(argv, False, '-d', '--debug') if new_argv is None: # invalid sys.exit(1) if cli_debug: os.environ['BLOCKSTACK_DEBUG'] = '1' log.setLevel(logging.DEBUG) log.debug("Activated debugging") # alternative config path? new_argv, cli_config_path = find_arg(argv, True, '-c', '--config') if new_argv is None: # invalid sys.exit(1) argv = new_argv if cli_config_path: cli_config_argv = True config_path = cli_config_path log.debug('Use config file {}'.format(config_path)) os.environ['BLOCKSTACK_CLIENT_CONFIG'] = config_path # CLI-given password? new_argv, cli_password = find_arg(argv, True, '-p', '--password') if new_argv is None: # invalid sys.exit(1) argv = new_argv if cli_password and os.environ.get( 'BLOCKSTACK_CLIENT_WALLET_PASSWORD') is None: log.debug("Use CLI password") os.environ["BLOCKSTACK_CLIENT_WALLET_PASSWORD"] = cli_password # assume YES to all prompts? new_argv, cli_default_yes = find_arg(argv, False, '-y', '--yes') if new_argv is None: # invalid sys.exit(1) if cli_default_yes or os.environ.get( "BLOCKSTACK_CLIENT_INTERACTIVE_YES") == "1": if cli_debug: print("Assume YES to all interactive prompts", file=sys.stderr) os.environ["BLOCKSTACK_CLIENT_INTERACTIVE_YES"] = '1' # API password? new_argv, cli_api_pass = find_arg(argv, True, '-a', '--api_password') if new_argv is None: # invalid sys.exit(1) argv = new_argv if cli_api_pass: os.environ['BLOCKSTACK_API_PASSWORD'] = cli_api_pass # dry-run? new_argv, cli_dry_run = find_arg(argv, False, '-n', '--dry-run') if new_argv is None: # invalid sys.exit(1) if cli_dry_run or os.environ.get("BLOCKSTACK_DRY_RUN") == "1": if cli_debug: print('Dry-run; no transactions will be sent', file=sys.stderr) os.environ['BLOCKSTACK_DRY_RUN'] = "1" if cli_config_argv or cli_debug or cli_dry_run: # re-exec to reset variables if cli_debug: print("Re-exec {} with {}".format(argv[0], argv), file=sys.stderr) os.execv(argv[0], argv) # do one-time opt-in request uuid_path = client_uuid_path(config_dir=os.path.dirname(config_path)) first_time = False client_uuid = None if not os.path.exists(uuid_path): first_time = True client_uuid = get_or_set_uuid(config_dir=os.path.dirname(config_path)) if os.environ.get('BLOCKSTACK_CLIENT_INTERACTIVE_YES') != '1': # interactive allowed # prompt for email print( "Would you like to receive an email when there is a new release of this software available?" ) email_addr = raw_input("Email address (leave blank to opt out): ") # will only process real email addresses when we email announcements out if len(email_addr) > 0: analytics_user_register(client_uuid, email_addr) conf = config.get_config( path=config_path, interactive=(os.environ.get('BLOCKSTACK_CLIENT_INTERACTIVE_YES') != '1')) if conf is None: return {'error': 'Failed to load config'} conf_version = conf.get('client_version', '') if not semver_match(conf_version, VERSION): # back up the config file if not cli_config_argv: # default config file backup_path = config.backup_config_file(config_path=config_path) if not backup_path: exit_with_error( "Failed to back up legacy configuration file {}".format( config_path)) else: exit_with_error( "Backed up legacy configuration file from {} to {} and re-generated a new, default configuration. Please restart." .format(config_path, backup_path)) advanced_mode = conf.get('advanced_mode', False) parser = argparse.ArgumentParser( description='Blockstack cli version {}'.format(config.VERSION)) all_methods = [] subparsers = parser.add_subparsers(dest='action') # add basic methods all_method_names = get_cli_methods() all_methods = parse_methods(all_method_names) build_method_subparsers(subparsers, all_methods) if not advanced_mode: # remove advanced methods all_methods = filter(lambda m: 'advanced' not in m['pragmas'], all_methods) # Print default help message, if no argument is given if len(argv) == 1 or '-h' in argv or '--help' in argv: parser.print_help() sys.exit(0) interactive, args, directive = False, None, None try: args, unknown_args = parser.parse_known_args(args=argv[1:]) directive = args.action except SystemExit: # bad arguments # special case: if the method is specified, but no method arguments are given, # then switch to prompting the user for individual arguments. try: directive_parser = argparse.ArgumentParser( description='Blockstack cli version {}'.format(config.VERSION)) directive_subparsers = directive_parser.add_subparsers( dest='action') # only parse the directive build_method_subparsers(directive_subparsers, all_methods, include_args=False, include_opts=False) directive_args, directive_unknown_args = directive_parser.parse_known_args( args=argv[1:]) # want interactive prompting interactive, directive = True, directive_args.action except SystemExit: # still invalid parser.print_help() return {'error': 'Invalid arguments. Try passing "-h".'} result = {} blockstack_server, blockstack_port = conf['server'], conf['port'] # initialize blockstack connection session(conf=conf, server_host=blockstack_server, server_port=blockstack_port, set_global=True) prompt_func = lambda help, name: raw_input('optional: {} ("{}"): '.format( help, name)) # dispatch to the apporpriate method for method_info in all_methods: if directive != method_info['command']: continue method = method_info['method'] pragmas = method_info['pragmas'] # interactive? if interactive: print('') print('Interactive prompt engaged. Press Ctrl+C to quit') print('Help for "{}": {}'.format(method_info['command'], method_info['help'])) print('') required_args = prompt_args(method_info['args'], prompt_func) if required_args is None: return {'error': 'Failed to prompt for arguments'} optional_args = prompt_args(method_info['opts'], prompt_func) if optional_args is None: return {'error': 'Failed to prompt for arguments'} full_args = [method_info['command'] ] + required_args + optional_args try: args, unknown_args = parser.parse_known_args(args=full_args) except SystemExit: # invalid arguments return {'error': 'Invalid arguments. Please try again.'} result = method(args, config_path=config_path) return {'status': True, 'result': result, 'pragmas': pragmas} # not found return {'error': 'No such command "{}"'.format(args.action)}