import sys, getopt
|
import tabulate
|
import subprocess,re
|
import psutil
|
|
def printHelp():
|
print('Python module tsmgr accept following switches:\n')
|
print('tsmgr [-n process number] [-R] [-h] [-r] [-s] [-c comment text] [-a comment text]\n')
|
print('[-n process number]: number of process for which -s -r -c or -a switch apply. Should be placed before any other switch');
|
print('[-R] : raw output for -s switch');
|
print('[-r] : run process');
|
print('[-s] : process status');
|
print('[-k] : kill process');
|
print('[-c comment text] : write new comment for process');
|
print('[-a comment text] : append additional comment for process');
|
print('[-h] : print help');
|
|
|
def getTrisurfVersion():
|
p = subprocess.Popen('trisurf --version', shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
lines=p.stdout.readlines()
|
version=re.findall(r'[0-9a-f]{7}(?:-dirty)?', lines[0].decode('ascii'))
|
p.wait()
|
if(len(version)):
|
return version[0]
|
else:
|
return "unknown version"
|
|
def start(Runs):
|
argv=sys.argv[1:]
|
processno=0
|
raw=False
|
try:
|
opts, args = getopt.getopt(argv,"Ra:n:hrskc:")
|
except getopt.GetoptError:
|
printHelp()
|
sys.exit(2)
|
for opt, arg in opts:
|
if opt == '-R':
|
raw=True
|
elif opt == '-h':
|
printHelp()
|
sys.exit()
|
elif opt == '-r':
|
if processno:
|
localRuns=[Runs[processno-1]]
|
else:
|
localRuns=Runs
|
for run in localRuns:
|
run.start()
|
elif opt == '-s':
|
report=[]
|
i=1
|
if processno:
|
localRuns=[Runs[processno-1]]
|
else:
|
localRuns=Runs
|
for run in localRuns:
|
line=run.getStatistics()
|
line.insert(0,i)
|
report.append(line)
|
i=i+1
|
if(raw):
|
print(report)
|
else:
|
print ("\n\nTrisurf running processes report\n")
|
print (tabulate.tabulate(report,headers=["Run no.", "Run start time", "ETA", "Status", "PID", "Path", "Comment"], tablefmt='fancy_grid'))
|
elif opt == '-n':
|
processno=int(arg)
|
if processno<1 or processno>len(Runs) :
|
processno=0
|
elif opt == '-c':
|
comment = arg
|
if processno:
|
Runs[processno-1].writeComment(arg)
|
elif opt == '-a':
|
comment = arg
|
if processno:
|
Runs[processno-1].writeComment("\n"+arg, 'a')
|
|
elif opt == '-k':
|
if processno:
|
p=psutil.Process(Runs[processno-1].getPID())
|
p.kill()
|
else:
|
printHelp()
|
sys.exit(2)
|
