X-Git-Url: https://git.ralfj.de/saartuer.git/blobdiff_plain/cf244d9312e36b30583da60d3b9918db57ab3610..7d24ee0e013ba6d9442a4822f44818e10772e3d6:/statemachine.py

diff --git a/statemachine.py b/statemachine.py
index 850eb0e..5226576 100644
--- a/statemachine.py
+++ b/statemachine.py
@@ -99,16 +99,30 @@ class StateMachine():
 			else:
 				raise Exception("Unknown command number: %d" % ev)
 	
-	class AbstractLockedState(State):
+	class AbstractNonStartState(State):
+		def handle_pins_event(self):
+			if self.pins().door_locked != (not self.pins().space_active):
+				self.actor().act(Actor.CMD_RED_ON)
+			else:
+				self.actor().act(Actor.CMD_RED_OFF)
+			return super().handle_pins_event()
+	
+	class AbstractLockedState(AbstractNonStartState):
 		'''A state with invariant "The space is locked", switching to StateAboutToOpen when the space becomes unlocked'''
+		def __init__(self, sm, nervlist = None):
+			super().__init__(sm, nervlist)
+			self.actor().act(Actor.CMD_GREEN_OFF)
 		def handle_pins_event(self):
 			if not self.pins().door_locked:
 				logger.info("Door unlocked, space is about to open")
 				return StateMachine.StateAboutToOpen(self.state_machine)
 			return super().handle_pins_event()
 	
-	class AbstractUnlockedState(State):
+	class AbstractUnlockedState(AbstractNonStartState):
 		'''A state with invariant "The space is unlocked", switching to StateZu when the space becomes locked'''
+		def __init__(self, sm, nervlist = None):
+			super().__init__(sm, nervlist)
+			self.actor().act(Actor.CMD_GREEN_ON)
 		def handle_pins_event(self):
 			if self.pins().door_locked:
 				logger.info("Door locked, closing space")