ud1

lib/FSM.py

@@ -87,7 +87,8 @@ class ProgFSM(Char):
         'Reboot',
         'Ping', 
         'Wifi',
-        'Defender'
+        'Defender',
+        'Update'
         ]
         
         for s in state_list:

lib/StatesProg.py

@@ -14,6 +14,8 @@ import ast
 import random
 from pathlib import Path
 import socket
+import git 
+
 #====================================     
 
 State = type("State", (object,), {})
@@ -591,6 +593,8 @@ def util_command(self, o, command):
 		self.FSM.ToTransition('toSelTheme')		
 	elif command == "Reboot":
 		self.FSM.ToTransition('toReboot')		
+	elif command == "Update":
+		self.FSM.ToTransition('toUpdate')	
 		#os.system("sudo reboot")
 	elif command == "Shutdown":
 		self.FSM.ToTransition('toShutdown')		
@@ -2861,3 +2865,61 @@ class Defender(State):
 		self.FSM.owner.pub.unregister("beat", self)        
 
 #==================================== 
+#==================================== 
+
+
+
+class Update(State):
+	def __init__(self,FSM):
+		super(Update, self).__init__(FSM)    
+		
+	def Enter(self):
+		
+		o = self.FSM.owner 
+		o.header_text = "Update"
+		o.pub.register("beat", self)
+		g = git.cmd.Git("/home/pi/zpc_ct/")
+		self.msg = g.pull()
+		if o.keyState[16] > 0 or o.keyState[17] > 0:
+			pass
+		else:
+			draw_header(o)
+			o.update_display(0)
+		super(Update, self).Enter()        
+		
+	def Execute(self):
+		o = self.FSM.owner 
+		menu1_actions(self, o)
+		if o.keyState[16] == 1:
+			draw_menu1(o)
+			o.update_display(0)
+		elif o.keyState[16] == 4:
+			draw_header(o)
+			#self.draw_footer(o)
+			self.draw_main(o)
+			o.update_display(0)
+
+		if o.keyState[0] == 1 or o.keyState[0] == 3 or o.keyState[1] == 1 or o.keyState[1] == 3 or o.keyState[2] == 1 or o.keyState[2] == 3 or o.keyState[3] == 1 or o.keyState[3] == 3:
+			draw_header(o)
+			#self.draw_footer(o)
+			self.draw_main(o)
+			o.update_display(0)	
+
+		if o.keyState[16] > 0 or o.keyState[17] > 0:
+			pass
+		else:			
+			pass
+			#o.update_display(0)
+	
+	def draw_main(self, o):
+		#o.center_text(self.msg, o.h1, o.width, 25, o.light_grey)
+		o.center_block(self.msg, o.h2, [0, 0, o.width, o.height], o.light_grey)
+
+	def ReceiveMessage(self, message):
+		o = self.FSM.owner 
+		play_seq(o, message)
+		
+	def Exit(self):
+		self.FSM.owner.pub.unregister("beat", self)        
+
+#====================================