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)
