def cmd_login(args): auth_service = "KBase" try: auth_service = conf.AUTH_SERVICE if conf.AUTH_SERVICE else auth_service except AttributeError: pass if ARAST_AUTH_SERVICE: auth_service = ARAST_AUTH_SERVICE auth_service = "RAST" if args.rast else auth_service auth.authenticate(service=auth_service, save=True) sys.stderr.write("[.] Logged in\n")
def cmd_login(args): auth_service = 'KBase' try: auth_service = conf.AUTH_SERVICE if conf.AUTH_SERVICE else auth_service except AttributeError: pass if ARAST_AUTH_SERVICE: auth_service = ARAST_AUTH_SERVICE auth_service = 'RAST' if args.rast else auth_service auth.authenticate(service=auth_service, save=True) sys.stderr.write('[.] Logged in\n')
def run_command(): parser = get_parser() args = parser.parse_args() usage = parser.format_usage() frmt = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s") sh = logging.StreamHandler() sh.setLevel(logging.DEBUG) sh.setFormatter(frmt) clientlog = logging.getLogger("client") clientlog.setLevel(logging.INFO) clientlog.addHandler(sh) if args.verbose: clientlog.setLevel(logging.DEBUG) clientlog.debug("Logger Debugging mode") if args.command == "login": cmd_login(args) sys.exit() if args.command == "logout": cmd_logout(args) sys.exit() a_user, a_token = auth.verify_token(ARAST_AUTH_USER, ARAST_AUTH_TOKEN) if not a_user or not a_token: if ARAST_ENVIRON: sys.exit("Please use the {} controls to authenticate".format(ARAST_ENVIRON)) else: sys.stderr.write("You can use the login/logout commands to authenticate\n") a_user, a_token = auth.authenticate() # main command options a_url = args.arast_url or ARAST_URL a_url = utils.verify_url(a_url) logging.info("ARAST_URL: {}".format(a_url)) aclient = client.Client(a_url, a_user, a_token) if args.command == "upload": cmd_upload(args, aclient, usage, clientlog) elif args.command == "run": cmd_run(args, aclient, usage, clientlog) elif args.command == "stat": cmd_stat(args, aclient) elif args.command == "get": cmd_get(args, aclient) elif args.command == "avail": cmd_avail(args, aclient) elif args.command == "kill": print aclient.kill_jobs(args.job)
def run_command(): parser = get_parser() args = parser.parse_args() usage = parser.format_usage() frmt = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') sh = logging.StreamHandler() sh.setLevel(logging.DEBUG) sh.setFormatter(frmt) clientlog = logging.getLogger('client') clientlog.setLevel(logging.INFO) clientlog.addHandler(sh) if args.verbose: clientlog.setLevel(logging.DEBUG) clientlog.debug("Logger Debugging mode") if args.command == 'login': cmd_login(args) sys.exit() if args.command == 'logout': cmd_logout(args) sys.exit() a_user, a_token = auth.verify_token(ARAST_AUTH_USER,ARAST_AUTH_TOKEN) if not a_user or not a_token: if ARAST_ENVIRON: sys.exit('Please use the {} controls to authenticate'.format(ARAST_ENVIRON)) else: sys.stderr.write('You can use the login/logout commands to authenticate\n') a_user, a_token = auth.authenticate() # main command options a_url = args.arast_url or ARAST_URL a_url = utils.verify_url(a_url) logging.info('ARAST_URL: {}'.format(a_url)) aclient = client.Client(a_url, a_user, a_token) if args.command == 'upload': cmd_upload(args, aclient, usage, clientlog) elif args.command == 'run': cmd_run(args, aclient, usage, clientlog) elif args.command == 'stat': cmd_stat(args, aclient) elif args.command == 'get': cmd_get(args, aclient) elif args.command == 'avail': cmd_avail(args, aclient) elif args.command == 'kill': print aclient.kill_jobs(args.job)