end game event

events/EndGameEvent.py

@@ -1,11 +1,15 @@
 from Event import Event
+from solenoids.MainFlipper import MainFlipper
+from solenoids.TopFlipper import TopFlipper
 
+#todo temporary
+import sys
 
 class EndGameEvent(Event):
-    def __init__(self, flipper):
-        self.flipper = flipper
+    def __init__(self):
         super().__init__("End Game Event")
 
     def trigger(self, _):
         super().trigger()
-        self.flipper.deactivate()
+        self.flipper.deactivate()
+        sys.exit(0)

events/EndOfBallEvent.py

@@ -1,5 +1,6 @@
 from Event import Event
 
+from events.EndGameEvent import EndGameEvent
 
 class EndOfBallEvent(Event):
     def __init__(self, playerState):
@@ -9,3 +10,6 @@ class EndOfBallEvent(Event):
     def trigger(self, _):
         super().trigger()
         self.playerState().removeBall()
+
+        if self.playerState().ballsLeft == 0:
+            EndGameEvent().trigger()