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