From ba73ab71fa4df14cbf55bf4111c791eea21a85c8 Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@fe.uni-lj.si> Date: Fri, 27 May 2016 08:20:01 +0000 Subject: [PATCH] Added some basic help on flags. Unformatted. --- python/trisurf/trisurf.py | 21 +++++++++++++-------- 1 files changed, 13 insertions(+), 8 deletions(-) diff --git a/python/trisurf/trisurf.py b/python/trisurf/trisurf.py index e3108b3..8b36d27 100644 --- a/python/trisurf/trisurf.py +++ b/python/trisurf/trisurf.py @@ -1,5 +1,3 @@ -#!/usr/bin/python3 - import configobj import xml.etree.ElementTree as ET import base64 @@ -84,6 +82,9 @@ ''' try: self.config=configobj.ConfigObj(tape) + with open (tape, "r") as myfile: + self.rawText=myfile.read() #read the file + except: print("Error reading or parsing tape file!\n") exit(1) @@ -91,6 +92,7 @@ def setTape(self, string): '''Method setTape(string) parses the string in memory that hold the tape contents.''' self.config=configobj.ConfigObj(io.StringIO(string)) + self.rawText=string return def getValue(self,key): @@ -364,13 +366,13 @@ else: try: with open (os.path.join(self.Dir.fullpath(),"tape"), "w") as myfile: - myfile.write("#This is automatically generated tape file from snapshot") - myfile.write(str(self.tape)) + #myfile.write("#This is automatically generated tape file from snapshot") + myfile.write(str(self.tape.rawText)) except: print("Error -- cannot make tapefile "+ os.path.join(self.Dir.fullpath(),"tape")+" from the snapshot in the running directory") exit(1) try: - os.symlink(os.path.abspath(self.snapshotFile), os.path.join(self.Dir.fullpath(),self.snapshotFile)) + os.symlink(os.path.abspath(self.snapshotFile), os.path.join(self.Dir.fullpath(),"initial_snapshot.vtu")) except: print("Error while symlinking "+os.path.abspath(self.snapshotFile)+" to "+os.path.join(self.Dir.fullpath(),self.snapshotFile)) @@ -383,7 +385,7 @@ self.Dir.goto() print("Starting trisurf-ng executable in "+self.Dir.fullpath()) if(self.fromSnapshot==True): - params=["trisurf", "--restore-from-vtk",self.snapshotFile]+self.runArgs + params=["trisurf", "--restore-from-vtk","initial_snapshot.vtu"]+self.runArgs else: #veify if dump exists. If not it is a first run and shoud be run with --force-from-tape if(os.path.isfile("dump.bin")==False): @@ -395,8 +397,6 @@ print("Process in "+self.Dir.fullpath()+" already running. Not starting.") return - def stop(self): - pass def setMaindir(self,prefix,variables): maindir="" @@ -441,6 +441,11 @@ report=["N/A","N/A",statustxt, pid, str(self.Dir.fullpath()), self.Comment.getText()] return report + + def stop(self): + p=psutil.Process(self.getPID()) + p.kill() + def writeComment(self, data, mode='w'): self.Dir=Directory(maindir=self.maindir,simdir=self.subdir) self.Comment=FileContent(os.path.join(self.Dir.fullpath(),".comment")) -- Gitblit v1.9.3