Homec4science
Diffusion blackdynamite (previous) Browse

R3127/bin/enterRun.py
a707f8ad9978previous

Last Change
Raw File

enterRun.py
View Options

#!/usr/bin/env python
import BlackDynamite as BD
import subprocess,os,sys
################################################################
parser = BD.BDParser()
parser.register_params(group="getRunInfo",
params={"run_id":int, "order":str},
mandatory={"run_id":True},
help={"run_id":"Select a run_id for switching to it"})
params = parser.parseBDParameters()
mybase = BD.Base(**params)
myrun = BD.Run(mybase)
run_id = params['run_id']
myrun["id"] = run_id
myrun.id = run_id
run_list = myrun.getMatchedObjectList()
if (len(run_list) == 0):
print ("no run found with id " + str(run_id))
sys.exit(1)
run = run_list[0]
myjob = BD.Job(mybase)
job_id = run['job_id']
myjob["id"] = job_id
myjob.id = job_id
job_list = myjob.getMatchedObjectList()
if (len(job_list) == 0):
print ("no job found with id " + str(job_id))
sys.exit(1)
job = job_list[0]
mybase.close()
separator = '-'*30
print separator
print "JOB INFO"
print separator
print job
print separator
print "RUN INFO"
print separator
print run
print separator
print "LOGGING TO '{0}'".format(run['machine_name'])
print separator
if run['state'] == 'CREATED':
print "Cannot enter run: not yet started"
sys.exit(-1)
bashrc_filename = os.path.join('/tmp','bashrc.run{0}'.format(run_id))
bashrc = open(bashrc_filename,'w')
bashrc.write('export PS1="\\u@\\h:<{0}|RUN-{1}> $ "\n'.format(params['study'],run_id))
bashrc.write('cd {0}\n'.format(run['run_path']))
bashrc.write('echo ' + separator)
bashrc.close()
subprocess.call('scp -q {0} {1}:{0}'.format(bashrc_filename,run['machine_name']),shell=True)
command_login = 'ssh -t {0} "bash --rcfile {1} -i"'.format(run['machine_name'],bashrc_filename)
#print command_login
subprocess.call(command_login,shell=True)
c4science · Help