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