def get_client(args, config):
"""Return instantiated client.
Config will provide global overrides.
"""
client = Client()
date = datetime.strftime(datetime.today(), "%Y_%m_%dT%H_%M_%S_%f")
client.ip = args.remote
client.jobname = "{0}-{1}-worker".format(client.ip, date)
if args.user is not None:
client.user = args.user
client.is_sudoer = True
if args.delay_pickup:
client.delay_pickup = True
if not config['DEFAULT']['output']:
if args.output is not None:
client.output = args.output
if not config['DEFAULT']['compress']:
if args.dont_compress:
client.compress = not client.compress
cprint(
"> Establishing secure connection {0}@{1}".format(
client.user, client.ip), 'blue')
client.pass_ = getpass.getpass()
return client
def __get_client__(args, config):
"""Return instantiated client.
Args will override global config defaults.
"""
client = Client()
client.ip = args.remote
# Check args for remote/local issues
if client.ip == 'local':
if args.socket:
sys.exit(colored("Can not use socket on local machine", 'red'))
elif args.compress:
sys.exit(colored("Can not compress with local transfer",
'red'))
if args.socket:
if args.compress:
sys.exit(
colored("Can not compress with socket usage transfer",
'red'))
else:
client.port = int(args.socket)
if args.case:
client.job_name = args.case
else:
client.job_name = "{0}_{1}".format(client.ip, config.date)
if args.user:
client.user = args.user
else:
client.user = '******'
if args.format:
client.format = args.format
else:
client.format = config.format
if args.digest:
if args.digest == 'None':
client.digest = ''
else:
client.digest = args.digest
else:
client.digest = config.digest
if args.output:
client.output = args.output
else:
client.output = config.output
client.compress = args.compress
client.output_dir = "{0}{1}/".format(config.output_dir,
client.job_name)
cprint(
"> Establishing secure connection {0}@{1}".format(
client.user, client.ip), 'blue')
if args.key:
client.key = args.key
cprint("> Please enter key pass phrase", 'blue')
client.pass_ = getpass.getpass()
return client
