added more file protection

intern-space-program · Dec 5, 2019 · d4e7f34 · d4e7f34
1 parent 7bc2821
commit d4e7f34
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/raspi/Final_Solution/flight_script.py b/raspi/Final_Solution/flight_script.py
@@ -75,10 +75,15 @@
 
 
 home = '/home/pi'
+
 store_dir = home + "/rocket_data"
 cmd = "mkdir " + store_dir
 os.system(cmd)
 
+store_dir = home + "/rocket_data/"+ str(absolute_tm[1]) + "-" + str(absolute_tm[2]) + "-" + str(absolute_tm[0]) + "_" + str(absolute_tm[3]) + "-" + str(absolute_tm[4]) + "-" + str(absolute_tm[5])
+cmd = "mkdir " + store_dir
+os.system(cmd)
+
 log_file = store_dir + "/system_log.txt"
 
 def get_time():
@@ -257,6 +262,8 @@ def store_buffer(self, mode):
 				return
 			self.stream_print("Storing READ Buffer: %d Bytes"%(self.get_buffer_size(STREAM_READ)))
 			self.read_file_handle.write(self.read_buffer.getvalue())
+			self.read_file_handle.close()
+			self.read_file_handle = open(self.read_store_file, 'ab')
 
 		if (mode & STREAM_WRITE == STREAM_WRITE):
 			if (not(self.mode & STREAM_WRITE)):
@@ -266,6 +273,8 @@ def store_buffer(self, mode):
 				return
 			self.stream_print("Storing WRITE Buffer: %d Bytes"%(self.get_buffer_size(STREAM_WRITE)))
 			self.write_file_handle.write(self.write_buffer.getvalue())
+			self.write_file_handle.close()
+			self.write_file_handle = open(self.write_store_file, 'ab')
 
 	def get_buffer_size(self, mode):
 		if (mode & STREAM_READ == STREAM_READ):