def __init__(self, delegate, args):
super(IntKeyWorkload, self).__init__(delegate, args)
self._auth_info = args.auth_info
self._urls = []
self._pending_batches = {}
self._lock = threading.Lock()
self._delegate = delegate
self._deps = {}
context = create_context('secp256k1')
crypto_factory = CryptoFactory(context=context)
if args.key_file is not None:
try:
with open(args.key_file, 'r') as infile:
signing_key = infile.read().strip()
private_key = Secp256k1PrivateKey.from_hex(signing_key)
self._signer = crypto_factory.new_signer(
private_key=private_key)
except ParseError as pe:
raise IntKeyCliException(str(pe))
except IOError as ioe:
raise IntKeyCliException(str(ioe))
else:
self._signer = crypto_factory.new_signer(
context.new_random_private_key())
def main(prog_name=os.path.basename(sys.argv[0]), args=None):
if args is None:
args = sys.argv[1:]
parser = create_parser(prog_name)
args = parser.parse_args(args)
if args.verbose is None:
verbose_level = 0
else:
verbose_level = args.verbose
setup_loggers(verbose_level=verbose_level)
if not args.command:
parser.print_help()
sys.exit(1)
if args.command == 'generate':
do_generate(args)
elif args.command == 'populate':
do_populate(args)
elif args.command == 'load':
do_load(args)
elif args.command == 'create_batch':
do_create_batch(args)
elif args.command == 'workload':
do_workload(args)
else:
raise IntKeyCliException("invalid command: {}".format(args.command))
def main(prog_name=os.path.basename(sys.argv[0]), args=sys.argv[1:]):
parser = create_parser(prog_name)
args = parser.parse_args(args)
if args.verbose is None:
verbose_level = 0
else:
verbose_level = args.verbose
setup_loggers(verbose_level=verbose_level)
if args.command == 'generate':
do_generate(args)
elif args.command == 'load':
do_load(args)
else:
raise IntKeyCliException("invalid command: {}".format(args.command))
def main(prog_name=os.path.basename(sys.argv[0]), args=None):
if args is None:
args = sys.argv[1:]
parser = create_parser(prog_name)
args = parser.parse_args(args)
if args.verbose is None:
verbose_level = 0
else:
verbose_level = args.verbose
setup_loggers(verbose_level=verbose_level)
if not args.command:
parser.print_help()
sys.exit(1)
if args.command == 'set':
do_set(args)
elif args.command == 'inc':
do_inc(args)
elif args.command == 'dec':
do_dec(args)
elif args.command == 'mul':
do_mul(args)
elif args.command == 'show':
do_show(args)
elif args.command == 'list':
do_list(args)
elif args.command == 'start':
start_rest_api(args, 'shell', 8888)
elif args.command == 'generate':
do_generate(args)
elif args.command == 'populate':
do_populate(args)
elif args.command == 'load':
do_load(args)
elif args.command == 'create_batch':
do_create_batch(args)
elif args.command == 'workload':
do_workload(args)
else:
raise IntKeyCliException("invalid command: {}".format(args.command))
