Add replay ball

events/EndBallEvent.py

@@ -1,7 +1,6 @@
 import config
 from Event import Event
 
-from events.EndGameEvent import EndGameEvent
 from events.StartBallEvent import StartBallEvent
 
 from solenoids.MainFlipper import MainFlipper
@@ -33,7 +32,7 @@ class EndBallEvent(Event):
         self.gameState.currentPlayer.deactivate()
 
         if self.gameState.currentPlayer.ballsLeft == 0 and self.gameState.currentPlayer.id == config.MAX_PLAYERS - 1:
-            EndGameEvent().trigger(None)
+            self.gameState.endGame()
             return
 
         self.gameState.nextPlayer()

events/EndGameEvent.py

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