From 3b83db7eaf329b61ffd66576fb7a8221311952a5 Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@gmail.com> Date: Sun, 15 May 2016 18:44:15 +0000 Subject: [PATCH] Nicer output --- python/trisurf/trisurf.py | 29 ++++++++++++++++++++--------- 1 files changed, 20 insertions(+), 9 deletions(-) diff --git a/python/trisurf/trisurf.py b/python/trisurf/trisurf.py index c57de3e..c70630a 100644 --- a/python/trisurf/trisurf.py +++ b/python/trisurf/trisurf.py @@ -13,6 +13,7 @@ import time import datetime import subprocess +import shutil # Process status TS_NOLOCK=0 # lock file does not exist @@ -323,8 +324,8 @@ return TS_NOLOCK if(psutil.pid_exists(int(pid))): proc= psutil.Process(int(pid)) - if proc.name()=="trisurf": - if proc.status()=="stopped": + if proc.name=="trisurf": + if proc.status=="stopped": return TS_STOPPED else: return TS_RUNNING @@ -335,6 +336,10 @@ def start(self): if(self.getStatus()==0 or self.getStatus()==TS_COMPLETED): + #check if executable exists + if(shutil.which('trisurf')==None): + print("Error. Trisurf executable not found in PATH. Please install trisurf prior to running trisurf manager.") + exit(1) self.Dir=Directory(maindir=self.maindir,simdir=self.subdir) #Symlinks tape file to the directory or create tape file from snapshot in the direcory... if(self.Dir.makeifnotexist()): @@ -347,7 +352,7 @@ else: try: with open (os.path.join(self.Dir.fullpath(),"tape"), "w") as myfile: - myfile.write("#This is automatically generated tape file from snapshot\n") + myfile.write("#This is automatically generated tape file from snapshot") myfile.write(str(self.tape)) except: print("Error -- cannot make tapefile "+ os.path.join(self.Dir.fullpath(),"tape")+" from the snapshot in the running directory") @@ -359,12 +364,12 @@ #check if the simulation has been completed. in this case notify user and stop executing. if(self.isCompleted() and ("--force-from-tape" not in self.runArgs) and ("--reset-iteration-count" not in self.runArgs)): - print("The simulation was completed. Not starting executable at localhost in "+self.Dir.fullpath()+"\n") + print("The simulation was completed. Not starting executable in "+self.Dir.fullpath()) return cwd=Directory(maindir=os.getcwd()) self.Dir.goto() - print("Starting trisurf-ng executable at localhost in "+self.Dir.fullpath()+"\n") + print("Starting trisurf-ng executable in "+self.Dir.fullpath()) if(self.fromSnapshot==True): params=["trisurf", "--restore-from-vtk",self.snapshotFile]+self.runArgs else: @@ -375,7 +380,7 @@ subprocess.Popen (params, stdout=subprocess.DEVNULL) cwd.goto() else: - print("Process already running. Not starting\n") + print("Process in "+self.Dir.fullpath()+" already running. Not starting.") return def stop(self): @@ -400,22 +405,28 @@ self.Dir=Directory(maindir=self.maindir,simdir=self.subdir) self.statistics=Statistics(self.Dir.fullpath(), statfile) self.Comment=FileContent(os.path.join(self.Dir.fullpath(),".comment")) - pid=self.getPID(); + pid=self.getPID() status=self.getStatus() + if(self.statistics.fileOK): + ETA=str(datetime.timedelta(microseconds=(int(self.tape.config['iterations'])-int(self.statistics.last))*self.statistics.dT)*1000000) if(status==TS_NONEXISTANT or status==TS_NOLOCK): statustxt="Not running" pid="" + ETA="" elif status==TS_STOPPED: statustxt="Stopped" + ETA="N/A" elif status==TS_COMPLETED: statustxt="Completed" + pid="" + ETA="" else: statustxt="Running" if(self.statistics.fileOK): - report=[time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(int(self.statistics.startDate))),str(datetime.timedelta(microseconds=(int(self.tape.config['iterations'])-int(self.statistics.last))*self.statistics.dT)*1000000), statustxt, pid, str(self.Dir.fullpath()), self.Comment.getText()] + report=[time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(int(self.statistics.startDate))),ETA, statustxt, pid, str(self.Dir.fullpath()), self.Comment.getText()] else: - report=["N/A","N/A\t",statustxt, pid, str(self.Dir.fullpath()), self.Comment.getText()] + report=["N/A","N/A",statustxt, pid, str(self.Dir.fullpath()), self.Comment.getText()] return report def writeComment(self, data, mode='w'): -- Gitblit v1.9.3