def play(self, level): # Display intro screen start_scene = StartScene(self.renderer) result = start_scene.run() if result not in ('quit'): # Display the main game screen game_scene = GameScene(self.renderer) result = game_scene.run(level - 1) if result in ('infected', 'died'): gameover_scene = GameOverScene(self.renderer, result) return gameover_scene.run() elif result in ('victory'): victory_scene = VictoryScene(self.renderer) return victory_scene.run() else: return result