| | |
|---|
| | | |
|---|
| | | from IPython import embed |
|---|
| | | |
|---|
| | | import __main__ as main |
|---|
| | | |
|---|
| | | #Color definitions for terminal |
|---|
| | | class bcolors: |
|---|
| | | HEADER = '\033[95m' |
|---|
| | |
|---|
| | | help='process number at host. If hostname is not specified, localhost is assumed. If no processes are specified all processes on all hosts are assumed.') |
|---|
| | | action_group=parser.add_mutually_exclusive_group(required=True) |
|---|
| | | action_group.add_argument('-c','--comment',nargs=1, help='append comment to current comment') |
|---|
| | | action_group.add_argument('--analysis', nargs='+', help='runs analysis function defined in configuration file') |
|---|
| | | action_group.add_argument('--delete-comment', help='delete comment',action='store_true') |
|---|
| | | action_group.add_argument('-k','--kill','--stop','--suspend', help='stop/kill the process', action='store_true') |
|---|
| | | action_group.add_argument('-r','--run','--start','--continue', help='start/continue process', action='store_true') |
|---|
| | |
|---|
| | | action_group.add_argument('--jump-to-ipython', help='loads the variables and jumps to IPython shell', action="store_true") |
|---|
| | | action_group.add_argument('-p','--preview',help='preview last VTU shape',action='store_true') |
|---|
| | | parser.add_argument('--force', help='if dangerous operation (killing all the processes) is requested, this flag is required to execute the operation. Otherwise, the request will be ignored.', action="store_true") |
|---|
| | | parser.add_argument('-H', '--host', nargs=1, help='specifies which host is itended for the operation. Defauts to localhost for all operations except --status and --version, where all configured hosts are assumed.') |
|---|
| | | parser.add_argument('-H', '--host', nargs='+', help='specifies which host is itended for the operation. Defauts to localhost for all operations except --status and --version, where all configured hosts are assumed.') |
|---|
| | | parser.add_argument('--html', help='Generate HTML output', action="store_true") |
|---|
| | | parser.add_argument('-n', nargs='+', metavar='PROC_NO', type=int, help='OBSOLETE. Specifies process numbers.') |
|---|
| | | parser.add_argument('-R','--raw',help='print status and the rest of the information in raw format', action="store_true") |
|---|
| | |
|---|
| | | port=22 #default ssh port |
|---|
| | | rm=Remote.Connection(hostname=host['address'],username=username, port=port) |
|---|
| | | rm.connect() |
|---|
| | | rm.send_file(__file__,'remote_control.py') |
|---|
| | | # print ("Sendind file:"+main.__file__) |
|---|
| | | rm.send_file(main.__file__,'remote_control.py') |
|---|
| | | for run in host['runs']: |
|---|
| | | try: |
|---|
| | | rm.send_file(run.tapeFile,run.tapeFile) |
|---|
| | |
|---|
| | | if target_runs==None: |
|---|
| | | target_runs=list(range(1,len(host['runs'])+1)) |
|---|
| | | report=[] |
|---|
| | | print("was here") |
|---|
| | | # print("was here") |
|---|
| | | for i in target_runs: |
|---|
| | | line=host['runs'][i-1].getStatistics() |
|---|
| | | line.insert(0,i) |
|---|
| | |
|---|
| | | print("Cannot start WebServer in python 2.7") |
|---|
| | | exit(0) |
|---|
| | | |
|---|
| | | def perform_action(args,host): |
|---|
| | | |
|---|
| | | def analyze(args,host,a_dict, analysis,hosts): |
|---|
| | | if len(a_dict)==0: |
|---|
| | | print ('Error: no analyses are specified in the tsmgr.start()!') |
|---|
| | | exit(1) |
|---|
| | | target_runs=getTargetRunIdxList(args) |
|---|
| | | if target_runs==None: |
|---|
| | | target_runs=list(range(1,len(host['runs'])+1)) |
|---|
| | | for i in target_runs: |
|---|
| | | |
|---|
| | | for anal in analysis: |
|---|
| | | if(anal not in a_dict): |
|---|
| | | print("Analysis '"+anal+"' is not known. Available analyses: "+", ".join(a_dict.keys())+".") |
|---|
| | | exit(0) |
|---|
| | | for anal in analysis: |
|---|
| | | retval=a_dict[anal](host['runs'][i-1],host=host, args=args, hosts=hosts) |
|---|
| | | #try: |
|---|
| | | if(retval): |
|---|
| | | exit(0) |
|---|
| | | #except: |
|---|
| | | # pass |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | def perform_action(args,host,**kwargs): |
|---|
| | | #find which flags have been used and act upon them. -r -s -k -v -c --delete-comment are mutually exclusive, so only one of them is active |
|---|
| | | if args['run']: |
|---|
| | | run_processes(args,host) |
|---|
| | |
|---|
| | | elif args['jump_to_ipython']: |
|---|
| | | print('Jumping to shell...') |
|---|
| | | embed() |
|---|
| | | exit(0) |
|---|
| | | elif args['analysis']!= None: |
|---|
| | | analyze(args,host,kwargs.get('analyses', {}), args['analysis'],kwargs.get('hosts',None)) |
|---|
| | | exit(0) |
|---|
| | | else: #version requested |
|---|
| | | print(getTrisurfVersion()) |
|---|
| | |
|---|
| | | |
|---|
| | | |
|---|
| | | |
|---|
| | | def start(hosts,argv=sys.argv[1:]): |
|---|
| | | def start(hosts,argv=sys.argv[1:], analyses={}): |
|---|
| | | args=vars(ParseCLIArguments(argv)) |
|---|
| | | #print(vars(args)) |
|---|
| | | |
|---|
| | | #Backward compatibility... If running just on localmode, the host specification is unnecessary. Check if only Runs are specified |
|---|
| | | try: |
|---|
| | | test_host=hosts[0]['name'] |
|---|
| | | except: |
|---|
| | | print("Network mode disabled. Old syntax detected.") |
|---|
| | | host={'name':socket.gethostname(),'address':'127.0.0.1', 'runs':hosts} |
|---|
| | | perform_action(args,host) |
|---|
| | | exit(0) |
|---|
| | | |
|---|
| | | print("Old syntax detected.") |
|---|
| | | hosts=({'name':socket.gethostname(),'address':'127.0.0.1', 'runs':hosts},) |
|---|
| | | |
|---|
| | | #find the host at which the action is attended |
|---|
| | | if args['host']==None: |
|---|
| | | #Only status and version commands are automatically executed on all the hosts. stopping or starting or other actions is not! |
|---|
| | | if(args['status']==False and args['version']==False): |
|---|
| | | hosts=getListOfHostConfigurationByHostname(hosts,socket.gethostname()) |
|---|
| | | else: |
|---|
| | |
|---|
| | | if len(hosts)==0: |
|---|
| | | print ('Hostname "{}" does not exist in configuration file. Please check the spelling'.format(args['host'][0])) |
|---|
| | | exit(1) |
|---|
| | | |
|---|
| | | if not args['local_only']: |
|---|
| | | hosts=copyConfigAndConnect(hosts) |
|---|
| | | |
|---|
| | | #do local stuff: |
|---|
| | | for host in hosts: |
|---|
| | | if host['name'] == socket.gethostname(): |
|---|
| | | if(args['html']): |
|---|
| | | print("Host <font color='orange'>"+host['name']+"</font> reports the following:") |
|---|
| | | print("Host <font color='orange'>"+host['name']+"</font> reports:") |
|---|
| | | else: |
|---|
| | | print("Host "+bcolors.WARNING+host['name']+bcolors.ENDC+" reports the following:") |
|---|
| | | perform_action(args,host) |
|---|
| | | print("Host "+bcolors.WARNING+host['name']+bcolors.ENDC+" reports:") |
|---|
| | | perform_action(args,host, analyses=analyses, hosts=hosts) |
|---|
| | | elif not args['local_only']: |
|---|
| | | output=host['_conn'].execute('python3 ./remote_control.py -x '+" ".join(argv)) |
|---|
| | | for line in output: |
|---|
