major changes! added save/load capability

1 files changed, 39 insertions, 2 deletions

diff --git a/aslclass.py b/aslclass.py
index c253970..3307d7c 100644
--- a/aslclass.py
+++ b/aslclass.py
@@ -9,23 +9,51 @@ import os
 import msvcrt
 from timeit import timeit as time
 from time import clock as clock
+from paths import listpaths
+import re
 
 class ASLreport:
 	def __init__(self):
 		self.shotcount=0
 		self.pauseinterval=[]
 		self.shotlist=[]
+		self.totaltime=0
 	
 	def metadata(self):
-		self.name=input("\nenter movie name:\n")
-		self.report=input("\nwant to generate report file?(y/n)\n")
+		load=input("Do you want to start a new state? (y/n +enter) (n=load state)\n")
+		if load=='y': pass
+		elif load=='n': self.loader()
+		self.name=input("\nEnter movie name:\n")
 		print("press enter to start")
 		keystroke=str(msvcrt.getch())[-2]
 		if keystroke=="r":
 			print("counter begins")
 			self.key()
 		else: print("Unknown keystroke ")
+
+	def loader(self):
+		pathlist=os.listdir(".")
+		print(pathlist)
+		serial=input("Enter the serial number of the savefile (starts from 0)\n")
+		if listpaths('.')[int(serial)][-9:len(listpaths('.')[int(serial)])]!="savestate": 
+			print("unknown filetype")
+			sys.exit(1)
+		loader=open(listpaths('.')[int(serial)], mode="r").read()
+		data=re.search(r"(?P<MovieName>.+)\n(?P<TotalTime>.+)\n(?P<ShotCount>\d+)",loader)
+		try:
+			self.name=data.group('MovieName')
+			self.totaltime=float(data.group('TotalTime'))
+			self.shotcount=int(data.group('ShotCount'))
+		except AttributeError: 
+			print("unknown filetype")
+			sys.exit(0)
+		print("press enter to start loaded state")
+		if str(msvcrt.getch())[-2]=="r":
+			print("loaded state")
+			self.key()
+		else: print("Unknown keystroke ")
 		
+	
 	def key(self):
 		clock()
 		tottime=0
@@ -59,6 +87,7 @@ class ASLreport:
 		else: print("Unknown keystroke")
 	
 	def reset(self):
+		print('\n==ASLreport for "%s"=='%self.name)
 		self.totaltime=clock()
 		for pausetime in self.pauseinterval: 
 			self.totaltime=self.totaltime-pausetime
@@ -68,12 +97,20 @@ class ASLreport:
 		self.ASL=self.totaltime/self.shotcount
 		print("number of shots=%d"%self.shotcount)	
 		print("ASL =%fseconds"%self.ASL)
+		self.report=input("\nDo you want to generate report file?(y/n)\n")
 		if self.report=="y": self.reportgenerator()
+		elif self.report=="n": sys.exit(0)
 		
 	def reportgenerator(self):
 		fileopen=open(r"%s.ASLreport"%self.name,mode="w")
 		fileopen.write(
 		"ASL report for %s \n\n Record Time=%f\n Recorded Shots=%d\n ASL=%f\nShot Timestamps:\n%s"%(self.name,self.totaltime,self.shotcount,self.ASL,str(self.shotlist)))
+	
+	def savestate(self):
+		saver=open("%s.savestate"%self.name, mode="w")
+		saver.write("%s %f %f"%(self.name,self.totaltime,self.shotcount))
+	
+	
 
 if __name__=="__main__":
 	ASLreport().metadata()
\ No newline at end of file